mse: Tons of fixes and improvements

[btrtrc.git] / mse / mse_test.go

diff --git a/mse/mse_test.go b/mse/mse_test.go
index d26d90be2916fc1f1d674424c8381bf3c8dc16b3..644a4cdfffab7efcc1859d7b6d5e3264eb550cf6 100644 (file)
--- a/mse/mse_test.go
+++ b/mse/mse_test.go
@@ -49,7 +49,7 @@ func TestHandshake(t *testing.T) {
        wg.Add(2)
        go func() {
                defer wg.Done()
-               a, err := Handshake(a, true, []byte("yep"))
+               a, err := InitiateHandshake(a, []byte("yep"))
                if err != nil {
                        t.Fatal(err)
                        return
@@ -61,7 +61,7 @@ func TestHandshake(t *testing.T) {
        }()
        go func() {
                defer wg.Done()
-               b, err := Handshake(b, false, []byte("yep"))
+               b, err := ReceiveHandshake(b, [][]byte{[]byte("nope"), []byte("yep"), []byte("maybe")})
                if err != nil {
                        t.Fatal(err)
                        return