dep-drawer/dependencies_test.go at master · egon12/dep-drawer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package main

import (
	"sort"
	"testing"
)

func TestGetRecursiveDependencies(t *testing.T) {
	a := GetRecursiveDependencies("diffpackage")
	for k, v := range a {
		t.Errorf("%v: %v", k, v)
	}
}

func TestGetImports(t *testing.T) {
	got := GetImports(".")

	want := []string{
		"flag",
		"bytes",
		"os",
		"strings",
		"time",
		"io/ioutil",
		"github.com/egon12/dep-drawer/gomod",
		"html/template",
		"errors",
		"github.com/egon12/dep-drawer/browser",
		"compress/gzip",
		"fmt",
		"path/filepath",
		"go/ast",
		"io",
		"go/parser",
		"go/token",
		"log",
	}

	sort.Strings(got)
	sort.Strings(want)

	for i := range got {
		if got[i] != want[i] {
			t.Errorf("want: %v\n got: %v\n", want[i], got[i])
		}
	}
}

func TestGetImports_diffpackage(t *testing.T) {
	got := GetImports("./diffpackage")

	want := []string{
		"github.com/egon12/dep-drawer/diffpackage/diffpackage2",
		"github.com/egon12/dep-drawer/diffpackage/diffpackage3",
	}

	sort.Strings(got)

	for i := range got {
		if got[i] != want[i] {
			t.Errorf("want: %v\n got: %v\n", want[i], got[i])
		}
	}
}