class ActionAdminTests(BasePbsTestCase): """ Test the functionality of the ActionAdmin. """ def setUp(self): super(ActionAdminTests, self).setUp() self.prescription = self.make(Prescription) user = User.objects.create(username="******") url = reverse("admin:risk_action_changelist", args=(quote(self.prescription.id),)) self.request = self._mocked_authenticated_request(url, user) self.admin = ActionAdmin(Prescription, site) self.admin.prescription = self.prescription def test_get_list_display(self): """ Test get_list_display with different request types. """ # if there are no pre_burn, day_of_burn or post_burn variables set Prescription.is_draft = True self.assertIn("add_action", self.admin.get_list_display(self.request)) Prescription.is_draft = False self.assertNotIn("add_action", self.admin.get_list_display(self.request)) # pre-burn self.request.GET = {"pre_burn": True} Prescription.is_draft = True self.assertEqual( ("__str__", "details", "pre_burn_resolved", "pre_burn_explanation"), self.admin.get_list_display(self.request), ) Prescription.is_draft = False self.assertEqual( ( "__str__", "details", "pre_burn_resolved", "pre_burn_explanation", "pre_burn_approved", "pre_burn_approver", ), self.admin.get_list_display(self.request), ) # day-of-burn self.request.GET = {"day_of_burn": True} Prescription.is_draft = True self.assertEqual( ( "__str__", "details", "day_of_burn_include", "day_of_burn_situation", "day_of_burn_mission", "day_of_burn_execution", "day_of_burn_administration", "day_of_burn_command", ), self.admin.get_list_display(self.request), ) Prescription.is_draft = False self.assertEqual( ( "__str__", "details", "day_of_burn_include", "day_of_burn_situation", "day_of_burn_mission", "day_of_burn_execution", "day_of_burn_administration", "day_of_burn_command", ), self.admin.get_list_display(self.request), ) # post-burn self.request.GET = {"post_burn": True} Prescription.is_draft = True self.assertIn("add_action", self.admin.get_list_display(self.request)) Prescription.is_draft = False self.assertNotIn("add_action", self.admin.get_list_display(self.request)) def test_get_list_editable(self): pass def test_get_readonly_fields(self): pass
class ActionAdminTests(BasePbsTestCase): """ Test the functionality of the ActionAdmin. """ def setUp(self): super(ActionAdminTests, self).setUp() self.prescription = self.make(Prescription) user = User.objects.create(username='******') url = reverse('admin:risk_action_changelist', args=(quote(self.prescription.id),)) self.request = self._mocked_authenticated_request(url, user) self.admin = ActionAdmin(Prescription, site) self.admin.prescription = self.prescription def test_get_list_display(self): """ Test get_list_display with different request types. """ # if there are no pre_burn, day_of_burn or post_burn variables set Prescription.is_draft = True self.assertIn("add_action", self.admin.get_list_display(self.request)) Prescription.is_draft = False self.assertNotIn("add_action", self.admin.get_list_display(self.request)) # pre-burn self.request.GET = {'pre_burn': True} Prescription.is_draft = True self.assertEqual(("__str__", "details", "pre_burn_resolved", "pre_burn_explanation"), self.admin.get_list_display(self.request)) Prescription.is_draft = False self.assertEqual(("__str__", "details", "pre_burn_resolved", "pre_burn_explanation", "pre_burn_approved", "pre_burn_approver"), self.admin.get_list_display(self.request)) # day-of-burn self.request.GET = {'day_of_burn': True} Prescription.is_draft = True self.assertEqual(("__str__", "details", "day_of_burn_include", "day_of_burn_situation", "day_of_burn_mission", "day_of_burn_execution", "day_of_burn_administration", "day_of_burn_command"), self.admin.get_list_display(self.request)) Prescription.is_draft = False self.assertEqual(("__str__", "details", "day_of_burn_include", "day_of_burn_situation", "day_of_burn_mission", "day_of_burn_execution", "day_of_burn_administration", "day_of_burn_command"), self.admin.get_list_display(self.request)) # post-burn self.request.GET = {'post_burn': True} Prescription.is_draft = True self.assertIn("add_action", self.admin.get_list_display(self.request)) Prescription.is_draft = False self.assertNotIn("add_action", self.admin.get_list_display(self.request)) def test_get_list_editable(self): pass def test_get_readonly_fields(self): pass