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()
}