package testsuite

import (
	"time"

	"git.hexq.cn/tiglog/mydb"
	"github.com/stretchr/testify/suite"
)

const TimeZone = "Asia/Shanghai"

var defaultTimeLocation, _ = time.LoadLocation(TimeZone)

type Helper interface {
	Session() mydb.Session

	Adapter() string

	TearUp() error
	TearDown() error
}

type Suite struct {
	suite.Suite

	Helper
}

func (s *Suite) AfterTest(suiteName, testName string) {
	err := s.TearDown()
	s.NoError(err)
}

func (s *Suite) BeforeTest(suiteName, testName string) {
	err := s.TearUp()
	s.NoError(err)
}