fix(testing): implement getExpectedRuntime on FakeTextProcessingProvider

Signed-off-by: Julius Härtl <jus@bitgrid.net>

3 files changed, 52 insertions, 1 deletions

diff --git a/apps/testing/lib/AppInfo/Application.php b/apps/testing/lib/AppInfo/Application.php
index 6e9421a030e..cbbbf6fc4ea 100644
--- a/apps/testing/lib/AppInfo/Application.php
+++ b/apps/testing/lib/AppInfo/Application.php
@@ -27,6 +27,7 @@ namespace OCA\Testing\AppInfo;
 use OCA\Testing\AlternativeHomeUserBackend;
 use OCA\Testing\Provider\FakeText2ImageProvider;
 use OCA\Testing\Provider\FakeTextProcessingProvider;
+use OCA\Testing\Provider\FakeTextProcessingProviderSync;
 use OCA\Testing\Provider\FakeTranslationProvider;
 use OCP\AppFramework\App;
 use OCP\AppFramework\Bootstrap\IBootContext;
@@ -41,6 +42,7 @@ class Application extends App implements IBootstrap {
 	public function register(IRegistrationContext $context): void {
 		$context->registerTranslationProvider(FakeTranslationProvider::class);
 		$context->registerTextProcessingProvider(FakeTextProcessingProvider::class);
+		$context->registerTextProcessingProvider(FakeTextProcessingProviderSync::class);
 		$context->registerTextToImageProvider(FakeText2ImageProvider::class);
 	}
 
diff --git a/apps/testing/lib/Provider/FakeTextProcessingProvider.php b/apps/testing/lib/Provider/FakeTextProcessingProvider.php
index f95b528dfbe..845acba7573 100644
--- a/apps/testing/lib/Provider/FakeTextProcessingProvider.php
+++ b/apps/testing/lib/Provider/FakeTextProcessingProvider.php
@@ -30,7 +30,7 @@ use OCP\TextProcessing\ITaskType;
 class FakeTextProcessingProvider implements IProvider {
 
 	public function getName(): string {
-		return 'Fake text processing provider';
+		return 'Fake text processing provider (asynchronous)';
 	}
 
 	public function process(string $prompt): string {
diff --git a/apps/testing/lib/Provider/FakeTextProcessingProviderSync.php b/apps/testing/lib/Provider/FakeTextProcessingProviderSync.php
new file mode 100644
index 00000000000..72532d62061
--- /dev/null
+++ b/apps/testing/lib/Provider/FakeTextProcessingProviderSync.php
@@ -0,0 +1,49 @@
+<?php
+/**
+ * @copyright Copyright (c) 2023 Julius Härtl <jus@bitgrid.net>
+ *
+ * @author Julius Härtl <jus@bitgrid.net>
+ *
+ * @license GNU AGPL version 3 or any later version
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+namespace OCA\Testing\Provider;
+
+use OCP\TextProcessing\FreePromptTaskType;
+use OCP\TextProcessing\IProviderWithExpectedRuntime;
+use OCP\TextProcessing\ITaskType;
+
+/**
+ * @template-implements IProviderWithExpectedRuntime<FreePromptTaskType|ITaskType>
+ */
+class FakeTextProcessingProviderSync implements IProviderWithExpectedRuntime {
+
+	public function getName(): string {
+		return 'Fake text processing provider (synchronous)';
+	}
+
+	public function process(string $prompt): string {
+		return strrev($prompt);
+	}
+
+	public function getTaskType(): string {
+		return FreePromptTaskType::class;
+	}
+
+	public function getExpectedRuntime(): int {
+		return 1;
+	}
+}