1
0
mirror of https://github.com/go-micro/go-micro.git synced 2024-12-06 08:16:03 +02:00
go-micro/function_test.go

64 lines
986 B
Go
Raw Normal View History

2017-06-01 13:11:52 +02:00
package micro
import (
2018-03-03 13:53:52 +02:00
"context"
2017-06-01 13:11:52 +02:00
"sync"
"testing"
2019-08-06 18:53:14 +02:00
proto "github.com/micro/go-micro/debug/proto"
2019-01-14 17:27:25 +02:00
"github.com/micro/go-micro/registry/memory"
2017-06-01 13:11:52 +02:00
)
func TestFunction(t *testing.T) {
var wg sync.WaitGroup
wg.Add(1)
2019-01-14 17:27:25 +02:00
r := memory.NewRegistry()
r.(*memory.Registry).Services = testData
2019-01-14 17:27:25 +02:00
2017-06-01 13:11:52 +02:00
// create service
fn := NewFunction(
2019-01-14 17:27:25 +02:00
Registry(r),
2018-05-28 16:40:28 +02:00
Name("test.function"),
2017-06-01 13:11:52 +02:00
AfterStart(func() error {
wg.Done()
return nil
}),
)
// we can't test fn.Init as it parses the command line
// fn.Init()
2018-05-28 16:40:28 +02:00
ch := make(chan error, 2)
2017-06-01 13:11:52 +02:00
go func() {
2018-05-28 16:40:28 +02:00
// run service
ch <- fn.Run()
2017-06-01 13:11:52 +02:00
}()
2018-05-28 16:40:28 +02:00
// wait for start
wg.Wait()
// test call debug
req := fn.Client().NewRequest(
"test.function",
"Debug.Health",
new(proto.HealthRequest),
)
rsp := new(proto.HealthResponse)
err := fn.Client().Call(context.TODO(), req, rsp)
if err != nil {
t.Fatal(err)
}
if rsp.Status != "ok" {
t.Fatalf("function response: %s", rsp.Status)
}
if err := <-ch; err != nil {
2018-05-25 15:39:50 +02:00
t.Fatal(err)
}
2017-06-01 13:11:52 +02:00
}