mirror of https://github.com/cbeuw/Cloak
Add and fix some tests
parent
e0e33e12d6
commit
237b9d131b
@ -0,0 +1,35 @@
|
|||||||
|
package multiplex
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math/rand"
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
"testing/quick"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestOobfs(t *testing.T) {
|
||||||
|
sessionKey := make([]byte, 32)
|
||||||
|
rand.Read(sessionKey)
|
||||||
|
obfuscator, err := GenerateObfs(0x01, sessionKey)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("failed to generate obfuscator %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
f := &Frame{}
|
||||||
|
_testFrame, _ := quick.Value(reflect.TypeOf(f), rand.New(rand.NewSource(42)))
|
||||||
|
testFrame := _testFrame.Interface().(*Frame)
|
||||||
|
obfsed, err := obfuscator.Obfs(testFrame)
|
||||||
|
if err != nil {
|
||||||
|
t.Error("failed to obfs ", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resultFrame, err := obfuscator.Deobfs(obfsed)
|
||||||
|
if err != nil {
|
||||||
|
t.Error("failed to deobfs ", err)
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(testFrame, resultFrame) {
|
||||||
|
t.Error("expecting", testFrame,
|
||||||
|
"got", resultFrame)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue