From b51400d119059248275986413a362cb3bb5f00ee Mon Sep 17 00:00:00 2001
From: Michael <chebotarev.michael@yandex.ru>
Date: Sun, 19 Mar 2023 16:16:58 +0000
Subject: [PATCH] fix cipher-otc timeout test

---
 otp/cipher_test.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/otp/cipher_test.go b/otp/cipher_test.go
index 6e5c7a5..67d7cda 100644
--- a/otp/cipher_test.go
+++ b/otp/cipher_test.go
@@ -54,6 +54,7 @@ func TestReader(t *testing.T) {
 
 		err    error
 		result []byte
+		limit  bool
 	}{
 		{
 			name:   "simple",
@@ -83,8 +84,9 @@ func TestReader(t *testing.T) {
 			name:   "timeout",
 			r:      iotest.TimeoutReader(bytes.NewBuffer(plaintext)),
 			prng:   bytes.NewBuffer(randomBytes),
-			result: ciphertext[:bytes.MinRead],
+			result: ciphertext,
 			err:    iotest.ErrTimeout,
+			limit:  true,
 		},
 	} {
 		t.Run(testCase.name, func(t *testing.T) {
@@ -92,7 +94,11 @@ func TestReader(t *testing.T) {
 
 			buf, err := ioutil.ReadAll(r)
 			require.ErrorIs(t, err, testCase.err)
-			require.Equal(t, testCase.result, buf)
+			if testCase.limit {
+				require.Equal(t, testCase.result[:len(buf)], buf)
+			} else {
+				require.Equal(t, testCase.result, buf)
+			}
 		})
 	}
 }