X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/7cd900d61edbd6d322db3cecb913adf574389320..aa91955aa4072310fa8113e2fc3e4be4bc188f45:/hotline/transfer_test.go?ds=sidebyside

diff --git a/hotline/transfer_test.go b/hotline/transfer_test.go
index fb7d39c..2bcee11 100644
--- a/hotline/transfer_test.go
+++ b/hotline/transfer_test.go
@@ -135,7 +135,7 @@ func Test_receiveFile(t *testing.T) {
 						},
 					}
 					fakeFileData := []byte{1, 2, 3}
-					b := testFile.BinaryMarshal()
+					b, _ := io.ReadAll(&testFile)
 					b = append(b, fakeFileData...)
 					return bytes.NewReader(b)
 				}(),
@@ -188,7 +188,7 @@ func Test_receiveFile(t *testing.T) {
 			targetFile := &bytes.Buffer{}
 			resForkFile := &bytes.Buffer{}
 			infoForkFile := &bytes.Buffer{}
-			err := receiveFile(tt.args.conn, targetFile, resForkFile, infoForkFile)
+			err := receiveFile(tt.args.conn, targetFile, resForkFile, infoForkFile, io.Discard)
 			if !tt.wantErr(t, err, fmt.Sprintf("receiveFile(%v, %v, %v)", tt.args.conn, targetFile, resForkFile)) {
 				return
 			}