def test_unregister_not_registered(self, logger_mock, permission_mock, content_type_mock): content_type_mock.objects.get.side_effect = ContentType.DoesNotExist content_type_mock.DoesNotExist = ContentType.DoesNotExist res = unregister_permissions() self.assertEqual(logger_mock.exception.call_count, 1) self.assertFalse(res)
def test_unregister_not_registered(self, logger_mock, permission_mock, content_type_mock): content_type_mock.objects.get.side_effect = ContentType.DoesNotExist content_type_mock.DoesNotExist = ContentType.DoesNotExist res = unregister_permissions() self.assertEqual(logger_mock.exception.call_count, 1) self.assertFalse(res)
def test_unregister(self, logger_mock, permission_mock, content_type_mock): content_type_obj = MagicMock() content_type_mock.objects.get.return_value = content_type_obj permissions_queryset = MagicMock() permission_mock.objects.filter.return_value = permissions_queryset content_type_queryset = MagicMock() content_type_mock.objects.filter.return_value = content_type_queryset res = unregister_permissions() self.assertEqual(content_type_queryset.delete.call_count, 1) self.assertEqual(permissions_queryset.delete.call_count, 1) self.assertTrue(res)
def test_unregister(self, logger_mock, permission_mock, content_type_mock): content_type_obj = MagicMock() content_type_mock.objects.get.return_value = content_type_obj permissions_queryset = MagicMock() permission_mock.objects.filter.return_value = permissions_queryset content_type_queryset = MagicMock() content_type_mock.objects.filter.return_value = content_type_queryset res = unregister_permissions() self.assertEqual(content_type_queryset.delete.call_count, 1) self.assertEqual(permissions_queryset.delete.call_count, 1) self.assertTrue(res)
def do_remove_audit(): """ Include steps to remove audit. Usable from migrations. """ unregister_permissions()