34 lines
711 B
Go
34 lines
711 B
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)
|
|
|
|
// 判断所有数据
|
|
gAll := []string{"g1", "g2", "g3"}
|
|
gs := GetGroupNames()
|
|
for _, v := range gs {
|
|
ast.Equal(true, utils.InArrStr(gAll, v))
|
|
}
|
|
}
|