From 39713c578255dd95a42d67096cec0c85c7ba8e8b Mon Sep 17 00:00:00 2001
From: Daniel Kesselberg <mail@danielkesselberg.de>
Date: Mon, 9 Sep 2024 20:52:52 +0200
Subject: [PATCH] test: update attachment tests to avoid deprecation warning

Signed-off-by: Daniel Kesselberg <mail@danielkesselberg.de>
---
 .../Unit/Service/TransmissionServiceTest.php  | 51 ++++++++-----------
 1 file changed, 21 insertions(+), 30 deletions(-)

diff --git a/tests/Unit/Service/TransmissionServiceTest.php b/tests/Unit/Service/TransmissionServiceTest.php
index 202240ef48..db060d2f30 100644
--- a/tests/Unit/Service/TransmissionServiceTest.php
+++ b/tests/Unit/Service/TransmissionServiceTest.php
@@ -19,14 +19,12 @@
 use OCA\Mail\Exception\AttachmentNotFoundException;
 use OCA\Mail\Exception\ServiceException;
 use OCA\Mail\Exception\SmimeSignException;
-use OCA\Mail\Model\Message;
 use OCA\Mail\Service\Attachment\AttachmentService;
 use OCA\Mail\Service\GroupsIntegration;
 use OCA\Mail\Service\SmimeService;
 use OCA\Mail\Service\TransmissionService;
 use OCP\AppFramework\Db\DoesNotExistException;
 use OCP\Files\SimpleFS\InMemoryFile;
-use OCP\Files\SimpleFS\ISimpleFile;
 use PHPUnit\Framework\MockObject\MockObject;
 use Psr\Log\LoggerInterface;
 
@@ -85,50 +83,43 @@ public function testGetAttachments() {
 		$this->assertEquals($expected, $actual);
 	}
 
-	public function testHandleAttachment() {
-		$id = 1;
-		$expected = [
-			'type' => 'local',
-			'id' => $id
-		];
-		[$localAttachment, $file] = [
-			new LocalAttachment(),
-			$this->createMock(ISimpleFile::class)
-		];
-		$message = $this->createMock(Message::class);
+	public function testHandleAttachment(): void {
 		$mailAccount = new MailAccount();
 		$mailAccount->setUserId('bob');
 		$account = new Account($mailAccount);
 
+		$attachment = new LocalAttachment();
+		$attachment->setFileName('test.txt');
+		$attachment->setMimeType('text/plain');
+
+		$file = new InMemoryFile(
+			'test.txt',
+			"Hello, I'm a test file."
+		);
+
 		$this->attachmentService->expects(self::once())
 			->method('getAttachment')
-			->willReturn([$localAttachment, $file]);
+			->willReturn([$attachment, $file]);
 		$this->logger->expects(self::never())
 			->method('warning');
 
-		$this->transmissionService->handleAttachment($account, $expected);
+		$part = $this->transmissionService->handleAttachment($account, ['id' => 1, 'type' => 'local']);
+
+		$this->assertEquals('test.txt', $part->getContentTypeParameter('name'));
 	}
 
-	public function testHandleAttachmentNoId() {
-		$attachment = [[
-			'type' => 'local',
-		]];
-		$message = $this->createMock(Message::class);
-		$account = new Account(new MailAccount());
+	public function testHandleAttachmentNoId(): void {
+		$mailAccount = new MailAccount();
+		$mailAccount->setUserId('bob');
+		$account = new Account($mailAccount);
 
 		$this->logger->expects(self::once())
 			->method('warning');
 
-		$this->transmissionService->handleAttachment($account, $attachment);
+		$this->transmissionService->handleAttachment($account, ['type' => 'local']);
 	}
 
-	public function testHandleAttachmentNotFound() {
-		$attachment = [
-			'id' => 1,
-			'type' => 'local',
-		];
-
-		$message = $this->createMock(Message::class);
+	public function testHandleAttachmentNotFound(): void {
 		$mailAccount = new MailAccount();
 		$mailAccount->setUserId('bob');
 		$account = new Account($mailAccount);
@@ -139,7 +130,7 @@ public function testHandleAttachmentNotFound() {
 		$this->logger->expects(self::once())
 			->method('warning');
 
-		$this->transmissionService->handleAttachment($account, $attachment);
+		$this->transmissionService->handleAttachment($account, ['id' => 1, 'type' => 'local']);
 	}
 
 	public function testGetSignMimePart() {