优化用户服务整体启用和禁用

internal/db/models/user_dao_test.go

@@ -1,6 +1,7 @@
-package models
+package models_test
 
 import (
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
 	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
 	_ "github.com/go-sql-driver/mysql"
 	_ "github.com/iwind/TeaGo/bootstrap"
@@ -9,7 +10,7 @@ import (
 )
 
 func TestUserDAO_UpdateUserFeatures(t *testing.T) {
-	var dao = NewUserDAO()
+	var dao = models.NewUserDAO()
 	var tx *dbs.Tx
 	err := dao.UpdateUsersFeatures(tx, []string{
 		userconfigs.UserFeatureCodeServerACME,
@@ -18,3 +19,17 @@ func TestUserDAO_UpdateUserFeatures(t *testing.T) {
 		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)
+	}
+}