Updated Evaluate Tests

weni-ai · Dec 12, 2019 · c87a023 · c87a023
1 parent 0a2b406
commit c87a023
Showing 1 changed file with 38 additions and 2 deletions.
diff --git a/bothub/api/v2/tests/test_evaluate.py b/bothub/api/v2/tests/test_evaluate.py
@@ -57,11 +57,12 @@ def setUp(self):
             intent="greet",
         )
 
-    def request(self, token, version=None):
+    def request(self, token, version=None, language=None):
         authorization_header = {"HTTP_AUTHORIZATION": "Token {}".format(token.key)}
         request = self.factory.get(
             f"/v2/evaluate/?repository_uuid={self.repository.uuid}"
-            + (f"&repository_version={version}" if version else ""),
+            + (f"&repository_version={version}" if version else "")
+            + (f"&language={language}" if language else ""),
             **authorization_header,
         )
         response = EvaluateViewSet.as_view({"get": "list"})(
@@ -78,6 +79,41 @@ def test_okay(self):
         self.assertEqual(len(content_data["results"]), 1)
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
+    def test_okay_language(self):
+        repository_version_language = RepositoryVersionLanguage.objects.create(
+            repository_version=self.repository_version,
+            language=languages.LANGUAGE_PT,
+            algorithm="statistical_model",
+        )
+
+        RepositoryExample.objects.create(
+            repository_version_language=repository_version_language,
+            text="test",
+            intent="greet",
+        )
+
+        RepositoryEvaluate.objects.create(
+            repository_version_language=repository_version_language,
+            text="test",
+            intent="greet",
+        )
+
+        RepositoryEvaluate.objects.create(
+            repository_version_language=repository_version_language,
+            text="test2",
+            intent="greet",
+        )
+        response, content_data = self.request(
+            self.owner_token, language=languages.LANGUAGE_PT
+        )
+
+        self.assertEqual(content_data["count"], 2)
+        self.assertEqual(len(content_data["results"]), 2)
+        self.assertEqual(
+            content_data["results"][0].get("language"), languages.LANGUAGE_PT
+        )
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
     def test_with_version(self):
         repository_version = RepositoryVersion.objects.create(
             repository=self.repository, name="new_test"