Minor comments and test improvement

[btrtrc.git] / mse / mse_test.go

diff --git a/mse/mse_test.go b/mse/mse_test.go
index d74dec97bbcedd9cb7d5b66b6700ea79951fa54e..6c2799542e79f02966e9d353940beaca1bf115b3 100644 (file)
--- a/mse/mse_test.go
+++ b/mse/mse_test.go
@@ -10,6 +10,7 @@ import (
        "testing"
 
        "github.com/bradfitz/iter"
+       "github.com/stretchr/testify/require"
 )
 
 func TestReadUntil(t *testing.T) {
@@ -117,5 +118,10 @@ func (tr *trackReader) Read(b []byte) (n int, err error) {
 
 func TestReceiveRandomData(t *testing.T) {
        tr := trackReader{rand.Reader, 0}
-       ReceiveHandshake(readWriter{&tr, ioutil.Discard}, nil)
+       _, err := ReceiveHandshake(readWriter{&tr, ioutil.Discard}, nil)
+       // No skey matches
+       require.Error(t, err)
+       // Establishing S, and then reading the maximum padding for giving up on
+       // synchronizing.
+       require.EqualValues(t, 96+532, tr.n)
 }