80 lines
1.5 KiB
Go
Executable File
80 lines
1.5 KiB
Go
Executable File
package dao
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"testing"
|
|
|
|
// 这里的依赖需要被导入，否则会报错
|
|
"atom/container"
|
|
"atom/database/models"
|
|
"atom/database/query"
|
|
_ "atom/providers"
|
|
"atom/utils"
|
|
|
|
"github.com/brianvoe/gofakeit/v6"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
"go.uber.org/dig"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UserRoleInjectParams struct {
|
|
dig.In
|
|
|
|
DB *gorm.DB
|
|
Dao *UserRoleDao
|
|
Query *query.Query
|
|
Faker *gofakeit.Faker
|
|
}
|
|
|
|
type UserRoleSuite struct {
|
|
suite.Suite
|
|
UserRoleInjectParams
|
|
}
|
|
|
|
func init() {
|
|
if err := container.Container.Provide(NewUserRoleDao); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func Test_UserRoleSuite(t *testing.T) {
|
|
err := container.Container.Invoke(func(p UserRoleInjectParams) {
|
|
s := &UserRoleSuite{}
|
|
s.UserRoleInjectParams = p
|
|
|
|
suite.Run(t, s)
|
|
})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func (s *UserRoleSuite) BeforeTest(suiteName, testName string) {
|
|
utils.TruncateTable(s.DB, s.Query.UserRole.TableName())
|
|
|
|
}
|
|
|
|
func (s *UserRoleSuite) Test_GetByUserID() {
|
|
Convey("Test_GetByUserID", s.T(), func() {
|
|
Reset(func() {
|
|
s.BeforeTest("_", "Test_GetByUserID")
|
|
})
|
|
|
|
Convey("not exists", func() {
|
|
has := s.Dao.Exists(context.Background(), 1)
|
|
So(has, ShouldBeFalse)
|
|
})
|
|
|
|
Convey("exists", func() {
|
|
_ = s.Query.UserRole.WithContext(context.Background()).Create(&models.UserRole{
|
|
UserID: 1,
|
|
RoleID: 1,
|
|
})
|
|
|
|
has := s.Dao.Exists(context.Background(), 1)
|
|
So(has, ShouldBeTrue)
|
|
})
|
|
})
|
|
}
|