def test_purges(self):
self.assertTrue(Program.objects.exists())
self.assertTrue(Notification.objects.exists())
self.assertTrue(
purge_objects.purge('schools') == purge_objects.error_msg)
self.assertTrue(purge_objects.purge('') == purge_objects.no_args_msg)
self.assertIn("test-programs", purge_objects.purge('test-programs'))
self.assertTrue(Program.objects.exists())
self.assertIn("programs", purge_objects.purge('programs'))
self.assertFalse(Program.objects.exists())
self.assertIn("notifications", purge_objects.purge('notifications'))
self.assertFalse(Notification.objects.exists())
def test_purges(self):
self.assertTrue(Program.objects.exists())
self.assertTrue(Notification.objects.exists())
self.assertTrue(purge_objects.purge('schools') ==
purge_objects.error_msg)
self.assertTrue(purge_objects.purge('') ==
purge_objects.no_args_msg)
self.assertIn("test-programs", purge_objects.purge('test-programs'))
self.assertTrue(Program.objects.exists())
self.assertIn("programs", purge_objects.purge('programs'))
self.assertFalse(Program.objects.exists())
self.assertIn("notifications", purge_objects.purge('notifications'))
self.assertFalse(Notification.objects.exists())
def handle(self, *args, **options):
msg = purge(options['objects'])
self.stdout.write(msg)