package hpas

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"
	"reflect"
	"runtime"
	"testing"

	"github.com/baidubce/bce-sdk-go/services/hpas/api"
	"github.com/baidubce/bce-sdk-go/util/log"
)

var (
	HPAS_CLIENT *Client
	Hpas_id     string
)

// For security reason, ak/sk should not hard write here.
type Conf struct {
	AK       string
	SK       string
	Endpoint string
}

func init() {
	_, f, _, _ := runtime.Caller(0)
	conf := filepath.Join(filepath.Dir(f), "config.json")
	fp, err := os.Open(conf)
	log.Error("config json file of ak/sk:", conf)
	if err != nil {
		log.Fatal("config json file of ak/sk not given:", conf)
		os.Exit(1)
	}
	decoder := json.NewDecoder(fp)
	confObj := &Conf{}
	decoder.Decode(confObj)
	fmt.Println(confObj)

	HPAS_CLIENT, _ = NewClient(confObj.AK, confObj.SK, confObj.Endpoint)
	log.SetLogLevel(log.WARN)
}

// ExpectEqual is the helper function for test each case
func ExpectEqual(alert func(format string, args ...interface{}),
	expected interface{}, actual interface{}) bool {
	expectedValue, actualValue := reflect.ValueOf(expected), reflect.ValueOf(actual)
	equal := false
	switch {
	case expected == nil && actual == nil:
		return true
	case expected != nil && actual == nil:
		equal = expectedValue.IsNil()
	case expected == nil && actual != nil:
		equal = actualValue.IsNil()
	default:
		if actualType := reflect.TypeOf(actual); actualType != nil {
			if expectedValue.IsValid() && expectedValue.Type().ConvertibleTo(actualType) {
				equal = reflect.DeepEqual(expectedValue.Convert(actualType).Interface(), actual)
			}
		}
	}
	if !equal {
		_, file, line, _ := runtime.Caller(1)
		alert("%s:%d: missmatch, expect %v but %v", file, line, expected, actual)
		return false
	}
	return true
}

func TestCreateInstance(t *testing.T) {
	createInstanceArgs := &api.CreateHpasReq{
		AppType:             "llama2_7B_train_ic5",
		AppPerformanceLevel: "10k",
		Name:                "create_hpas_test",
		ApplicationName:     "app-name",
		AutoSeqSuffix:       true,
		PurchaseNum:         1,
		ZoneName:            "cn-bj-a",
		ImageId:             "m-Xz6svNFM",
		SubnetId:            "sbn-s2haxxwvw8yi",
		SecurityGroupIds:    []string{"g-9vjwstn24c2v"},
		Password:            "7**************************************************************5",
		KeypairId:           "k-dadadad",
		Tags:                []api.TagModel{{TagKey: "test1", TagValue: "test1"}},
		InternalIps:         []string{"192.168.48.12"},
		UserData:            "userData",
		ReservedInstance:    &api.CreateCombinedCouponReq{
			Name: "test_reserved",
			PurchaseNum:     1,
			AutoRenewPeriod: 3,
			ReservedInstancePeriod: 6,
		},
	}
	createResult, err := HPAS_CLIENT.CreateHpas(createInstanceArgs)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(createResult)
	Hpas_id = createResult.HpasIds[0]
}

