1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2026-01-23 01:15:31 -05:00

Fix test break

This commit is contained in:
V2Ray
2015-09-20 00:11:14 +02:00
parent 075753c030
commit b40be74b86
15 changed files with 50 additions and 48 deletions

View File

@@ -1,21 +1,21 @@
package mocks
import (
"github.com/v2ray/v2ray-core/proxy/vmess/protocol"
"github.com/v2ray/v2ray-core/proxy/vmess/protocol/user"
)
type MockUserSet struct {
UserIds []protocol.ID
UserIds []user.ID
UserHashes map[string]int
Timestamps map[string]int64
}
func (us *MockUserSet) AddUser(user protocol.User) error {
func (us *MockUserSet) AddUser(user user.User) error {
us.UserIds = append(us.UserIds, user.Id)
return nil
}
func (us *MockUserSet) GetUser(userhash []byte) (*protocol.ID, int64, bool) {
func (us *MockUserSet) GetUser(userhash []byte) (*user.ID, int64, bool) {
idx, found := us.UserHashes[string(userhash)]
if found {
return &us.UserIds[idx], us.Timestamps[string(userhash)], true