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)
	res, err := CheckLegalRelations(session, names, 3)
	if err != nil {
		log.Println(res, err)
	}
	log.Println(res)
}