]> git.r.bdr.sh - rbdr/mobius/blobdiff - hotline/transfer_test.go
Update goreleaser config
[rbdr/mobius] / hotline / transfer_test.go
index fb7d39cfa89da66614d2db0797c994dc852ce335..2bcee11ae7f5b3106ce2c31ec133665afdcff5f5 100644 (file)
@@ -135,7 +135,7 @@ func Test_receiveFile(t *testing.T) {
                                                },
                                        }
                                        fakeFileData := []byte{1, 2, 3}
                                                },
                                        }
                                        fakeFileData := []byte{1, 2, 3}
-                                       b := testFile.BinaryMarshal()
+                                       b, _ := io.ReadAll(&testFile)
                                        b = append(b, fakeFileData...)
                                        return bytes.NewReader(b)
                                }(),
                                        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{}
                        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
                        }
                        if !tt.wantErr(t, err, fmt.Sprintf("receiveFile(%v, %v, %v)", tt.args.conn, targetFile, resForkFile)) {
                                return
                        }