1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-11-16 09:26:21 -05:00
v2fly/common/protocol/server_spec_test.go

72 lines
1.7 KiB
Go
Raw Normal View History

2016-07-25 10:48:09 -04:00
package protocol_test
import (
"testing"
2016-07-25 15:31:16 -04:00
"time"
2016-07-25 10:48:09 -04:00
2018-08-26 18:33:27 -04:00
"v2ray.com/core/common"
2017-12-17 18:07:50 -05:00
"v2ray.com/core/common/net"
2016-08-20 14:55:45 -04:00
. "v2ray.com/core/common/protocol"
2017-12-17 18:07:50 -05:00
"v2ray.com/core/common/uuid"
"v2ray.com/core/proxy/vmess"
2017-10-24 10:15:35 -04:00
. "v2ray.com/ext/assert"
2016-07-25 10:48:09 -04:00
)
2016-07-25 15:31:16 -04:00
func TestAlwaysValidStrategy(t *testing.T) {
2017-10-24 10:15:35 -04:00
assert := With(t)
2016-07-25 15:31:16 -04:00
strategy := AlwaysValid()
2017-10-24 10:15:35 -04:00
assert(strategy.IsValid(), IsTrue)
2016-07-25 15:31:16 -04:00
strategy.Invalidate()
2017-10-24 10:15:35 -04:00
assert(strategy.IsValid(), IsTrue)
2016-07-25 15:31:16 -04:00
}
func TestTimeoutValidStrategy(t *testing.T) {
2017-10-24 10:15:35 -04:00
assert := With(t)
2016-07-25 15:31:16 -04:00
strategy := BeforeTime(time.Now().Add(2 * time.Second))
2017-10-24 10:15:35 -04:00
assert(strategy.IsValid(), IsTrue)
2016-07-25 15:31:16 -04:00
time.Sleep(3 * time.Second)
2017-10-24 10:15:35 -04:00
assert(strategy.IsValid(), IsFalse)
2016-07-25 15:31:16 -04:00
strategy = BeforeTime(time.Now().Add(2 * time.Second))
strategy.Invalidate()
2017-10-24 10:15:35 -04:00
assert(strategy.IsValid(), IsFalse)
2016-07-25 15:31:16 -04:00
}
2017-12-17 18:07:50 -05:00
func TestUserInServerSpec(t *testing.T) {
assert := With(t)
uuid1 := uuid.New()
uuid2 := uuid.New()
2018-08-26 18:33:27 -04:00
toAccount := func(a *vmess.Account) Account {
account, err := a.AsAccount()
common.Must(err)
return account
}
2018-08-26 18:11:32 -04:00
spec := NewServerSpec(net.Destination{}, AlwaysValid(), &MemoryUser{
2017-12-17 18:07:50 -05:00
Email: "test1@v2ray.com",
2018-08-26 18:33:27 -04:00
Account: toAccount(&vmess.Account{Id: uuid1.String()}),
2017-12-17 18:07:50 -05:00
})
2018-08-26 18:11:32 -04:00
assert(spec.HasUser(&MemoryUser{
2017-12-17 18:07:50 -05:00
Email: "test1@v2ray.com",
2018-08-26 18:33:27 -04:00
Account: toAccount(&vmess.Account{Id: uuid2.String()}),
2017-12-17 18:07:50 -05:00
}), IsFalse)
2018-08-26 18:11:32 -04:00
spec.AddUser(&MemoryUser{Email: "test2@v2ray.com"})
assert(spec.HasUser(&MemoryUser{
2017-12-17 18:07:50 -05:00
Email: "test1@v2ray.com",
2018-08-26 18:33:27 -04:00
Account: toAccount(&vmess.Account{Id: uuid1.String()}),
2017-12-17 18:07:50 -05:00
}), IsTrue)
}
func TestPickUser(t *testing.T) {
assert := With(t)
2018-08-26 18:11:32 -04:00
spec := NewServerSpec(net.Destination{}, AlwaysValid(), &MemoryUser{Email: "test1@v2ray.com"}, &MemoryUser{Email: "test2@v2ray.com"}, &MemoryUser{Email: "test3@v2ray.com"})
2017-12-17 18:07:50 -05:00
user := spec.PickUser()
assert(user.Email, HasSuffix, "@v2ray.com")
}