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()