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