mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-03 15:00:27 +08:00
37 lines
810 B
Go
37 lines
810 B
Go
package models_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
_ "github.com/iwind/TeaGo/bootstrap"
|
|
"github.com/iwind/TeaGo/dbs"
|
|
)
|
|
|
|
func TestUserDAO_UpdateUserFeatures(t *testing.T) {
|
|
var dao = models.NewUserDAO()
|
|
var tx *dbs.Tx
|
|
err := dao.UpdateUsersFeatures(tx, []string{
|
|
userconfigs.UserFeatureCodeServerACME,
|
|
}, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestUserDAO_CheckUserServersEnabled(t *testing.T) {
|
|
dbs.NotifyReady()
|
|
|
|
var dao = models.NewUserDAO()
|
|
var tx *dbs.Tx
|
|
for _, userId := range []int64{1, 2, 1000000} {
|
|
isEnabled, err := dao.CheckUserServersEnabled(tx, userId)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log("user:", userId, "isEnabled:", isEnabled)
|
|
}
|
|
}
|