package test

import (
	"fmt"
	"git.rtzhtech.cn/iss/public-lib/dto"
	"git.rtzhtech.cn/iss/public-lib/logic"
	"git.rtzhtech.cn/iss/public-lib/model"
	"testing"
)

func TestAddLinkStrategy(t *testing.T) {
	l := logic.NewLinkStrategyLogic()
	in := &dto.AddLinkStrategyRes{}
	in.Appid = 123
	in.Strategyname = "Strategyname"
	in.Linktype = 1
	in.Src = append(in.Src, &model.LinkRelationObj{Linkfromid: 1})
	in.Src = append(in.Src, &model.LinkRelationObj{Linkfromid: 2})
	in.Dst = append(in.Dst, &model.LinkRelationAction{Linktoid: 1, Actindex: 1, Actid: 1, Delay: 1})
	in.Dst = append(in.Dst, &model.LinkRelationAction{Linktoid: 1, Actindex: 2, Actid: 2, Delay: 2})
	in.Dst = append(in.Dst, &model.LinkRelationAction{Linktoid: 2, Actindex: 3, Actid: 1, Delay: 1})
	in.Dst = append(in.Dst, &model.LinkRelationAction{Linktoid: 2, Actindex: 4, Actid: 2, Delay: 2})
	for i := 0; i < 10; i++ {
		in.ID = 0
		in.Strategyid = 1000000 + int64(i)
		err := l.AddLinkStrategy(in)
		if err != nil {
			t.Errorf("TestAdd:%v\n", err)
		}
	}
}

func TestDelLinkStrategy(t *testing.T) {
	l := logic.NewLinkStrategyLogic()
	err := l.DelLinkStrategy(1000001)
	if err != nil {
		t.Errorf("TestAdd:%v\n", err)
	}
}

func TestUpdateLinkStrategy(t *testing.T) {
	l := logic.NewLinkStrategyLogic()
	in := &dto.AddLinkStrategyRes{}
	in.Strategyid = 1000000
	in.Appid = 123
	in.Strategyname = "Strategyname3"
	in.Linktype = 2
	in.Src = append(in.Src, &model.LinkRelationObj{Linkfromid: 1})
	in.Src = append(in.Src, &model.LinkRelationObj{Linkfromid: 2})
	in.Dst = append(in.Dst, &model.LinkRelationAction{Linktoid: 1, Actindex: 1, Actid: 1, Delay: 1})
	in.Dst = append(in.Dst, &model.LinkRelationAction{Linktoid: 1, Actindex: 2, Actid: 2, Delay: 2})
	in.Dst = append(in.Dst, &model.LinkRelationAction{Linktoid: 2, Actindex: 3, Actid: 1, Delay: 1})
	in.Dst = append(in.Dst, &model.LinkRelationAction{Linktoid: 2, Actindex: 4, Actid: 2, Delay: 2})

	do := l.SvcCtx.LinkStrategy
	err := l.ModifyLinkStrategy(in, do.Appid.String(), do.Strategyname.String(), do.Linktype.String())
	if err != nil {
		t.Errorf("TestAdd:%v\n", err)
	}

	in.Strategyid = 1000002
	//更新全部字段
	err = l.ModifyLinkStrategy(in)
	if err != nil {
		t.Errorf("TestAdd:%v\n", err)
	}
}

func LinkStrategyWhere() *dto.GetLinkStrategyReq {
	req := &dto.GetLinkStrategyReq{}
	req.Page = 2
	req.Limit = 3
	req.AppId = 123
	req.StrategyName = "name"
	return req
}

func TestGetLinkStrategy(t *testing.T) {
	l := logic.NewLinkStrategyLogic()
	out, err := l.GetLinkStrategy(LinkStrategyWhere())
	if err != nil {
		t.Errorf("TestPage:%v\n", err)
	}
	fmt.Printf("total:%d\n", out.Total)
	for _, v := range out.List {
		fmt.Println(v)
	}
}

func TestGetLinkStrategyCount(t *testing.T) {
	l := logic.NewLinkStrategyLogic()
	count, err := l.GetLinkStrategyCount(LinkStrategyWhere())
	if err != nil {
		t.Errorf("TestPage:%v\n", err)
	}
	fmt.Println(count)
}