central-api/golang/central-api_test.go

package central_api

import (
	"io/ioutil"
	"testing"

	"golang.org/x/net/context"
	"google.golang.org/grpc"
)

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)

}