51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package dbdata
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/bjdgyc/anylink/pkg/utils"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetGroupNames(t *testing.T) {
|
|
ast := assert.New(t)
|
|
|
|
preIpData()
|
|
defer closeIpdata()
|
|
|
|
// 添加 group
|
|
g1 := Group{Name: "g1", ClientDns: []ValData{{Val: "114.114.114.114"}}}
|
|
err := SetGroup(&g1)
|
|
ast.Nil(err)
|
|
g2 := Group{Name: "g2", ClientDns: []ValData{{Val: "114.114.114.114"}}}
|
|
err = SetGroup(&g2)
|
|
ast.Nil(err)
|
|
g3 := Group{Name: "g3", ClientDns: []ValData{{Val: "114.114.114.114"}}}
|
|
err = SetGroup(&g3)
|
|
ast.Nil(err)
|
|
|
|
authData := map[string]interface{}{
|
|
"type": "radius",
|
|
"radius": map[string]string{
|
|
"addr": "192.168.8.12:1044",
|
|
"secret": "43214132",
|
|
},
|
|
}
|
|
g4 := Group{Name: "g4", ClientDns: []ValData{{Val: "114.114.114.114"}}, Auth: authData}
|
|
err = SetGroup(&g4)
|
|
ast.Nil(err)
|
|
g5 := Group{Name: "g5", ClientDns: []ValData{{Val: "114.114.114.114"}}, DsIncludeDomains: "baidu.com,163.com"}
|
|
err = SetGroup(&g5)
|
|
ast.Nil(err)
|
|
g6 := Group{Name: "g6", ClientDns: []ValData{{Val: "114.114.114.114"}}, DsExcludeDomains: "com.cn,qq.com"}
|
|
err = SetGroup(&g6)
|
|
ast.Nil(err)
|
|
|
|
// 判断所有数据
|
|
gAll := []string{"g1", "g2", "g3", "g4", "g5", "g6"}
|
|
gs := GetGroupNames()
|
|
for _, v := range gs {
|
|
ast.Equal(true, utils.InArrStr(gAll, v))
|
|
}
|
|
}
|