move test data

rfebriya · Nov 16, 2019 · eeed493 · eeed493
1 parent 90d7a87
commit eeed493
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 11 deletions.
diff --git a/function_test.go b/function_test.go
@@ -7,13 +7,14 @@ import (
 
 	proto "github.com/micro/go-micro/debug/proto"
 	"github.com/micro/go-micro/registry/memory"
+	"github.com/micro/go-micro/util/test"
 )
 
 func TestFunction(t *testing.T) {
 	var wg sync.WaitGroup
 	wg.Add(1)
 
-	r := memory.NewRegistry(memory.Services(testData))
+	r := memory.NewRegistry(memory.Services(test.Data))
 
 	// create service
 	fn := NewFunction(

diff --git a/service.go b/service.go
@@ -14,8 +14,8 @@ import (
 	"github.com/micro/go-micro/debug/profile/pprof"
 	"github.com/micro/go-micro/plugin"
 	"github.com/micro/go-micro/server"
-	"github.com/micro/go-micro/util/wrapper"
 	"github.com/micro/go-micro/util/log"
+	"github.com/micro/go-micro/util/wrapper"
 )
 
 type service struct {

diff --git a/service_test.go b/service_test.go
@@ -11,6 +11,7 @@ import (
 	proto "github.com/micro/go-micro/debug/proto"
 	"github.com/micro/go-micro/registry/memory"
 	"github.com/micro/go-micro/util/log"
+	"github.com/micro/go-micro/util/test"
 )
 
 func testShutdown(wg *sync.WaitGroup, cancel func()) {
@@ -29,7 +30,7 @@ func testService(ctx context.Context, wg *sync.WaitGroup, name string) Service {
 	// add self
 	wg.Add(1)
 
-	r := memory.NewRegistry(memory.Services(testData))
+	r := memory.NewRegistry(memory.Services(test.Data))
 
 	// create service
 	return NewService(

diff --git a/common_test.go → util/test/test.go b/common_test.go → util/test/test.go
@@ -1,12 +1,12 @@
-package micro
+package test
 
 import (
 	"github.com/micro/go-micro/registry"
 )
 
 var (
-	// mock data
-	testData = map[string][]*registry.Service{
+	// mock registry data
+	Data = map[string][]*registry.Service{
 		"foo": {
 			{
 				Name:    "foo",

diff --git a/util/wrapper/wrapper.go b/util/wrapper/wrapper.go
@@ -49,9 +49,9 @@ func (c *clientWrapper) Publish(ctx context.Context, p client.Message, opts ...c
 // FromService wraps a client to inject From-Service header into metadata
 func FromService(name string, c client.Client) client.Client {
 	return &clientWrapper{
-                c,
-                metadata.Metadata{
-                        HeaderPrefix + "From-Service": name,
-                },
-        }
+		c,
+		metadata.Metadata{
+			HeaderPrefix + "From-Service": name,
+		},
+	}
 }