interfaceを使うように変更

harakeishi · Feb 12, 2023 · 5f6850a · 5f6850a
1 parent 41dbd19
commit 5f6850a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/trv/db.go b/trv/db.go
@@ -86,7 +86,7 @@ func (d *DB) fetchDBInfo(client *github.Client, ctx context.Context, source Sour
 					continue
 				}
 				var table Table
-				schema, err := table.fetchTableInfoInMarkdownFromGitHub(client, ctx, source.Owner, source.Repo, path)
+				schema, err := table.fetchTableInfoInMarkdownFromGitHub(client.Repositories, ctx, source.Owner, source.Repo, path)
 				if err != nil {
 					return fmt.Errorf("fech DB info fail:%w", err)
 				}

diff --git a/trv/table.go b/trv/table.go
@@ -26,9 +26,13 @@ func (t Table) getFullName(i int) string {
 	return t.Name + "." + t.Columns[i].Name
 }
 
+type Repositories interface {
+	GetContents(ctx context.Context, owner string, repo string, path string, opt *github.RepositoryContentGetOptions) (fileContent *github.RepositoryContent, directoryContent []*github.RepositoryContent, resp *github.Response, err error)
+}
+
 // Get the schema information output in markdown from GitHub.
-func (t *Table) fetchTableInfoInMarkdownFromGitHub(client *github.Client, ctx context.Context, owner, repo, path string) (schema, error) {
-	content, _, _, err := client.Repositories.GetContents(ctx, owner, repo, path, nil)
+func (t *Table) fetchTableInfoInMarkdownFromGitHub(repositories Repositories, ctx context.Context, owner, repo, path string) (schema, error) {
+	content, _, _, err := repositories.GetContents(ctx, owner, repo, path, nil)
 	if err != nil {
 		return "", fmt.Errorf("fetch table info fail:%w", err)
 	}