How to use EXPECT method of dot_imports Package

1package locator_test2import (3	"go/ast"4	"strconv"5	"github.com/maxbrunsfeld/counterfeiter/model"6	. "github.com/maxbrunsfeld/counterfeiter/locator"7	. "github.com/onsi/ginkgo"8	. "github.com/onsi/gomega"9)10var _ = Describe("Locator", func() {11	Describe("finding a named interface in a file", func() {12		var interfaceName string13		var model *model.InterfaceToFake14		var err error15		JustBeforeEach(func() {16			model, err = GetInterfaceFromFilePath(interfaceName, "../fixtures/something.go")17		})18		Context("when it exists", func() {19			BeforeEach(func() {20				interfaceName = "Something"21			})22			It("should have the correct name", func() {23				Expect(model.Name).To(Equal("Something"))24			})25			It("should have the correct package name", func() {26				Expect(model.PackageName).To(Equal("fixtures"))27			})28			It("should have the correct import path", func() {29				Expect(model.ImportPath).To(HavePrefix("github.com"))30				Expect(model.ImportPath).To(HaveSuffix("counterfeiter/fixtures"))31			})32			It("should have the correct methods", func() {33				Expect(model.Methods).To(HaveLen(4))34				Expect(model.Methods[0].Field.Names[0].Name).To(Equal("DoThings"))35				Expect(model.Methods[0].Imports).To(HaveLen(1))36				Expect(model.Methods[1].Field.Names[0].Name).To(Equal("DoNothing"))37				Expect(model.Methods[1].Imports).To(HaveLen(1))38				Expect(model.Methods[2].Field.Names[0].Name).To(Equal("DoASlice"))39				Expect(model.Methods[2].Imports).To(HaveLen(1))40				Expect(model.Methods[3].Field.Names[0].Name).To(Equal("DoAnArray"))41				Expect(model.Methods[3].Imports).To(HaveLen(1))42			})43			It("does not return an error", func() {44				Expect(err).ToNot(HaveOccurred())45			})46		})47		Context("when it does not exist", func() {48			BeforeEach(func() {49				interfaceName = "GARBAGE"50			})51			It("returns an error", func() {52				Expect(err).To(HaveOccurred())53			})54		})55	})56	Describe("finding an interface described by a named function from a file", func() {57		var interfaceName string58		var model *model.InterfaceToFake59		var err error60		JustBeforeEach(func() {61			model, err = GetInterfaceFromFilePath(interfaceName, "../fixtures/request_factory.go")62		})63		Context("when it exists", func() {64			BeforeEach(func() {65				interfaceName = "RequestFactory"66			})67			It("returns a model representing the named function alias", func() {68				Expect(model.Name).To(Equal("RequestFactory"))69				Expect(model.RepresentedByInterface).To(BeFalse())70			})71			It("should have a single method", func() {72				Expect(model.Methods).To(HaveLen(1))73				Expect(model.Methods[0].Field.Names[0].Name).To(Equal("RequestFactory"))74				Expect(model.Methods[0].Imports).To(HaveLen(1))75			})76			It("does not return an error", func() {77				Expect(err).ToNot(HaveOccurred())78			})79		})80		Context("when it does not exist", func() {81			BeforeEach(func() {82				interfaceName = "Whoops!"83			})84			It("returns an error", func() {85				Expect(err).To(HaveOccurred())86			})87		})88	})89	Describe("finding an interface with duplicate imports", func() {90		var model *model.InterfaceToFake91		var err error92		JustBeforeEach(func() {93			model, err = GetInterfaceFromFilePath("AB", "../fixtures/dup_packages/dup_packagenames.go")94			Expect(err).NotTo(HaveOccurred())95		})96		It("returns a model representing the named function alias", func() {97			Expect(model.Name).To(Equal("AB"))98			Expect(model.RepresentedByInterface).To(BeTrue())99		})100		It("should have methods", func() {101			Expect(model.Methods).To(HaveLen(4))102			Expect(model.Methods[0].Field.Names[0].Name).To(Equal("A"))103			Expect(collectImports(model.Methods[0].Imports)).To(ConsistOf(104				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages/a/v1",105				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages/b/v1",106				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages"))107			Expect(model.Methods[1].Field.Names[0].Name).To(Equal("FromA"))108			Expect(collectImports(model.Methods[1].Imports)).To(ConsistOf(109				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages/a/v1"))110			Expect(model.Methods[2].Field.Names[0].Name).To(Equal("B"))111			Expect(collectImports(model.Methods[2].Imports)).To(ConsistOf(112				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages/a/v1",113				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages/b/v1",114				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages"))115			Expect(model.Methods[3].Field.Names[0].Name).To(Equal("FromB"))116			Expect(collectImports(model.Methods[3].Imports)).To(ConsistOf(117				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages/b/v1"))118		})119	})120	Describe("finding an interface with duplicate indirect imports", func() {121		var model *model.InterfaceToFake122		var err error123		JustBeforeEach(func() {124			model, err = GetInterfaceFromFilePath("DupAB", "../fixtures/dup_packages/dupAB.go")125			Expect(err).NotTo(HaveOccurred())126		})127		It("returns a model representing the named function alias", func() {128			Expect(model.Name).To(Equal("DupAB"))129			Expect(model.RepresentedByInterface).To(BeTrue())130		})131		It("should have methods", func() {132			Expect(model.Methods).To(HaveLen(2))133			Expect(model.Methods[0].Field.Names[0].Name).To(Equal("A"))134			Expect(collectImports(model.Methods[0].Imports)).To(ConsistOf(135				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages/a/v1",136				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages"))137			Expect(model.Methods[1].Field.Names[0].Name).To(Equal("B"))138			Expect(collectImports(model.Methods[1].Imports)).To(ConsistOf(139				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages/b/v1",140				"github.com/maxbrunsfeld/counterfeiter/fixtures/dup_packages"))141		})142	})143	Describe("finding an interface with dot imports", func() {144		var model *model.InterfaceToFake145		var err error146		JustBeforeEach(func() {147			model, err = GetInterfaceFromFilePath("DotImports", "../fixtures/dot_imports.go")148			Expect(err).NotTo(HaveOccurred())149		})150		It("returns a model representing the named function alias", func() {151			Expect(model.Name).To(Equal("DotImports"))152			Expect(model.RepresentedByInterface).To(BeTrue())153		})154		It("should have a single method", func() {155			Expect(model.Methods).To(HaveLen(1))156			// Expect(model.Methods[0].Names[0].Name).To(Equal("DoThings"))157		})158	})159	Describe("finding an interface in vendored code", func() {160		var model *model.InterfaceToFake161		var err error162		Context("when the vendor dir is in the same directory", func() {163			JustBeforeEach(func() {164				model, err = GetInterfaceFromFilePath("FooInterface", "../fixtures/vendored/foo.go")165				Expect(err).NotTo(HaveOccurred())166			})167			It("returns a model representing the named function alias", func() {168				Expect(model.Name).To(Equal("FooInterface"))169				Expect(model.RepresentedByInterface).To(BeTrue())170			})171			It("should have a single method", func() {172				Expect(model.Methods).To(HaveLen(1))173				Expect(model.Methods[0].Field.Names[0].Name).To(Equal("FooVendor"))174			})175		})176		Context("when the vendor dir is in a parent directory", func() {177			JustBeforeEach(func() {178				model, err = GetInterfaceFromFilePath("BazInterface", "../fixtures/vendored/baz/baz.go")179				Expect(err).NotTo(HaveOccurred())180			})181			It("returns a model representing the named function alias", func() {182				Expect(model.Name).To(Equal("BazInterface"))183				Expect(model.RepresentedByInterface).To(BeTrue())184			})185			It("should have a single method", func() {186				Expect(model.Methods).To(HaveLen(1))187				Expect(model.Methods[0].Field.Names[0].Name).To(Equal("FooVendor"))188			})189		})190		Context("when the vendor code shadows a higher level", func() {191			JustBeforeEach(func() {192				model, err = GetInterfaceFromFilePath("BarInterface", "../fixtures/vendored/bar/bar.go")193				Expect(err).NotTo(HaveOccurred())194			})195			It("returns a model representing the named function alias", func() {196				Expect(model.Name).To(Equal("BarInterface"))197				Expect(model.RepresentedByInterface).To(BeTrue())198			})199			It("should have a single method", func() {200				Expect(model.Methods).To(HaveLen(1))201				Expect(model.Methods[0].Field.Names[0].Name).To(Equal("BarVendor"))202			})203		})204	})205})206func collectImports(specs map[string]*ast.ImportSpec) []string {207	imports := []string{}208	for _, v := range specs {209		s, err := strconv.Unquote(v.Path.Value)210		Expect(err).NotTo(HaveOccurred())211		imports = append(imports, s)212	}213	return imports214}...

1package integration_test2import (3	"fmt"4	"go/build"5	"io/ioutil"6	"log"7	"os"8	"path/filepath"9	"strings"10	"testing"11	. "github.com/onsi/gomega"12	"github.com/sclevine/spec"13	"github.com/maxbrunsfeld/counterfeiter/v6/generator"14)15func runTests(t *testing.T, when spec.G, it spec.S) {16	log.SetOutput(ioutil.Discard) // Comment this out to see verbose log output17	log.SetFlags(log.Llongfile)18	var (19		baseDir             string20		relativeDir         string21		originalGopath      string22		originalBuildGopath string23		originalGo111module string24		testDir             string25		copyDirFunc         func()26		copyFileFunc        func(name string)27		initModuleFunc      func()28		writeToTestData     bool29	)30	name := "working with a module"31	it.Before(func() {32		RegisterTestingT(t)33		originalGo111module = os.Getenv("GO111MODULE")34		os.Setenv("GO111MODULE", "on")35		originalGopath = os.Getenv("GOPATH")36		originalBuildGopath = build.Default.GOPATH37		var err error38		testDir, err = ioutil.TempDir("", "counterfeiter-integration")39		Expect(err).NotTo(HaveOccurred())40		os.Unsetenv("GOPATH")41		baseDir = testDir42		err = os.MkdirAll(baseDir, 0777)43		Expect(err).ToNot(HaveOccurred())44		relativeDir = filepath.Join("..", "fixtures")45		copyDirFunc = func() {46			err = os.MkdirAll(baseDir, 0777)47			Expect(err).ToNot(HaveOccurred())48			err = Copy(relativeDir, baseDir)49			Expect(err).ToNot(HaveOccurred())50		}51		copyFileFunc = func(name string) {52			dir := baseDir53			d := filepath.Dir(name)54			if d != "." {55				dir = filepath.Join(dir, d)56			}57			err = os.MkdirAll(dir, 0777)58			Expect(err).ToNot(HaveOccurred())59			b, err := ioutil.ReadFile(filepath.Join(relativeDir, name))60			Expect(err).ToNot(HaveOccurred())61			err = ioutil.WriteFile(filepath.Join(baseDir, name), b, 0755)62			Expect(err).ToNot(HaveOccurred())63		}64		initModuleFunc = func() {65			copyFileFunc("blank.go")66			err := ioutil.WriteFile(filepath.Join(baseDir, "go.mod"), []byte("module github.com/maxbrunsfeld/counterfeiter/v6/fixtures"), 0755)67			Expect(err).ToNot(HaveOccurred())68		}69		// Set this to true to write the output of tests to the testdata/output70		// directory 🙃 happy debugging!71		// writeToTestData = true72	})73	it.After(func() {74		if originalGo111module != "" {75			os.Setenv("GO111MODULE", originalGo111module)76		} else {77			os.Unsetenv("GO111MODULE")78		}79		if originalGopath != "" {80			os.Setenv("GOPATH", originalGopath)81		} else {82			os.Unsetenv("GOPATH")83		}84		build.Default.GOPATH = originalBuildGopath85		if baseDir == "" {86			return87		}88		err := os.RemoveAll(testDir)89		Expect(err).ToNot(HaveOccurred())90	})91	when("generating a fake for stdlib interfaces", func() {92		const (93			noHeader   = "noheader"94			withHeader = "header"95		)96		t := func(header, variant string) {97			it("succeeds", func() {98				initModuleFunc()99				cache := &generator.FakeCache{}100				f, err := generator.NewFake(generator.InterfaceOrFunction, "WriteCloser", "io", "FakeWriteCloser", "custom", header, baseDir, cache)101				Expect(err).NotTo(HaveOccurred())102				b, err := f.Generate(true) // Flip to false to see output if goimports fails103				Expect(err).NotTo(HaveOccurred())104				if writeToTestData {105					WriteOutput(b, filepath.Join("testdata", "output", "write_closer", "actual."+variant+".go"))106				}107				WriteOutput(b, filepath.Join(baseDir, "fixturesfakes", "fake_write_closer."+variant+".go"))108				RunBuild(baseDir)109				b2, err := ioutil.ReadFile(filepath.Join("testdata", "expected_fake_writecloser."+variant+".txt"))110				Expect(err).NotTo(HaveOccurred())111				Expect(string(b2)).To(Equal(string(b)))112			})113		}114		t("", noHeader)115		t("// some header\n//\n\n", withHeader)116	})117	when("generating an interface for a package", func() {118		it("succeeds", func() {119			initModuleFunc()120			cache := &generator.FakeCache{}121			f, err := generator.NewFake(generator.Package, "", "os", "Os", "custom", "", baseDir, cache)122			Expect(err).NotTo(HaveOccurred())123			b, err := f.Generate(true) // Flip to false to see output if goimports fails124			Expect(err).NotTo(HaveOccurred())125			if writeToTestData {126				WriteOutput(b, filepath.Join("testdata", "output", "package_mode", "actual.go"))127			}128			WriteOutput(b, filepath.Join(baseDir, "fixturesfakes", "fake_os.go"))129			RunBuild(baseDir)130		})131	})132	when(name, func() {133		t := func(interfaceName string, filename string, subDir string, files ...string) {134			when("working with "+filename, func() {135				it.Before(func() {136					if subDir != "" {137						baseDir = filepath.Join(baseDir, subDir)138						relativeDir = filepath.Join(relativeDir, subDir)139					}140					log.Println(testDir)141					copyFileFunc(filename)142					for i := range files {143						copyFileFunc(files[i])144					}145				})146				it("succeeds", func() {147					suffix := strings.Replace(subDir, "\\", "/", -1)148					if suffix != "" {149						suffix = "/" + suffix150					}151					WriteOutput([]byte(fmt.Sprintf("module github.com/maxbrunsfeld/counterfeiter/v6/fixtures%s\n", suffix)), filepath.Join(baseDir, "go.mod"))152					cache := &generator.FakeCache{}153					f, err := generator.NewFake(generator.InterfaceOrFunction, interfaceName, fmt.Sprintf("github.com/maxbrunsfeld/counterfeiter/v6/fixtures%s", suffix), "Fake"+interfaceName, "fixturesfakes", "", baseDir, cache)154					Expect(err).NotTo(HaveOccurred())155					b, err := f.Generate(true) // Flip to false to see output if goimports fails156					Expect(err).NotTo(HaveOccurred())157					if writeToTestData {158						WriteOutput(b, filepath.Join("testdata", "output", strings.Replace(filename, ".go", "", -1), "actual.go"))159					}160					WriteOutput(b, filepath.Join(baseDir, "fixturesfakes", "fake_"+filename))161					RunBuild(baseDir)162				})163			})164		}165		t("SomethingElse", "compound_return.go", "")166		t("DotImports", "dot_imports.go", "")167		t("EmbedsInterfaces", "embeds_interfaces.go", "", filepath.Join("another_package", "types.go"))168		t("AliasedInterface", "aliased_interfaces.go", "", filepath.Join("another_package", "types.go"))169		t("HasImports", "has_imports.go", "")170		t("HasOtherTypes", "has_other_types.go", "", "other_types.go")171		t("HasVarArgs", "has_var_args.go", "")172		t("HasVarArgsWithLocalTypes", "has_var_args.go", "")173		t("ImportsGoHyphenPackage", "imports_go_hyphen_package.go", "", filepath.Join("go-hyphenpackage", "fixture.go"))174		t("FirstInterface", "multiple_interfaces.go", "")175		t("SecondInterface", "multiple_interfaces.go", "")176		t("InlineStructParams", "inline_struct_params.go", "")177		t("RequestFactory", "request_factory.go", "")178		t("ReusesArgTypes", "reuses_arg_types.go", "")179		t("SomethingWithForeignInterface", "something_remote.go", "", filepath.Join("aliased_package", "in_aliased_package.go"))180		t("Something", "something.go", "")181		t("SomethingFactory", "typed_function.go", "")182		t("SyncSomething", "interface.go", "sync")183		when("working with duplicate packages", func() {184			t := func(interfaceName string, offset string, fakePackageName string) {185				when("working with "+interfaceName, func() {186					it.Before(func() {187						relativeDir = filepath.Join(relativeDir, "dup_packages")188						copyDirFunc()189					})190					it("succeeds", func() {191						pkgPath := "github.com/maxbrunsfeld/counterfeiter/v6/fixtures/dup_packages"192						if offset != "" {193							pkgPath = pkgPath + "/" + offset194						}195						cache := &generator.FakeCache{}196						f, err := generator.NewFake(generator.InterfaceOrFunction, interfaceName, pkgPath, "Fake"+interfaceName, fakePackageName, "", baseDir, cache)197						Expect(err).NotTo(HaveOccurred())198						b, err := f.Generate(false) // Flip to false to see output if goimports fails199						Expect(err).NotTo(HaveOccurred())200						if writeToTestData {201							WriteOutput(b, filepath.Join("testdata", "output", "dup_"+strings.ToLower(interfaceName), "actual.go"))202						}203						WriteOutput(b, filepath.Join(baseDir, offset, fakePackageName, "fake_"+strings.ToLower(interfaceName)+".go"))204						RunBuild(filepath.Join(baseDir, offset, fakePackageName))205					})206				})207			}208			t("MultiAB", "foo", "foofakes")209			t("AliasV1", "", "dup_packagesfakes")210		})211	})212}...

1package dot_imports2import (3func TestDotImports(t *testing.T) {4	RegisterTestingT(t)5	Expect(true).To(BeTrue())6}7package dot_imports8import (9func TestDotImports(t *testing.T) {10	RegisterTestingT(t)11	Assert(true)12}13package dot_imports14import (15func TestDotImports(t *testing.T) {16	RegisterTestingT(t)17	Ω(true).Should(BeTrue())18}19package dot_imports20import (21func TestDotImports(t *testing.T) {22	RegisterTestingT(t)23	Ω(true).Should(BeTrue())24}25package dot_imports26import (27func TestDotImports(t *testing.T) {28	RegisterTestingT(t)29	Ω(true).Should(BeTrue())30}31package dot_imports32import (33func TestDotImports(t *testing.T) {34	RegisterTestingT(t)35	Ω(true).Should(BeTrue())36}37package dot_imports38import (39func TestDotImports(t *testing.T) {40	RegisterTestingT(t)41	Ω(true).Should(BeTrue())42}43package dot_imports44import (45func TestDotImports(t *testing.T) {46	RegisterTestingT(t)

1import (2	. "github.com/GoLangTutorials/dot_imports"3func main() {4	fmt.Println("Hello World!")5	EXPECT(1, 2)6}

1import (2    . "github.com/sergey-dryabzhinsky/dot_imports"3func main() {4    fmt.Println("Hello, world!")5    EXPECT(1, 1)6    EXPECT(1, 2)7}8import (9    . "github.com/sergey-dryabzhinsky/dot_imports"10func main() {11    fmt.Println("Hello, world!")12    ASSERT(1, 1)13    ASSERT(1, 2)14}15import (16    . "github.com/sergey-dryabzhinsky/dot_imports"17func main() {18    fmt.Println("Hello, world!")19    EXPECT(1, 1)20    ASSERT(1, 1)21    EXPECT(1, 2)22    ASSERT(1, 2)23}24import (25    x "github.com/sergey-dryabzhinsky/dot_imports"26func main() {27    fmt.Println("Hello, world!")28    x.EXPECT(1, 1)29    x.ASSERT(1, 1)30    x.EXPECT(1, 2)31    x.ASSERT(1, 2)32}33import (34    . "github.com/sergey-dryabzhinsky/dot_imports"35func main() {36    fmt.Println("Hello, world!")37    EXPECT(1, 1)38    ASSERT(1, 1)39    EXPECT(1, 2)40    ASSERT(1, 2)41}42import (43    . "github.com/sergey-dryabzhinsky/dot_imports"

1import (2    . "github.com/deepakgupta1/dot_imports"3func main() {4    fmt.Println("Hello, World")5    EXPECT(1, 1, "1 is not equal to 1")6}7package dot_imports8import (9func EXPECT(a interface{}, b interface{}, message string) {10    if a != b {11        fmt.Println(message)12        os.Exit(1)13    }14}15import (16    "github.com/deepakgupta1/dot_imports"17func main() {18    fmt.Println("Hello, World")19    dot_imports.EXPECT(1, 1, "1 is not equal to 1")20}21package dot_imports22import (23func EXPECT(a interface{}, b interface{}, message string) {24    if a != b {25        fmt.Println(message)26        os.Exit(1)27    }28}29import (30    "github.com/deepakgupta1/dot_imports"31func main() {32    fmt.Println("Hello, World")33    dot_imports.EXPECT(1, 1, "1 is not equal to 1")34}35import (36    "github.com/deepakgupta1/dot_imports"37func main() {38    fmt.Println("

1import (2func main() {3    fmt.Println("Hello World")4    EXPECT("Hello World", "Hello World")5}6GoLangTutorials/DotImports/1.go:8:2: imported and not used: "github.com/GoLangTutorials/DotImports"7This is because you can not import the sa