mirror of https://github.com/bjdgyc/anylink.git
31 lines
606 B
Go
31 lines
606 B
Go
package sessdata
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewSession(t *testing.T) {
|
|
assert := assert.New(t)
|
|
sessions = make(map[string]*Session)
|
|
sess := NewSession("")
|
|
token := sess.Token
|
|
v, ok := sessions[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()
|
|
}
|