7x31/anylink
1
0
Fork 0
mirror of https://github.com/bjdgyc/anylink.git synced 2025-09-26 14:23:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复测试文件

Browse Source
This commit is contained in:
bjdgyc
2025-02-26 17:50:38 +08:00
parent d33eeed180
commit d5e3e99b48
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
server/dbdata/user_test.go
View File

@@ -3,10 +3,12 @@ package dbdata
import ( import (
"testing" "testing"
"github.com/bjdgyc/anylink/base"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
) )
func TestCheckUser(t *testing.T) { func TestCheckUser(t *testing.T) {
base.Test()
ast := assert.New(t) ast := assert.New(t)
preIpData() preIpData()
@@ -38,7 +40,10 @@ func TestCheckUser(t *testing.T) {
// 单独验证密码 // 单独验证密码
u.DisableOtp = true u.DisableOtp = true
_ = SetUser(&u) _ = SetUser(&u)
err = CheckUser("aaa", pincode, group) ext := map[string]any{
"mac_addr": "",
}
err = CheckUser("aaa", pincode, group, ext)
ast.Nil(err) ast.Nil(err)
// 添加一个radius组 // 添加一个radius组
@@ -53,9 +58,9 @@ func TestCheckUser(t *testing.T) {
g2 := Group{Name: group2, Status: 1, ClientDns: dns, RouteInclude: route, Auth: authData} g2 := Group{Name: group2, Status: 1, ClientDns: dns, RouteInclude: route, Auth: authData}
err = SetGroup(&g2) err = SetGroup(&g2)
ast.Nil(err) ast.Nil(err)
err = CheckUser("aaa", "bbbbbbb", group2) err = CheckUser("aaa", "bbbbbbb", group2, ext)
if ast.NotNil(err) { if ast.NotNil(err) {
ast.Equal("aaa Radius服务器连接异常, 请检测服务器和端口", err.Error()) ast.Contains(err.Error(), "aaa Radius服务器连接异常")
} }
// 添加用户策略 // 添加用户策略
dns2 := []ValData{{Val: "8.8.8.8"}} dns2 := []ValData{{Val: "8.8.8.8"}}
@@ -63,7 +68,7 @@ func TestCheckUser(t *testing.T) {
p1 := Policy{Username: "aaa", Status: 1, ClientDns: dns2, RouteInclude: route2} p1 := Policy{Username: "aaa", Status: 1, ClientDns: dns2, RouteInclude: route2}
err = SetPolicy(&p1) err = SetPolicy(&p1)
ast.Nil(err) ast.Nil(err)
err = CheckUser("aaa", pincode, group) err = CheckUser("aaa", pincode, group, ext)
ast.Nil(err) ast.Nil(err)
// 添加一个ldap组 // 添加一个ldap组
group3 := "group3" group3 := "group3"
@@ -83,7 +88,7 @@ func TestCheckUser(t *testing.T) {
g3 := Group{Name: group3, Status: 1, ClientDns: dns, RouteInclude: route, Auth: authData} g3 := Group{Name: group3, Status: 1, ClientDns: dns, RouteInclude: route, Auth: authData}
err = SetGroup(&g3) err = SetGroup(&g3)
ast.Nil(err) ast.Nil(err)
err = CheckUser("aaa", "bbbbbbb", group3) err = CheckUser("aaa", "bbbbbbb", group3, ext)
if ast.NotNil(err) { if ast.NotNil(err) {
ast.Equal("aaa LDAP服务器连接异常, 请检测服务器和端口", err.Error()) ast.Equal("aaa LDAP服务器连接异常, 请检测服务器和端口", err.Error())
} }