go-interview-problems/03-web-crawler/task_test.go

package main

import (
	"fmt"
	"reflect"
	"sort"
	"testing"
)

// fakeFetcher is Fetcher that returns canned results.
type fakeFetcher map[string]*fakeResult

type fakeResult struct {
	body string
	urls []string
}

func (f fakeFetcher) Fetch(url string) (string, []string, error) {
	if res, ok := f[url]; ok {
		return res.body, res.urls, nil
	}
	return "", nil, fmt.Errorf("not found: %s", url)
}

func TestCrawl(t *testing.T) {
	tests := []struct {
		name    string
		url     string
		depths  int
		fetcher fakeFetcher
		result  []string
		err     error
	}{
		{
			name:   "default",
			url:    "https://golang.org/",
			depths: 4,
			fetcher: fakeFetcher{
				"https://golang.org/": &fakeResult{
					"The Go Programming Language",
					[]string{
						"https://golang.org/pkg/",
						"https://golang.org/cmd/",
					},
				},
				"https://golang.org/pkg/": &fakeResult{
					"Packages",
					[]string{
						"https://golang.org/",
						"https://golang.org/cmd/",
						"https://golang.org/pkg/fmt/",
						"https://golang.org/pkg/os/",
					},
				},
				"https://golang.org/pkg/fmt/": &fakeResult{
					"Package fmt",
					[]string{
						"https://golang.org/",
						"https://golang.org/pkg/",
					},
				},
				"https://golang.org/pkg/os/": &fakeResult{
					"Package os",
					[]string{
						"https://golang.org/",
						"https://golang.org/pkg/",
					},
				},
			},
			result: []string{
				"The Go Programming Language",
				"Packages",
				"Package fmt",
				"Package os",
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result, err := Crawl(tt.url, tt.depths, tt.fetcher)
			if err != tt.err {
				t.Error(err)
			}
			sort.Strings(result)
			sort.Strings(tt.result)

			if !reflect.DeepEqual(tt.result, result) {
				t.Errorf("Wrong result. Expected: %+q, Got: %+q", tt.result, result)
			}
		})
	}
}