make seen a non-primitive type

This commit is contained in:
Haiyan Meng
2020-01-14 11:03:29 -08:00
parent 230e0ca752
commit 72eda992bd
4 changed files with 36 additions and 20 deletions

View File

@@ -75,7 +75,7 @@ func newCrawler(matchPrefix string, err error,
// Crawl implements the Crawler interface for testing.
func (c testCrawler) Crawl(_ context.Context,
output chan<- CrawledDocument, _ map[string]struct{}) error {
output chan<- CrawledDocument, _ SeenMap) error {
for i, d := range c.docs {
isResource := true
@@ -181,7 +181,7 @@ func TestCrawlGithubRunner(t *testing.T) {
defer close(output)
defer wg.Done()
seen := map[string]struct{}{}
seen := NewSeenMap()
errs := CrawlGithubRunner(context.Background(),
output, test.tc, seen)
@@ -323,7 +323,7 @@ resources:
visited[d.ID()]++
return nil
},
make(map[string]struct{}),
NewSeenMap(),
)
if lv, lc := len(visited), len(tc.corpus); lv != lc {
t.Errorf("error: %d of %d documents visited.", lv, lc)