kirillius
/
Cloak
mirror of https://github.com/cbeuw/Cloak
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix tests

This commit is contained in:
Andy Wang 2019-10-20 15:18:27 +01:00
parent 57f0c3d20a
commit 3649283a49
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
internal/server/activeuser_test.go
View File

@ -1,13 +1,29 @@
package server package server
import ( import (
"crypto/rand"
"encoding/base64" "encoding/base64"
mux "github.com/cbeuw/Cloak/internal/multiplex" mux "github.com/cbeuw/Cloak/internal/multiplex"
"github.com/cbeuw/Cloak/internal/server/usermanager" "github.com/cbeuw/Cloak/internal/server/usermanager"
"github.com/cbeuw/Cloak/internal/util"
"os" "os"
"testing" "testing"
) )
func getSeshConfig(unordered bool) *mux.SessionConfig {
sessionKey := make([]byte, 32)
rand.Read(sessionKey)
obfuscator, _ := mux.GenerateObfs(0x00, sessionKey, true)
seshConfig := &mux.SessionConfig{
Obfuscator: obfuscator,
Valve: nil,
UnitRead: util.ReadTLS,
Unordered: unordered,
}
return seshConfig
}
func TestActiveUser_Bypass(t *testing.T) { func TestActiveUser_Bypass(t *testing.T) {
manager, err := usermanager.MakeLocalManager(MOCK_DB_NAME) manager, err := usermanager.MakeLocalManager(MOCK_DB_NAME)
if err != nil { if err != nil {
@ -20,7 +36,7 @@ func TestActiveUser_Bypass(t *testing.T) {
var existing bool var existing bool
var sesh1 *mux.Session var sesh1 *mux.Session
t.Run("get first session", func(t *testing.T) { t.Run("get first session", func(t *testing.T) {
sesh0, existing, err = user.GetSession(0, &mux.SessionConfig{}) sesh0, existing, err = user.GetSession(0, getSeshConfig(false))
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }
@ -47,7 +63,7 @@ func TestActiveUser_Bypass(t *testing.T) {
} }
}) })
t.Run("get second session", func(t *testing.T) { t.Run("get second session", func(t *testing.T) {
sesh1, existing, err = user.GetSession(1, &mux.SessionConfig{}) sesh1, existing, err = user.GetSession(1, getSeshConfig(false))
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }
@ -79,7 +95,7 @@ func TestActiveUser_Bypass(t *testing.T) {
} }
}) })
t.Run("get session again after termination", func(t *testing.T) { t.Run("get session again after termination", func(t *testing.T) {
seshx, existing, err := user.GetSession(0, &mux.SessionConfig{}) seshx, existing, err := user.GetSession(0, getSeshConfig(false))
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }