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

func TestCheckLegalRelationships(t *testing.T) {
	client, err := NewNebulaClient(HostList, UserName, PassWord)
	if err != nil {
		log.Fatal("连接失败:", err)
	}
	defer client.Close()
	names := []string{"华芳创业投资有限公司", "北京剑鱼信息技术有限公司", "河南折扣牛哟有限公司", "上海元藩投资有限公司"}

	has, res, err := client.CheckLegalRelationships(names, 3, 0)
	log.Println(has, res, err)
}

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)
}