diff options
| -rw-r--r-- | archaeological_finds/tests.py | 42 |
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 |