1 files changed, 108 insertions, 0 deletions
diff --git a/apps/testing/lib/TaskProcessing/FakeTextToImageProvider.php b/apps/testing/lib/TaskProcessing/FakeTextToImageProvider.php
new file mode 100644
index 00000000000..455d6c0b518
--- /dev/null
+++ b/apps/testing/lib/TaskProcessing/FakeTextToImageProvider.php
@@ -0,0 +1,108 @@
+<?php
+
+/**
+ * SPDX-FileCopyrightText: 2023 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+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;
+use RuntimeException;
+
+class FakeTextToImageProvider implements ISynchronousProvider {
+
+	public function __construct(
+		protected IAppConfig $appConfig,
+	) {
+	}
+
+	public function getId(): string {
+		return Application::APP_ID . '-text2image';
+	}
+
+	public function getName(): string {
+		return 'Fake text2image task processing provider';
+	}
+
+	public function getTaskTypeId(): string {
+		return TextToImage::ID;
+	}
+
+	public function getExpectedRuntime(): int {
+		return 1;
+	}
+
+	public function getInputShapeEnumValues(): array {
+		return [];
+	}
+
+	public function getInputShapeDefaults(): array {
+		return [
+			'numberOfImages' => 1,
+		];
+	}
+
+	public function getOptionalInputShape(): array {
+		return [
+			'size' => new ShapeDescriptor(
+				'Size',
+				'Optional. The size of the generated images. Must be in 256x256 format.',
+				EShapeType::Text
+			),
+		];
+	}
+
+	public function getOptionalInputShapeEnumValues(): array {
+		return [];
+	}
+
+	public function getOptionalInputShapeDefaults(): array {
+		return [];
+	}
+
+	public function getOutputShapeEnumValues(): array {
+		return [];
+	}
+
+	public function getOptionalOutputShape(): array {
+		return [];
+	}
+
+	public function getOptionalOutputShapeEnumValues(): array {
+		return [];
+	}
+
+	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');
+		}
+		$prompt = $input['input'];
+
+		$nbImages = 1;
+		if (isset($input['numberOfImages']) && is_int($input['numberOfImages'])) {
+			$nbImages = $input['numberOfImages'];
+		}
+
+		$fakeContent = file_get_contents(__DIR__ . '/../../img/logo.png');
+
+		$output = ['images' => []];
+		foreach (range(1, $nbImages) as $i) {
+			$output['images'][] = $fakeContent;
+		}
+		/** @var array<string, list<numeric|string>|numeric|string> $output */
+		return $output;
+	}
+}