XiaoMo/frp
1
0
Fork 0
mirror of https://github.com/fatedier/frp.git synced 2025-06-01 13:08:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

update test package (#4814)

Browse Source
This commit is contained in:
fatedier 2025-05-26 14:54:03 +08:00 committed by GitHub
parent ce366ee17f
commit 1ced733340
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 36 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/stale.yml vendored
View File

@ -1,4 +1,4 @@
name: "Close stale issues"
name: "Close stale issues and PRs"
on:
schedule:
- cron: "20 0 * * *"

8
pkg/proto/udp/udp_test.go
View File

@ -3,16 +3,16 @@ package udp
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestUdpPacket(t *testing.T) {
assert := assert.New(t)
require := require.New(t)
buf := []byte("hello world")
udpMsg := NewUDPPacket(buf, nil, nil)
newBuf, err := GetContent(udpMsg)
assert.NoError(err)
assert.EqualValues(buf, newBuf)
require.NoError(err)
require.EqualValues(buf, newBuf)
}

12
pkg/util/metric/counter_test.go
View File

@ -3,21 +3,21 @@ package metric
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestCounter(t *testing.T) {
assert := assert.New(t)
require := require.New(t)
c := NewCounter()
c.Inc(10)
assert.EqualValues(10, c.Count())
require.EqualValues(10, c.Count())
c.Dec(5)
assert.EqualValues(5, c.Count())
require.EqualValues(5, c.Count())
cTmp := c.Snapshot()
assert.EqualValues(5, cTmp.Count())
require.EqualValues(5, cTmp.Count())
c.Clear()
assert.EqualValues(0, c.Count())
require.EqualValues(0, c.Count())
}

18
pkg/util/metric/date_counter_test.go
View File

@ -3,25 +3,25 @@ package metric
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestDateCounter(t *testing.T) {
assert := assert.New(t)
require := require.New(t)
dc := NewDateCounter(3)
dc.Inc(10)
assert.EqualValues(10, dc.TodayCount())
require.EqualValues(10, dc.TodayCount())
dc.Dec(5)
assert.EqualValues(5, dc.TodayCount())
require.EqualValues(5, dc.TodayCount())
counts := dc.GetLastDaysCount(3)
assert.EqualValues(3, len(counts))
assert.EqualValues(5, counts[0])
assert.EqualValues(0, counts[1])
assert.EqualValues(0, counts[2])
require.EqualValues(3, len(counts))
require.EqualValues(5, counts[0])
require.EqualValues(0, counts[1])
require.EqualValues(0, counts[2])
dcTmp := dc.Snapshot()
assert.EqualValues(5, dcTmp.TodayCount())
require.EqualValues(5, dcTmp.TodayCount())
}

36
pkg/util/util/util_test.go
View File

@ -3,45 +3,41 @@ package util
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestRandId(t *testing.T) {
assert := assert.New(t)
require := require.New(t)
id, err := RandID()
assert.NoError(err)
require.NoError(err)
t.Log(id)
assert.Equal(16, len(id))
require.Equal(16, len(id))
}
func TestGetAuthKey(t *testing.T) {
assert := assert.New(t)
require := require.New(t)
key := GetAuthKey("1234", 1488720000)
assert.Equal("6df41a43725f0c770fd56379e12acf8c", key)
require.Equal("6df41a43725f0c770fd56379e12acf8c", key)
}
func TestParseRangeNumbers(t *testing.T) {
assert := assert.New(t)
require := require.New(t)
numbers, err := ParseRangeNumbers("2-5")
if assert.NoError(err) {
assert.Equal([]int64{2, 3, 4, 5}, numbers)
}
require.NoError(err)
require.Equal([]int64{2, 3, 4, 5}, numbers)
numbers, err = ParseRangeNumbers("1")
if assert.NoError(err) {
assert.Equal([]int64{1}, numbers)
}
require.NoError(err)
require.Equal([]int64{1}, numbers)
numbers, err = ParseRangeNumbers("3-5,8")
if assert.NoError(err) {
assert.Equal([]int64{3, 4, 5, 8}, numbers)
}
require.NoError(err)
require.Equal([]int64{3, 4, 5, 8}, numbers)
numbers, err = ParseRangeNumbers(" 3-5,8, 10-12 ")
if assert.NoError(err) {
assert.Equal([]int64{3, 4, 5, 8, 10, 11, 12}, numbers)
}
require.NoError(err)
require.Equal([]int64{3, 4, 5, 8, 10, 11, 12}, numbers)
_, err = ParseRangeNumbers("3-a")
assert.Error(err)
require.Error(err)
}