From 50dbe5a65dbce77b445880cc702404de1e598e76 Mon Sep 17 00:00:00 2001
From: Ross Light <light@google.com>
Date: Mon, 2 Apr 2018 16:35:59 -0700
Subject: [PATCH] goose: add a test for imports across packages

Reviewed-by: Tuo Shan <shantuo@google.com>
Reviewed-by: Herbie Ong <herbie@google.com>
---
 internal/goose/testdata/PkgImport/bar/bar.go  |  8 ++++++
 internal/goose/testdata/PkgImport/foo/foo.go  | 26 +++++++++++++++++++
 .../goose/testdata/PkgImport/foo/foo_goose.go |  7 +++++
 internal/goose/testdata/PkgImport/out.txt     |  1 +
 internal/goose/testdata/PkgImport/pkg         |  1 +
 5 files changed, 43 insertions(+)
 create mode 100644 internal/goose/testdata/PkgImport/bar/bar.go
 create mode 100644 internal/goose/testdata/PkgImport/foo/foo.go
 create mode 100644 internal/goose/testdata/PkgImport/foo/foo_goose.go
 create mode 100644 internal/goose/testdata/PkgImport/out.txt
 create mode 100644 internal/goose/testdata/PkgImport/pkg

diff --git a/internal/goose/testdata/PkgImport/bar/bar.go b/internal/goose/testdata/PkgImport/bar/bar.go
new file mode 100644
index 0000000..c5a1de4
--- /dev/null
+++ b/internal/goose/testdata/PkgImport/bar/bar.go
@@ -0,0 +1,8 @@
+package bar
+
+type Bar int
+
+//goose:provide Bar
+func ProvideBar() Bar {
+	return 1
+}
diff --git a/internal/goose/testdata/PkgImport/foo/foo.go b/internal/goose/testdata/PkgImport/foo/foo.go
new file mode 100644
index 0000000..35e73a1
--- /dev/null
+++ b/internal/goose/testdata/PkgImport/foo/foo.go
@@ -0,0 +1,26 @@
+package main
+
+import (
+	"fmt"
+
+	"bar"
+)
+
+func main() {
+	fmt.Println(injectFooBar())
+}
+
+type Foo int
+type FooBar int
+
+//goose:provide Set
+func provideFoo() Foo {
+	return 41
+}
+
+//goose:import Set "bar".Bar
+
+//goose:provide Set
+func provideFooBar(foo Foo, barVal bar.Bar) FooBar {
+	return FooBar(foo) + FooBar(barVal)
+}
diff --git a/internal/goose/testdata/PkgImport/foo/foo_goose.go b/internal/goose/testdata/PkgImport/foo/foo_goose.go
new file mode 100644
index 0000000..73f5093
--- /dev/null
+++ b/internal/goose/testdata/PkgImport/foo/foo_goose.go
@@ -0,0 +1,7 @@
+//+build gooseinject
+
+package main
+
+//goose:use Set
+
+func injectFooBar() FooBar
diff --git a/internal/goose/testdata/PkgImport/out.txt b/internal/goose/testdata/PkgImport/out.txt
new file mode 100644
index 0000000..d81cc07
--- /dev/null
+++ b/internal/goose/testdata/PkgImport/out.txt
@@ -0,0 +1 @@
+42
diff --git a/internal/goose/testdata/PkgImport/pkg b/internal/goose/testdata/PkgImport/pkg
new file mode 100644
index 0000000..257cc56
--- /dev/null
+++ b/internal/goose/testdata/PkgImport/pkg
@@ -0,0 +1 @@
+foo