func TestDeleteHpas(t *testing.T) {
	deleteInstanceArgs := &api.DeleteHpasReq{
		HpasIds: []string{"hpas-yQtvbIDe"},
	}
	err := HPAS_CLIENT.DeleteHpas(deleteInstanceArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestStopHpas(t *testing.T) {
	stopHpasArgs := &api.StopHpasReq{
		HpasIds:   []string{"hpas-0eGs9gbM"},
		ForceStop: true,
	}
	err := HPAS_CLIENT.StopHpas(stopHpasArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestStartHpas(t *testing.T) {
	startInstanceArgs := &api.StartHpasReq{
		HpasIds: []string{"hpas-0eGs9gbM"},
	}
	err := HPAS_CLIENT.StartHpas(startInstanceArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestRebootHpas(t *testing.T) {
	rebootInstanceArgs := &api.RebootHpasReq{
		HpasIds:   []string{"hpas-0eGs9gbM"},
		ForceStop: true,
	}
	err := HPAS_CLIENT.RebootHpas(rebootInstanceArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestResetHpas(t *testing.T) {
	resetHpasArgs := &api.ResetHpasReq{
		HpasIds:   []string{"hpas-YuZJjrZ1"},
		ImageId:   "m-Xz6svNFM",
		Password:  "71fa62c0059fa8624a4fbe110e236ab31ceede74cc7349df2f75f7ed2a279665",
		KeypairId: "k-dadadad",
		UserData:  "userData",
		CleanLastUserData: false,
	}
	err := HPAS_CLIENT.ResetHpas(resetHpasArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestModifyPasswordHpas(t *testing.T) {
	modifyPasswordHpasArgs := &api.ModifyPasswordHpasReq{
		HpasId:   "hpas-0eGs9gbM",
		Password: "71fa62c0059fa8624a4fbe110e236ab31ceede74cc7349df2f75f7ed2a279665",
	}
	err := HPAS_CLIENT.ModifyPasswordHpas(modifyPasswordHpasArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestModifyInstancesAttribute(t *testing.T) {
	modifyInstancesAttributeArgs := &api.ModifyInstancesAttributeReq{
		HpasIds: []string{"hpas-FRUqoSQk"},
		Name:    "newName0324",
	}
	err := HPAS_CLIENT.ModifyInstancesAttribute(modifyInstancesAttributeArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestClient_ModifyInstancesSubnet(t *testing.T) {
	modifyInstancesSubnetArgs := &api.ModifyInstancesSubnetRequest{
		HpasIds:  []string{"hpas-FRUqoSQk"},
		SubnetId: "sbn-s2haxxwvw8yi",
	}
	resp, err := HPAS_CLIENT.ModifyInstancesSubnet(modifyInstancesSubnetArgs)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestClient_ModifyInstanceVpc(t *testing.T) {
	modifyInstanceVpcArgs := &api.ModifyInstanceVpcRequest{
		HpasId:            "hpasId",
		SubnetId:          "subnetId",
		PrivateIp:         "privateIp",
		SecurityGroupType: "securityGroupType",
		SecurityGroupIds:  nil,
	}
	resp, err := HPAS_CLIENT.ModifyInstanceVpc(modifyInstanceVpcArgs)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestCreateReservedHpas(t *testing.T) {
	createReservedHpasReq := &api.CreateReservedHpasReq{
		AppType:             "llama2_7B_train",
		AppPerformanceLevel: "10k",
		Name:                "create_hpas_test",
		PurchaseNum:         1,
		ZoneName:            "cn-bj-a",
		Tags:                []api.TagModel{{TagKey: "test1", TagValue: "test1"}},
	}
	createResult, err := HPAS_CLIENT.CreateReservedHpas(createReservedHpasReq)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(createResult)
	Hpas_id = createResult.ReservedHpasIds[0]
}

func TestDescribeReservedHpas(t *testing.T) {
	listReservedHpasPageReq := &api.ListReservedHpasPageReq{
		ReservedHpasIds: []string{"k-eZgiHDOY"},
		PageNo:          1,
		PageSize:        10,
	}
	describeResult, err := HPAS_CLIENT.DescribeReservedHpas(listReservedHpasPageReq)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(describeResult)
}

func TestListHpas(t *testing.T) {
	listHpasPageReq := &api.ListHpasPageReq{
		HpasIds:    []string{"hpas-hS7So6Qy"},
		Name:       "real",
		ZoneName:   "cn-bj-a",
		HpasStatus: "Active",
		AppType:    "llama2_7B_train",
		PageNo:     1,
		PageSize:   10,
	}
	listResult, err := HPAS_CLIENT.ListHpas(listHpasPageReq)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(listResult)
}

func TestListHpasWithRdmaTopo(t *testing.T) {
	listHpasPageReq := &api.ListHpasPageReq{
		HpasIds:      []string{"hpas-Gw8dTmsL"},
		Name:         "real",
		ZoneName:     "cn-bj-a",
		HpasStatus:   "Active",
		ShowRdmaTopo: true,
		PageNo:       1,
		PageSize:     10,
	}
	listResult, err := HPAS_CLIENT.ListHpas(listHpasPageReq)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(listResult)
}

func TestImageList(t *testing.T) {
	body := &api.DescribeHpasImageReq{
		Marker:  "m-11111",
		MaxKeys: 2,
	}
	resp, err := HPAS_CLIENT.ImageList(body)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestCreateImage(t *testing.T) {
	body := &api.CreateImageReq{
		HpasId:    "hpas-1111",
		ImageName: "name",
	}
	resp, err := HPAS_CLIENT.CreateImage(body)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestModifyImageAttribute(t *testing.T) {
	body := &api.ModifyImageAttributeReq{
		ImageId:   "m-rJhg2N26",
		ImageName: "test_custom_image",
	}
	resp, err := HPAS_CLIENT.ModifyImageAttribute(body)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestDeleteImages(t *testing.T) {
	body := &api.DeleteImagesReq{
		ImageIds: []string{"m-11111"},
	}
	resp, err := HPAS_CLIENT.DeleteImages(body)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestDescribeHPASInstancesByMaker(t *testing.T) {
	listHpasByMakerReq := &api.ListHpasByMakerReq{
		HpasIds:      []string{"hpas-hS7So6Qy"},
		Name:         "real",
		ZoneName:     "zoneA",
		HpasStatus:   "active",
		AppType:      "llama2_7B_train",
		ShowRdmaTopo: true,
		Marker:       "marker123",
		MaxKeys:      10,
		VpcId: "vpc-c2fdxvjuvhiu",
		PrivateIps: []string{"10.172.192.106", "172.16.0.11"},
	}

	resp, err := HPAS_CLIENT.DescribeHPASInstancesByMaker(listHpasByMakerReq)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestListReservedHpasByMakerReq(t *testing.T) {
	req := &api.ListReservedHpasByMakerReq{
		ReservedHpasIds:    []string{"k-fAqX67kN"},
		Name:               "test",
		ZoneName:           "zoneG",
		ReservedHpasStatus: "active",
		AppType:            "llama2_7B_train",
		Marker:             "marker123",
		HpasId:             "hpas-hS7So6Qy",
		MaxKeys:            10,
	}

	resp, err := HPAS_CLIENT.DescribeReservedHpasByMaker(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestClient_ModifyReservedHpasName(t *testing.T) {
	modifyReservedHpasNameArgs := &api.ModifyReservedHpasNameReq{
		ReservedInstanceIds: []string{"k-xx13SR5s"},
		Name:                "name",
	}
	resp, err := HPAS_CLIENT.ModifyReservedHpasName(modifyReservedHpasNameArgs)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestClient_ModifyReservedHpas(t *testing.T) {
	modifyReservedHpasArgs := &api.ModifyReservedHpasReq{
		ModifyReservedHpasList: []api.ModifyReservedHpasModel{
			{
				ReservedInstanceId: "k-xx13SR5s",
				ZoneName:           "cn-bj-g",
				EhcClusterId:       "ehc-ARLMSBZs",
			},
		},
	}
	resp, err := HPAS_CLIENT.ModifyReservedHpas(modifyReservedHpasArgs)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestAssignIpv4ReqWithPrivateIps(t *testing.T) {
	req := &api.AssignIpv4Req{
		HpasId:                  "hpas-hS7So6Qy",
		PrivateIps:              []string{"172.16.0.8", "172.16.0.11"},
		SecondaryPrivateIpCount: 2,
	}

	resp, err := HPAS_CLIENT.AssignPrivateIpAddresses(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestAssignIpv4ReqWithPrivateIpCount(t *testing.T) {
	req := &api.AssignIpv4Req{
		HpasId:                  "hpas-hS7So6Qy",
		SecondaryPrivateIpCount: 2,
	}

	resp, err := HPAS_CLIENT.AssignPrivateIpAddresses(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestUnAssignIpv4Req(t *testing.T) {
	req := &api.UnAssignIpv4Req{
		HpasId:     "hpas-hS7So6Qy",
		PrivateIps: []string{"172.16.0.8", "172.16.0.11"}}

	resp, err := HPAS_CLIENT.UnAssignPrivateIpAddresses(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestAssignIpv6ReqWithIpv6Addresses(t *testing.T) {
	req := &api.AssignIpv6Req{
		HpasId:        "hpas-hS7So6Qy",
		Ipv6Addresses: []string{"2400:da00:e003:0:78c::2", "2400:da00:e003:0:78c::3"},
		Reboot:        true,
	}

	resp, err := HPAS_CLIENT.AssignIpv6Addresses(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestAssignIpv6ReqWithIpv6AddressCount(t *testing.T) {
	req := &api.AssignIpv6Req{
		HpasId:           "hpas-hS7So6Qy",
		Ipv6AddressCount: 4,
		Reboot:           false,
	}

	resp, err := HPAS_CLIENT.AssignIpv6Addresses(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestUnAssignIpv6Req(t *testing.T) {
	req := &api.UnAssignIpv6Req{
		HpasId:        "hpas-hS7So6Qy",
		Ipv6Addresses: []string{"2400:da00:e003:0:78c::2", "2400:da00:e003:0:78c::3"},
		Reboot:        true,
	}

	resp, err := HPAS_CLIENT.UnAssignIpv6Addresses(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestAttachTags(t *testing.T) {
	attachTagsArgs := &api.TagsOperationRequest{
		ResourceType: "hpasri",
		ResourceIds:  []string{"k-NEbO71Mz"},
		Tags: []api.TagModel{
			{
				TagKey:   "test1",
				TagValue: "test1",
			},
		},
	}
	err := HPAS_CLIENT.AttachTags(attachTagsArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestDetachTags(t *testing.T) {
	attachTagsArgs := &api.TagsOperationRequest{
		ResourceType: "hpasri",
		ResourceIds:  []string{"k-NEbO71Mz"},
		Tags: []api.TagModel{
			{
				TagKey:   "test1",
				TagValue: "test1",
			},
		},
	}
	err := HPAS_CLIENT.DetachTags(attachTagsArgs)
	ExpectEqual(t.Errorf, err, nil)
}

func TestDescribeHpasVncUrl(t *testing.T) {
	req := &api.DescribeHpasVncUrlReq{
		HpasId: "hpas-xxxxxxx",
	}
	resp, err := HPAS_CLIENT.DescribeHpasVncUrl(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestAttachSecurityGroups(t *testing.T) {
	req := &api.SecurityGroupsReq{
		HpasIds:           []string{"hpas-xxxxxxx"},
		SecurityGroupIds:  []string{"sg-xxxxxxx"},
		SecurityGroupType: "normal",
	}
	resp, err := HPAS_CLIENT.AttachSecurityGroups(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestCreatePreInstance(t *testing.T) {
	createInstanceArgs := &api.CreateHpasReq{
		AppType:             "unixbench_indexscore",
		AppPerformanceLevel: "1.6k",
		Name:                "create_hpas_test",
		ApplicationName:     "app-name",
		AutoSeqSuffix:       true,
		PurchaseNum:         1,
		ZoneName:            "cn-bj-a",
		ImageId:             "m-JtPe0qoJ",
		SubnetId:            "sbn-mmbsnu0kccsm",
		RootDiskSizeInGb:    400,
		RootDiskStorageType: "enhanced_ssd_pl1",
		SecurityGroupIds:    []string{"g-ktj27g1f7wuy"},
		KeypairId:           "k-GY5hmntP",
		BillingModel: api.BillingModel{
			ChargeType: "Prepaid",
			Period:     1,
			PeriodUnit: "Month",
		},
	}
	createResult, err := HPAS_CLIENT.CreateHpas(createInstanceArgs)
	ExpectEqual(t.Errorf, err, nil)
	Hpas_id = createResult.HpasIds[0]
	fmt.Println(Hpas_id)
}

func TestRenewHpasInstances(t *testing.T) {
	req := &api.RenewHpasReq{
		HpasIds:    []string{"hpas-mDyozAkj"},
		Period:     1,
		PeriodUnit: "Month",
	}
	resp, err := HPAS_CLIENT.RenewHpasInstances(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}


func TestReplaceSecurityGroups(t *testing.T) {
	req := &api.SecurityGroupsReq{
		HpasIds:           []string{"hpas-xxxxxxx"},
		SecurityGroupIds:  []string{"sg-xxxxxxx"},
		SecurityGroupType: "normal",
	}
	resp, err := HPAS_CLIENT.ReplaceSecurityGroups(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestDetachSecurityGroups(t *testing.T) {
	req := &api.SecurityGroupsReq{
		HpasIds:           []string{"hpas-xxxxxxx"},
		SecurityGroupIds:  []string{"sg-xxxxxxx"},
		SecurityGroupType: "normal",
	}
	resp, err := HPAS_CLIENT.DetachSecurityGroups(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestDescribeHpasSpecs(t *testing.T) {
	req := &api.DescribeHpasSpecsReq{
		ZoneName: "cn-bj-a",
		AppType:  "llama2_7B_train",
		AppPerformanceLevel: "10k",
	}
	resp, err := HPAS_CLIENT.DescribeHpasSpecs(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}

func TestDescribeInstanceInventoryQuantity(t *testing.T) {
	req := &api.DescribeInstanceInventoryQuantityReq{
		ZoneName:            "cn-bj-a",
		AppType:             "llama2_7B_train",
		AppPerformanceLevel: "10k",
	}
	resp, err := HPAS_CLIENT.DescribeInstanceInventoryQuantity(req)
	ExpectEqual(t.Errorf, err, nil)
	fmt.Println(resp)
}