mirror of https://github.com/bjdgyc/anylink.git
32 lines
600 B
Go
32 lines
600 B
Go
package sessdata
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewSession(t *testing.T) {
|
|
assert := assert.New(t)
|
|
sessions = sync.Map{}
|
|
sess := NewSession()
|
|
token := sess.Token
|
|
v, ok := sessions.Load(token)
|
|
assert.True(ok)
|
|
assert.Equal(sess, v)
|
|
}
|
|
|
|
func TestConnSession(t *testing.T) {
|
|
assert := assert.New(t)
|
|
preIpData()
|
|
defer closeIpdata()
|
|
sess := NewSession()
|
|
cSess := sess.NewConn()
|
|
cSess.RateLimit(100, true)
|
|
assert.Equal(cSess.BandwidthUp, uint32(100))
|
|
cSess.RateLimit(200, false)
|
|
assert.Equal(cSess.BandwidthDown, uint32(200))
|
|
cSess.Close()
|
|
}
|