fix test

Signed-off-by: alperozturk <alper_ozturk@proton.me>

Author: alperozturk96

library/src/androidTest/java/com/owncloud/android/lib/resources/assistant/AssistantIT.kt

@@ -9,6 +9,9 @@
 package com.owncloud.android.lib.resources.assistant
 
 import com.owncloud.android.AbstractIT
+import com.owncloud.android.lib.resources.assistant.model.TaskInputShape
+import com.owncloud.android.lib.resources.assistant.model.TaskOutputShape
+import com.owncloud.android.lib.resources.assistant.model.TaskTypeData
 import com.owncloud.android.lib.resources.status.NextcloudVersion
 import junit.framework.TestCase.assertEquals
 import junit.framework.TestCase.assertTrue
@@ -21,12 +24,34 @@ class AssistantIT : AbstractIT() {
         testOnlyOnServer(NextcloudVersion.nextcloud_28)
     }
 
+    private fun getTaskType(): TaskTypeData {
+        return TaskTypeData(
+            "core:text2text",
+            "Free text to text prompt",
+            "Runs an arbitrary prompt through a language model that returns a reply",
+            listOf(
+                TaskInputShape(
+                    "Prompt",
+                    "Describe a task that you want the assistant to do or ask a question",
+                    "Text"
+                )
+            ),
+            listOf(
+                TaskOutputShape(
+                    "Generated reply",
+                    "The generated text from the assistant",
+                    "Text"
+                )
+            )
+        )
+    }
+
     @Test
     fun testGetTaskTypes() {
         val result = GetTaskTypesRemoteOperation().execute(nextcloudClient)
         assertTrue(result.isSuccess)
 
-        val taskTypes = result.resultData.types
+        val taskTypes = result.resultData
         assertTrue(taskTypes.isNotEmpty())
     }
 
@@ -38,8 +63,8 @@ class AssistantIT : AbstractIT() {
 
         // create one task
         val input = "Give me some random output for test purpose"
-        val type = "OCP\\TextProcessing\\FreePromptTaskType"
-        assertTrue(CreateTaskRemoteOperation(input, type).execute(nextcloudClient).isSuccess)
+        val taskType = getTaskType()
+        assertTrue(CreateTaskRemoteOperation(input, taskType).execute(nextcloudClient).isSuccess)
 
         result = GetTaskListRemoteOperation("assistant").execute(nextcloudClient)
         assertTrue(result.isSuccess)
@@ -52,8 +77,8 @@ class AssistantIT : AbstractIT() {
     fun testDeleteTask() {
         // create one task
         val input = "Give me some random output for test purpose"
-        val type = "OCP\\TextProcessing\\FreePromptTaskType"
-        assertTrue(CreateTaskRemoteOperation(input, type).execute(nextcloudClient).isSuccess)
+        val taskType = getTaskType()
+        assertTrue(CreateTaskRemoteOperation(input, taskType).execute(nextcloudClient).isSuccess)
 
         var result = GetTaskListRemoteOperation("assistant").execute(nextcloudClient)
         assertTrue(result.isSuccess)

library/src/main/java/com/owncloud/android/lib/resources/assistant/GetTaskTypesRemoteOperation.kt

@@ -20,7 +20,6 @@ import com.owncloud.android.lib.resources.assistant.model.TaskTypes
 import org.apache.commons.httpclient.HttpStatus
 
 class GetTaskTypesRemoteOperation : OCSRemoteOperation<List<TaskTypeData>>() {
-
     private val supportedTaskType = "Text"
 
     @Suppress("TooGenericExceptionCaught")
@@ -39,17 +38,19 @@ class GetTaskTypesRemoteOperation : OCSRemoteOperation<List<TaskTypeData>>() {
                         object : TypeToken<ServerResponse<TaskTypes>>() {}
                     )?.ocs?.data?.types
 
-                val taskTypeList = response?.map { (key, value) ->
-                    value.copy(id = value.id ?: key)
-                }
+                val taskTypeList =
+                    response?.map { (key, value) ->
+                        value.copy(id = value.id ?: key)
+                    }
 
-                val supportedTaskTypeList = taskTypeList?.filter { taskType ->
-                    taskType.inputShape?.any { inputShape ->
-                        inputShape.type == supportedTaskType
-                    } == true && taskType.outputShape?.any { outputShape ->
-                        outputShape.type == supportedTaskType
-                    } == true
-                }
+                val supportedTaskTypeList =
+                    taskTypeList?.filter { taskType ->
+                        taskType.inputShape?.any { inputShape ->
+                            inputShape.type == supportedTaskType
+                        } == true && taskType.outputShape?.any { outputShape ->
+                            outputShape.type == supportedTaskType
+                        } == true
+                    }
 
                 result = RemoteOperationResult(true, getMethod)
                 result.setResultData(supportedTaskTypeList)

library/src/main/java/com/owncloud/android/lib/resources/assistant/model/TaskTypes.kt

@@ -8,7 +8,7 @@
 
 package com.owncloud.android.lib.resources.assistant.model
 
-data class TaskTypes(val types:  Map<String, TaskTypeData>)
+data class TaskTypes(val types: Map<String, TaskTypeData>)
 
 data class TaskTypeData(
     val id: String?,
@@ -21,11 +21,11 @@ data class TaskTypeData(
 data class TaskInputShape(
     val name: String?,
     val description: String?,
-    val type: String?,
+    val type: String?
 )
 
 data class TaskOutputShape(
     val name: String?,
     val description: String?,
-    val type: String?,
+    val type: String?
 )