package test

import (
	"testing"

	"github.com/dtm-labs/dtm/client/dtmcli"
	"github.com/dtm-labs/dtm/client/dtmcli/dtmimp"
	"github.com/dtm-labs/dtm/test/busi"
	"github.com/go-resty/resty/v2"
	"github.com/stretchr/testify/assert"
)

func TestXaCoverDBError(t *testing.T) {
	oldDriver := busi.BusiConf.Driver
	gid := dtmimp.GetFuncName()
	err := dtmcli.XaGlobalTransaction(DtmServer, gid, func(xa *dtmcli.Xa) (*resty.Response, error) {
		req := busi.GenReqHTTP(30, false, false)
		_, err := xa.CallBranch(req, busi.Busi+"/TransOutXa")
		assert.Nil(t, err)
		busi.BusiConf.Driver = "no-driver"
		_, err = xa.CallBranch(req, busi.Busi+"/TransInXa")
		assert.Error(t, err)
		return nil, err
	})
	assert.Error(t, err)
	waitTransProcessed(gid)
	busi.BusiConf.Driver = oldDriver
	cronTransOnceForwardNow(t, gid, 500) // rollback succeeded here
	assert.Equal(t, StatusFailed, getTransStatus(gid))
	assert.Equal(t, []string{StatusSucceed, StatusPrepared}, getBranchesStatus(gid))
}

func TestXaCoverDTMError(t *testing.T) {
	gid := dtmimp.GetFuncName()
	err := dtmcli.XaGlobalTransaction("localhost:01", gid, func(xa *dtmcli.Xa) (*resty.Response, error) {
		return nil, nil
	})
	assert.Error(t, err)
}

func TestXaCoverGidError(t *testing.T) {
	if dtmimp.GetCurrentDBType() != dtmimp.DBTypeMysql {
		return
	}
	gid := dtmimp.GetFuncName() + "-'  '"
	err := dtmcli.XaGlobalTransaction(DtmServer, gid, func(xa *dtmcli.Xa) (*resty.Response, error) {
		req := busi.GenReqHTTP(30, false, false)
		_, err := xa.CallBranch(req, busi.Busi+"/TransOutXa")
		assert.Error(t, err)
		return nil, err
	})
	assert.Error(t, err)
	waitTransProcessed(gid)
}