central-api/golang/server_test.go

61 lines
1.0 KiB
Go
Raw Normal View History

2018-04-06 10:16:19 +00:00
package server_grpc
import (
"testing"
"google.golang.org/grpc"
"golang.org/x/net/context"
"io/ioutil"
)
func TestTarDis_saveAllTarget(t *testing.T) {
contents,_ := ioutil.ReadFile("../dh.json")
si := &ServerInput{}
si.Target = "TargetDiscoveryService"
si.Method = "saveAllTarget"
si.Params = append(si.Params, string(contents))
si.Params = append(si.Params, "{\"id\":1}")
callRPC(t, si)
}
func TestRPCNP_readAllByDomain(t *testing.T) {
si := &ServerInput{}
si.Target = "NoAuthProbeService"
si.Method = "readAllByDomain"
//sp := &ServerParam{}
//sp.IsCollection = false
//sp.Type = "com.loafle.overflow.module.domain.model.Domain"
//sp.Data = "{\"id\":1}"
si.Params = append(si.Params, "{\"id\":1}")
callRPC(t, si)
}
func callRPC(t *testing.T, si *ServerInput) {
conn, err := grpc.Dial(":50006", grpc.WithInsecure())
if err != nil {
t.Log(err)
}
client := NewOverflowApiServerClient(conn)
out, err := client.Exec(context.Background(), si)
if err != nil {
t.Fatal(err)
}
t.Log(out)
}