class TaskTestCase(TestCase): fixtures = ['countries', 'user', 'work', 'drafts'] def setUp(self): self.task = Task(title='foo', description='bar') def test_discard_bad_work_country(self): na = Country.objects.get(country__pk='NA') self.task.country = na self.task.work = Work.objects.get(frbr_uri='/za/act/2014/10') self.task.clean() self.assertEqual(self.task.country, na) self.assertIsNone(self.task.work) def test_discard_bad_work_locality(self): za = Country.objects.get(country__pk='ZA') loc = za.localities.all()[0] self.task.country = za self.task.locality = loc self.task.work = Work.objects.get(frbr_uri='/za/act/2014/10') self.task.clean() self.assertEqual(self.task.country, za) self.assertEqual(self.task.locality, loc) self.assertIsNone(self.task.work)
class TaskTestCase(TestCase): fixtures = ['countries', 'user', 'work', 'drafts'] def setUp(self): self.user = User.objects.get(pk=1) self.task = Task( title='foo', description='bar', created_by_user=self.user) def test_discard_bad_work_country(self): na = Country.objects.get(country__pk='NA') self.task.country = na self.task.work = Work.objects.get(frbr_uri='/za/act/2014/10') self.task.clean() self.assertEqual(self.task.country, na) self.assertIsNone(self.task.work) def test_discard_bad_work_locality(self): za = Country.objects.get(country__pk='ZA') loc = za.localities.all()[0] self.task.country = za self.task.locality = loc self.task.work = Work.objects.get(frbr_uri='/za/act/2014/10') self.task.clean() self.assertEqual(self.task.country, za) self.assertEqual(self.task.locality, loc) self.assertIsNone(self.task.work) def test_cancel(self): za = Country.objects.get(country__pk='ZA') self.task.country = za self.task.cancel(self.user) self.task.save() self.task.refresh_from_db() self.assertEqual(Task.CANCELLED, self.task.state) self.assertIsNotNone(self.task.closed_at) self.assertIsNone(self.task.reviewed_by_user) def test_reopen(self): za = Country.objects.get(country__pk='ZA') self.task.country = za self.task.cancel(self.user) self.task.save() self.task.refresh_from_db() self.task.reopen(self.user) self.assertEqual(Task.OPEN, self.task.state) self.assertIsNone(self.task.closed_at) self.assertIsNone(self.task.reviewed_by_user) def test_close(self): za = Country.objects.get(country__pk='ZA') self.task.state = Task.PENDING_REVIEW self.task.country = za self.task.close(self.user) self.task.save() self.task.refresh_from_db() self.assertEqual(Task.DONE, self.task.state) self.assertIsNotNone(self.task.closed_at) self.assertEqual(self.user, self.task.reviewed_by_user)