test(fakeAI): Allow to specify whether the fake providers should failbugfix/noid/allow-to-fail-fake-AI-providers

Signed-off-by: Joas Schilling <coding@schilljs.com>

1 files changed, 9 insertions, 1 deletions

diff --git a/apps/testing/lib/TaskProcessing/FakeTextToImageProvider.php b/apps/testing/lib/TaskProcessing/FakeTextToImageProvider.php
index 429b3af9d57..cd508be9155 100644
--- a/apps/testing/lib/TaskProcessing/FakeTextToImageProvider.php
+++ b/apps/testing/lib/TaskProcessing/FakeTextToImageProvider.php
@@ -9,7 +9,9 @@ declare(strict_types=1);
 namespace OCA\Testing\TaskProcessing;
 
 use OCA\Testing\AppInfo\Application;
+use OCP\AppFramework\Services\IAppConfig;
 use OCP\TaskProcessing\EShapeType;
+use OCP\TaskProcessing\Exception\ProcessingException;
 use OCP\TaskProcessing\ISynchronousProvider;
 use OCP\TaskProcessing\ShapeDescriptor;
 use OCP\TaskProcessing\TaskTypes\TextToImage;
@@ -17,7 +19,9 @@ use RuntimeException;
 
 class FakeTextToImageProvider implements ISynchronousProvider {
 
-	public function __construct() {
+	public function __construct(
+		protected IAppConfig $appConfig,
+	) {
 	}
 
 	public function getId(): string {
@@ -77,6 +81,10 @@ class FakeTextToImageProvider implements ISynchronousProvider {
 	}
 
 	public function process(?string $userId, array $input, callable $reportProgress): array {
+		if ($this->appConfig->getAppValueBool('fail-' . $this->getId())) {
+			throw new ProcessingException('Failing as set by AppConfig');
+		}
+
 		if (!isset($input['input']) || !is_string($input['input'])) {
 			throw new RuntimeException('Invalid prompt');
 		}