From c87a023745e49d63967eaa6d5ce0ccadab514c25 Mon Sep 17 00:00:00 2001
From: Daniel Yohan <dyohan9@gmail.com>
Date: Thu, 12 Dec 2019 11:29:26 -0300
Subject: [PATCH] Updated Evaluate Tests

---
 bothub/api/v2/tests/test_evaluate.py | 40 ++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/bothub/api/v2/tests/test_evaluate.py b/bothub/api/v2/tests/test_evaluate.py
index 4e91e66e..61d165fa 100644
--- 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"