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{"北京剑鱼信息技术有限公司", "河南拓普计算机网络工程有限公司", "上海元藩投资有限公司"}
	names := []string{"万达集团股份有限公司", "万达石化有限公司", "山东万达电缆有限公司", "山东万达化工有限公司", "山东万达热电有限公司", "山东万达进出口有限公司", "山东耐斯特炭黑有限公司", "山东万达宝通轮胎有限公司", "山东明宇化学有限公司", "大连万达集团股份有限公司", "大连万达（上海）金融集团有限公司", "大连万达集团咨询服务有限公司", "北京万达足球俱乐部有限公司", "北京红舸科技文化有限公司", "北京万达文化产业集团有限公司"}
	has, result, err := client.CheckLegalRelationships(names, 4, 1)
	log.Println(has, result, err)
}

func TestCheckLegalRelationshipsByStep(t *testing.T) {
	client, err := NewNebulaClient(HostList, UserName, PassWord)
	if err != nil {
		log.Fatal("连接失败:", err)
	}
	defer client.Close()

	names := []string{"河南剑鱼数字科技有限公司", "河南拓普计算机网络工程有限公司", "上海元藩投资有限公司"}
	//names := []string{"万达集团股份有限公司", "万达石化有限公司", "山东万达电缆有限公司", "山东万达化工有限公司", "山东万达热电有限公司", "山东万达进出口有限公司", "山东耐斯特炭黑有限公司", "山东万达宝通轮胎有限公司", "山东明宇化学有限公司", "大连万达集团股份有限公司", "大连万达（上海）金融集团有限公司", "大连万达集团咨询服务有限公司", "北京万达足球俱乐部有限公司", "北京红舸科技文化有限公司", "北京万达文化产业集团有限公司"}
	has, result, err := client.CheckLegalRelationshipsByStep(names, 8, 1)
	log.Println(has, result, 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)
}

// TestFindSuspectInvestRelationsByNames 测试 疑似关系的判断
func TestFindSuspectInvestRelationsByNames(t *testing.T) {
	client, err := NewNebulaClient(HostList, UserName, PassWord)
	if err != nil {
		log.Fatal("连接失败:", err)
	}
	defer client.Close()
	names := []string{"南雄市宝虹农业开发有限公司", "南雄市龙旭园林景观设计有限公司", "香港豪华国际酒店有限公司", "香港豪华酒店（中国）国际管理有限公司"}

	res, err := client.FindSuspectInvestRelationsByNames(names)

	log.Println(res, err)

}

// TestNebulaClient_FindExecutivesInvestRelationsByNames 测试 董事高管 关系
func TestNebulaClient_FindExecutivesInvestRelationsByNames(t *testing.T) {
	client, err := NewNebulaClient(HostList, UserName, PassWord)
	if err != nil {
		log.Fatal("连接失败:", err)
	}
	defer client.Close()
	names := []string{"南雄市宝虹农业开发有限公司", "南雄市龙旭园林景观设计有限公司", "香港豪华国际酒店有限公司", "香港豪华酒店（中国）国际管理有限公司"}

	res, err := client.FindExecutivesInvestRelationsByNames(names)

	log.Println(res, err)
}