Tests: find basket search

1 files changed, 42 insertions, 0 deletions

diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py
index bb249e2ea..c9ee0f9ed 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -1075,6 +1075,48 @@ class FindTest(FindInit, TestCase):
         )
 
 
+class FindBasketTest(FindInit, TestCase):
+    fixtures = WAREHOUSE_FIXTURES
+    model = models.FindBasket
+    SEARCH_URL = "get-findbasket"
+
+    def setUp(self):
+        self.create_finds(force=True)
+        self.create_finds(force=True)
+        self.username = "myuser"
+        self.password = "mypassword"
+        u = User.objects.create_superuser(self.username, "myemail@test.com",
+                                          self.password)
+        self.client = Client()
+        b = models.FindBasket.objects.create(label="test basket", user_id=u.pk)
+        b.items.add(self.finds[0])
+        b.items.add(self.finds[1])
+        models.FindBasket.objects.create(label="other", user_id=u.pk)
+
+    def test_search(self):
+        c = Client()
+        search = {}
+
+        # no result when no authentication
+        response = c.get(reverse("get-findbasket"), search)
+        content = response.content.decode()
+        res = json.loads(content)
+        self.assertEqual(res, [])
+
+        c.login(username=self.username, password=self.password)
+
+        response = c.get(reverse("get-findbasket"), search)
+        self.assertEqual(response.status_code, 200)
+        content = response.content.decode()
+        res = json.loads(content)
+        self.assertEqual(res["recordsTotal"], 2)
+        search = {"search_vector" : 'other'}
+        response = c.get(reverse("get-findbasket"), search)
+        content = response.content.decode()
+        res = json.loads(content)
+        self.assertEqual(res["recordsTotal"], 1)
+
+
 class FindSearchTest(FindInit, TestCase, SearchText):
     fixtures = WAREHOUSE_FIXTURES
     model = models.Find