diff options
| -rw-r--r-- | archaeological_finds/tests.py | 61 | ||||
| -rw-r--r-- | archaeological_finds/views.py | 12 |
2 files changed, 68 insertions, 5 deletions
diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index e273069b8..df891e57e 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -1463,6 +1463,67 @@ class FindSearchTest(FindInit, TestCase, SearchText): ) +class FindBasketWizardModificationTest(WizardTest, FindInit, TestCase): + fixtures = WAREHOUSE_FIXTURES + url_name = "find_basket_modification" + wizard_name = "find_basket_edit_wizard" + steps = views.basket_modification_steps + redirect_url = "/find_basket_modification/selec-find_basket_modification?"\ + "open_item={last_id}" + model = models.FindBasket + form_datas = [ + FormData( + "Basket modification", + form_datas={ + "selec-find_basket_modification": {"pk": ""}, + "basket-find_basket_modification": { + "label": "Super OK", + "slug": "super-ok", + "comment": "No", + "shared_with": [], + "shared_write_with": [], + }, + }, + ) + ] + + def pre_wizard(self): + self.basket = models.FindBasket.objects.create( + label="OK", + user=IshtarUser.objects.get(pk=self.user.pk), + ) + self.basket_number = models.FindBasket.objects.count() + user1 = User.objects.create_user( + "user1", "nomail@nomail.com", "user1" + ) + self.ishtar_user1 = IshtarUser.objects.get(user_ptr=user1) + user2 = User.objects.create_user( + "user2", "nomail@nomail.com", "user2" + ) + self.ishtar_user2 = IshtarUser.objects.get(user_ptr=user2) + + data = self.form_datas[0].form_datas + data["selec-find_basket_modification"]["pk"] = self.basket.pk + data["basket-find_basket_modification"]["shared_with"] = [self.ishtar_user1.pk] + data["basket-find_basket_modification"]["shared_write_with"] = [ + self.ishtar_user2.pk, self.ishtar_user1.pk + ] + super().pre_wizard() + + def post_wizard(self): + # no creation + self.assertEqual(models.FindBasket.objects.count(), self.basket_number) + basket = models.FindBasket.objects.get(pk=self.basket.pk) + self.assertEqual( + list(basket.shared_with.values_list("pk", flat=True)), + [self.ishtar_user1.pk] + ) + self.assertEqual( + list(basket.shared_write_with.values_list("pk", flat=True)), + [self.ishtar_user1.pk, self.ishtar_user2.pk] + ) + + class FindAutocompleteTest(FindInit, TestCase): fixtures = WAREHOUSE_FIXTURES model = models.Find diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py index 514c0650a..5d1747032 100644 --- a/archaeological_finds/views.py +++ b/archaeological_finds/views.py @@ -284,12 +284,14 @@ basket_search_wizard = wizards.FindBasketSearch.as_view( url_name="find_basket_search", ) +basket_modification_steps = [ + ("selec-find_basket_modification", forms.FindBasketForWriteFormSelection), + ("basket-find_basket_modification", forms.FindBasketForm), + ("final-find_basket_modification", FinalForm), +] + basket_modify_wizard = wizards.FindBasketEditWizard.as_view( - [ - ("selec-find_basket_modification", forms.FindBasketForWriteFormSelection), - ("basket-find_basket_modification", forms.FindBasketForm), - ("final-find_basket_modification", FinalForm), - ], + basket_modification_steps, label=_("Basket modify"), url_name="find_basket_modification", ) |