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: on:
schedule: schedule:
- cron: "20 0 * * *" - cron: "20 0 * * *"

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

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

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

@ -3,21 +3,21 @@ package metric
import ( import (
"testing" "testing"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/require"
) )
func TestCounter(t *testing.T) { func TestCounter(t *testing.T) {
assert := assert.New(t) require := require.New(t)
c := NewCounter() c := NewCounter()
c.Inc(10) c.Inc(10)
assert.EqualValues(10, c.Count()) require.EqualValues(10, c.Count())
c.Dec(5) c.Dec(5)
assert.EqualValues(5, c.Count()) require.EqualValues(5, c.Count())
cTmp := c.Snapshot() cTmp := c.Snapshot()
assert.EqualValues(5, cTmp.Count()) require.EqualValues(5, cTmp.Count())
c.Clear() 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 ( import (
"testing" "testing"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/require"
) )
func TestDateCounter(t *testing.T) { func TestDateCounter(t *testing.T) {
assert := assert.New(t) require := require.New(t)
dc := NewDateCounter(3) dc := NewDateCounter(3)
dc.Inc(10) dc.Inc(10)
assert.EqualValues(10, dc.TodayCount()) require.EqualValues(10, dc.TodayCount())
dc.Dec(5) dc.Dec(5)
assert.EqualValues(5, dc.TodayCount()) require.EqualValues(5, dc.TodayCount())
counts := dc.GetLastDaysCount(3) counts := dc.GetLastDaysCount(3)
assert.EqualValues(3, len(counts)) require.EqualValues(3, len(counts))
assert.EqualValues(5, counts[0]) require.EqualValues(5, counts[0])
assert.EqualValues(0, counts[1]) require.EqualValues(0, counts[1])
assert.EqualValues(0, counts[2]) require.EqualValues(0, counts[2])
dcTmp := dc.Snapshot() 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 ( import (
"testing" "testing"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/require"
) )
func TestRandId(t *testing.T) { func TestRandId(t *testing.T) {
assert := assert.New(t) require := require.New(t)
id, err := RandID() id, err := RandID()
assert.NoError(err) require.NoError(err)
t.Log(id) t.Log(id)
assert.Equal(16, len(id)) require.Equal(16, len(id))
} }
func TestGetAuthKey(t *testing.T) { func TestGetAuthKey(t *testing.T) {
assert := assert.New(t) require := require.New(t)
key := GetAuthKey("1234", 1488720000) key := GetAuthKey("1234", 1488720000)
assert.Equal("6df41a43725f0c770fd56379e12acf8c", key) require.Equal("6df41a43725f0c770fd56379e12acf8c", key)
} }
func TestParseRangeNumbers(t *testing.T) { func TestParseRangeNumbers(t *testing.T) {
assert := assert.New(t) require := require.New(t)
numbers, err := ParseRangeNumbers("2-5") numbers, err := ParseRangeNumbers("2-5")
if assert.NoError(err) { require.NoError(err)
assert.Equal([]int64{2, 3, 4, 5}, numbers) require.Equal([]int64{2, 3, 4, 5}, numbers)
}
numbers, err = ParseRangeNumbers("1") numbers, err = ParseRangeNumbers("1")
if assert.NoError(err) { require.NoError(err)
assert.Equal([]int64{1}, numbers) require.Equal([]int64{1}, numbers)
}
numbers, err = ParseRangeNumbers("3-5,8") numbers, err = ParseRangeNumbers("3-5,8")
if assert.NoError(err) { require.NoError(err)
assert.Equal([]int64{3, 4, 5, 8}, numbers) require.Equal([]int64{3, 4, 5, 8}, numbers)
}
numbers, err = ParseRangeNumbers(" 3-5,8, 10-12 ") numbers, err = ParseRangeNumbers(" 3-5,8, 10-12 ")
if assert.NoError(err) { require.NoError(err)
assert.Equal([]int64{3, 4, 5, 8, 10, 11, 12}, numbers) require.Equal([]int64{3, 4, 5, 8, 10, 11, 12}, numbers)
}
_, err = ParseRangeNumbers("3-a") _, err = ParseRangeNumbers("3-a")
assert.Error(err) require.Error(err)
} }