test: Test AI watermarking

+ add TextToImage test

Signed-off-by: Marcel Klehr <[email protected]>

Author: marcelklehr

tests/res/speech.mp3

@@ -18,11 +18,13 @@
 use OCA\OpenAi\Service\OpenAiAPIService;
 use OCA\OpenAi\Service\OpenAiSettingsService;
 use OCA\OpenAi\Service\QuotaRuleService;
+use OCA\OpenAi\Service\WatermarkingService;
 use OCA\OpenAi\TaskProcessing\ChangeToneProvider;
 use OCA\OpenAi\TaskProcessing\EmojiProvider;
 use OCA\OpenAi\TaskProcessing\HeadlineProvider;
 use OCA\OpenAi\TaskProcessing\ProofreadProvider;
 use OCA\OpenAi\TaskProcessing\SummaryProvider;
+use OCA\OpenAi\TaskProcessing\TextToImageProvider;
 use OCA\OpenAi\TaskProcessing\TextToSpeechProvider;
 use OCA\OpenAi\TaskProcessing\TextToTextProvider;
 use OCA\OpenAi\TaskProcessing\TranslateProvider;
@@ -581,11 +583,16 @@ public function testTextToSpeechProvider(): void {
 			$this->createMock(\Psr\Log\LoggerInterface::class),
 			\OCP\Server::get(IAppConfig::class),
 			self::TEST_USER1,
+			\OCP\Server::get(WatermarkingService::class),
 		);
 
 		$inputText = 'This is a test prompt';
 
-		$response = 'BINARYDATA';
+		$response = file_get_contents(__DIR__ . '/../../res/speech.mp3');
+
+		if (!$response) {
+			throw new \RuntimeException('Could not read test resourcce `speech.mp3`');
+		}
 
 		$url = self::OPENAI_API_BASE . 'audio/speech';
 
@@ -605,7 +612,7 @@ public function testTextToSpeechProvider(): void {
 		$this->iClient->expects($this->once())->method('post')->with($url, $options)->willReturn($iResponse);
 
 		$result = $TTSProvider->process(self::TEST_USER1, ['input' => $inputText], fn () => null);
-		$this->assertEquals(['speech' => 'BINARYDATA'], $result);
+		$this->assertArrayHasKey('speech', $result);
 
 		// Check that token usage is logged properly (should be 21 characters)
 		$usage = $this->quotaUsageMapper->getQuotaUnitsOfUser(self::TEST_USER1, Application::QUOTA_TYPE_SPEECH);
@@ -614,4 +621,59 @@ public function testTextToSpeechProvider(): void {
 		$this->quotaUsageMapper->deleteUserQuotaUsages(self::TEST_USER1);
 	}
 
+	public function testTextToImageProvider(): void {
+		$TTSProvider = new TextToImageProvider(
+			$this->openAiApiService,
+			$this->createMock(\OCP\IL10N::class),
+			$this->createMock(\Psr\Log\LoggerInterface::class),
+			\OCP\Server::get(IClientService::class),
+			\OCP\Server::get(IAppConfig::class),
+			self::TEST_USER1,
+			\OCP\Server::get(WatermarkingService::class),
+		);
+
+		$inputText = 'This is a test prompt';
+
+		$responseImage = file_get_contents(__DIR__ . '/../../res/trees.jpg');
+
+		if (!$responseImage) {
+			throw new \RuntimeException('Could not read test resourcce `trees.jpg`');
+		}
+
+		$response = json_encode([
+			'data' => [
+				[
+					'b64_json' => base64_encode($responseImage),
+				]
+			]
+		]);
+
+		$url = self::OPENAI_API_BASE . 'images/generations';
+
+		$options = ['timeout' => Application::OPENAI_DEFAULT_REQUEST_TIMEOUT, 'headers' => ['User-Agent' => Application::USER_AGENT, 'Authorization' => self::AUTHORIZATION_HEADER, 'Content-Type' => 'application/json']];
+		$options['body'] = json_encode([
+			'prompt' => $inputText,
+			'size' => '1024x1024',
+			'n' => 1,
+			'model' => Application::DEFAULT_IMAGE_MODEL_ID,
+		]);
+
+		$iResponse = $this->createMock(\OCP\Http\Client\IResponse::class);
+		$iResponse->method('getHeader')->with('Content-Type')->willReturn('application/json');
+		$iResponse->method('getBody')->willReturn($response);
+		$iResponse->method('getStatusCode')->willReturn(200);
+
+		$this->iClient->expects($this->once())->method('post')->with($url, $options)->willReturn($iResponse);
+
+		$result = $TTSProvider->process(self::TEST_USER1, ['input' => $inputText], fn () => null);
+		$this->assertArrayHasKey('images', $result);
+		$this->assertArrayHasKey(0, $result['images']);
+
+		// Check that token usage is logged properly (should be 21 characters)
+		$usage = $this->quotaUsageMapper->getQuotaUnitsOfUser(self::TEST_USER1, Application::QUOTA_TYPE_IMAGE);
+		$this->assertEquals(1, $usage);
+		// Clear quota usage
+		$this->quotaUsageMapper->deleteUserQuotaUsages(self::TEST_USER1);
+	}
+
 }