fix(answers): serialization and e2e tests

Author: aallam

src/commonTest/kotlin/serialize/answers/TestAnswers.kt

@@ -0,0 +1,57 @@
+package serialize.answers
+
+import com.algolia.search.ExperimentalAlgoliaClientAPI
+import com.algolia.search.model.Attribute
+import com.algolia.search.model.search.AnswersQuery
+import com.algolia.search.model.search.Language
+import com.algolia.search.serialize.KeyAttributesForPrediction
+import com.algolia.search.serialize.KeyFilters
+import com.algolia.search.serialize.KeyNbHits
+import com.algolia.search.serialize.KeyParams
+import com.algolia.search.serialize.KeyQuery
+import com.algolia.search.serialize.KeyQueryLanguages
+import com.algolia.search.serialize.KeyThreshold
+import com.algolia.search.serialize.internal.JsonNoDefaults
+import kotlinx.serialization.json.JsonElement
+import kotlinx.serialization.json.add
+import kotlinx.serialization.json.buildJsonArray
+import kotlinx.serialization.json.buildJsonObject
+import kotlinx.serialization.json.put
+import serialize.TestSerializer
+
+@OptIn(ExperimentalAlgoliaClientAPI::class)
+internal class TestAnswers : TestSerializer<AnswersQuery>(AnswersQuery.serializer(), JsonNoDefaults) {
+
+    override val items: List<Pair<AnswersQuery, JsonElement>> = listOf(
+
+        AnswersQuery(
+            query = "query",
+            queryLanguages = listOf(Language.English),
+            attributesForPrediction = listOf(Attribute("a1"), Attribute("a2")),
+            nbHits = 10,
+            threshold = 20f
+        ) to buildJsonObject {
+            put(KeyQuery, "query")
+            put(KeyQueryLanguages, buildJsonArray { add(Language.English.raw) })
+            put(KeyAttributesForPrediction, buildJsonArray {
+                add("a1")
+                add("a2")
+            })
+            put(KeyNbHits, 10)
+            put(KeyThreshold, 20f)
+        },
+
+        AnswersQuery(
+            query = "query",
+            queryLanguages = listOf(Language.English),
+        ).apply {
+            filters = "brand:sony"
+        } to buildJsonObject {
+            put(KeyQuery, "query")
+            put(KeyQueryLanguages, buildJsonArray { add(Language.English.raw) })
+            put(KeyParams, buildJsonObject {
+                put(KeyFilters, "brand:sony")
+            })
+        }
+    )
+}

src/commonTest/kotlin/suite/TestSuiteAnswers.kt

@@ -2,14 +2,13 @@ package suite
 
 import clientAnswers
 import com.algolia.search.ExperimentalAlgoliaClientAPI
+import com.algolia.search.model.Attribute
 import com.algolia.search.model.IndexName
 import com.algolia.search.model.search.AnswersQuery
 import com.algolia.search.model.search.Language
-import kotlinx.serialization.json.jsonObject
-import kotlinx.serialization.json.jsonPrimitive
-import runBlocking
-import shouldEqual
 import kotlin.test.Test
+import kotlin.test.assertFalse
+import runBlocking
 
 internal class TestSuiteAnswers {
 
@@ -18,55 +17,18 @@ internal class TestSuiteAnswers {
 
     @OptIn(ExperimentalAlgoliaClientAPI::class)
     @Test
-    fun withResult() {
-        runBlocking {
-            val query = AnswersQuery(
-                query = "sir ken robinson",
-                queryLanguages = listOf(Language.English)
-            )
-
-            val response = answers.findAnswers(query)
-
-            response.nbHits shouldEqual 10
-        }
-    }
-
-    @OptIn(ExperimentalAlgoliaClientAPI::class)
-    @Test
-    fun withoutResult() {
+    fun testFindAnswers() {
         runBlocking {
             val query = AnswersQuery(
-                query = "what",
-                queryLanguages = listOf(Language.English)
-            )
-
-            val response = answers.findAnswers(query)
-
-            response.nbHits shouldEqual 0
-        }
-    }
-
-    @OptIn(ExperimentalAlgoliaClientAPI::class)
-    @Test
-    fun withHighlight() {
-        runBlocking {
-            val query = AnswersQuery(
-                query = "sarah jones",
+                query = "when do babies start learning?",
                 queryLanguages = listOf(Language.English),
-                nbHits = 2,
-            ).apply {
-                highlightPreTag = "_pre_"
-                highlightPostTag = "_post_"
-            }
+                nbHits = 20,
+                attributesForPrediction = listOf(Attribute("description"), Attribute("title"), Attribute("transcript"))
+            )
 
             val response = answers.findAnswers(query)
 
-            response.nbHits shouldEqual 2
-            response.hits[0]
-                .highlightResult["main_speaker"]
-                ?.jsonObject?.get("value")
-                ?.jsonPrimitive?.content shouldEqual "_pre_Sarah_post_ _pre_Jones_post_"
-
+            assertFalse(response.hits.map { it.answer }.isEmpty())
         }
     }
 }