package tests

import (
	"testing"

	"github.com/gin-gonic/gin"
	"github.com/pingcap/dbaas/billing2/api"
	"github.com/pingcap/dbaas/billing2/utils/ginutils"
	"github.com/pingcap/dbaas/billing2/utils/tests"
	"github.com/stretchr/testify/suite"
)

type MigrateTestSuite struct {
	tests.BillingTestSuite
	router *gin.Engine
}

func (suite *MigrateTestSuite) SetupTest() {
	suite.BillingTestSuite.SetupTest()
	suite.router = gin.New()
	tests.MockCentralHandler()
	suite.router.Use(ginutils.ErrorMiddleWare)
	api.InitializeMigrateRouter(ginutils.MyRoute{Route: suite.router})
}

func TestMigratePayment(t *testing.T) {
	suite.Run(t, new(MigrateTestSuite))
}