jyps/graph/graph_test.go

package main

import (
	"log"
	"testing"
)

func TestCheckInvestRelation(t *testing.T) {
	session, pool, err := ConnectToNebula(HostList, UserName, PassWord)
	if err != nil {
		log.Fatalf("Failed to connect to Nebula Graph: %v", err)
	}
	defer pool.Close()
	defer session.Release()
	names := []string{"华芳创业投资有限公司", "北京剑鱼信息技术有限公司", "河南折扣牛哟有限公司", "上海元藩投资有限公司"}
	//res, err := CheckLegalRelationsGraph(session, names, 3)
	has, res, err := CheckLegalRelationships4(session, names, 3, 1)
	if err != nil {
		log.Println(res, err, has)
	}
	log.Println(has, res)
}

func TestFetchLegalByVid(t *testing.T) {
	session, pool, err := ConnectToNebula(HostList, UserName, PassWord)
	if err != nil {
		log.Fatalf("Failed to connect to Nebula Graph: %v", err)
	}
	defer pool.Close()
	defer session.Release()

	vid := "fb7c72258ec3f57bedd0657b0c3e90d2"

	res, err := getLegalByVid(session, vid)
	log.Println(res, err)
}