Fix a write_test case to embed a jpeg file

tagatac · Nov 18, 2024 · 81d5161 · 81d5161
1 parent 3c6af3e
commit 81d5161
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/bagoup/write_test.go b/internal/bagoup/write_test.go
@@ -512,7 +512,7 @@ func TestWriteFile(t *testing.T) {
 					ofMock.EXPECT().WriteAttachment("attachment1.heic"),
 					osMock.EXPECT().FileExist("attachment2.jpeg").Return(true, nil),
 					osMock.EXPECT().HEIC2JPG("attachment2.jpeg").Return("attachment2.jpeg", nil),
-					ofMock.EXPECT().WriteAttachment("attachment2.jpeg"),
+					ofMock.EXPECT().WriteAttachment("attachment2.jpeg").Return(true, nil),
 					ofMock.EXPECT().Name().Return("messages-export/friend/iMessage;-;friend@gmail.com;;;iMessage;-;friend@hotmail.com.pdf"),
 					ofMock.EXPECT().ReferenceAttachment("att3transfer.png"),
 					ofMock.EXPECT().Stage(),
@@ -521,6 +521,7 @@ func TestWriteFile(t *testing.T) {
 				)
 			},
 			wantJPGs:     1,
+			wantEmbedded: 1,
 			wantConvFail: 1,
 		},
 		{