7x31
/
anylink
mirror of https://github.com/bjdgyc/anylink.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

修复测试文件

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