mirror of
https://github.com/nezhahq/nezha.git
synced 2025-01-23 13:18:13 -05:00
53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type testSt struct {
|
|
input string
|
|
output string
|
|
}
|
|
|
|
func TestNotification(t *testing.T) {
|
|
cases := []testSt{
|
|
{
|
|
input: "103.80.236.249/d5ce:d811:cdb8:067a:a873:2076:9521:9d2d",
|
|
output: "103.****.249/d5ce:d811:****:9521:9d2d",
|
|
},
|
|
{
|
|
input: "3.80.236.29/d5ce::cdb8:067a:a873:2076:9521:9d2d",
|
|
output: "3.****.29/d5ce::****:9521:9d2d",
|
|
},
|
|
{
|
|
input: "3.80.236.29/d5ce::cdb8:067a:a873:2076::9d2d",
|
|
output: "3.****.29/d5ce::****::9d2d",
|
|
},
|
|
{
|
|
input: "3.80.236.9/d5ce::cdb8:067a:a873:2076::9d2d",
|
|
output: "3.****.9/d5ce::****::9d2d",
|
|
},
|
|
{
|
|
input: "3.80.236.9/d5ce::cdb8:067a:a873:2076::9d2d",
|
|
output: "3.****.9/d5ce::****::9d2d",
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
assert.Equal(t, IPDesensitize(c.input), c.output)
|
|
}
|
|
}
|
|
|
|
func TestGenerGenerateRandomString(t *testing.T) {
|
|
generatedString := make(map[string]bool)
|
|
for i := 0; i < 100; i++ {
|
|
str, err := GenerateRandomString(32)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, len(str), 32)
|
|
assert.False(t, generatedString[str])
|
|
generatedString[str] = true
|
|
}
|
|
}
|