def test_timezone(self): from django.conf import settings for switch in [False, True]: settings.USE_TZ = switch test_job = models.TestJob() self.assertRaisesMessage(AssertionError, 'RuntimeWarning not triggered by save', self.assertWarnsRegex, RuntimeWarning, 'DateTimeField TestJob.closure received a naive datetime', test_job.save)
def test_job_with_data_file(self): test_file = ContentFile('DUMMY CONTENT', 'foobar.txt') # Base case test_job = models.TestJob() test_job.save() expected_path = self.build_path('testjob', '1', 'data', 'foobar.txt') self.assertFalse(os.path.exists(expected_path)) annotation = models.TestFile(job=test_job, data=test_file) annotation.save() self.assertTrue(os.path.exists(expected_path)) self.assertTrue(os.path.isfile(expected_path)) # test_job_two = models.TestJobTwo() test_job_two.save() expected_path = self.build_path('testjobtwo', '1', 'data', 'foobar.txt') self.assertFalse(os.path.exists(expected_path)) annotation = models.JobDataFuncTestFile(job=test_job_two, data=test_file) annotation.save() self.assertTrue(os.path.exists(expected_path)) # # Not supported anymore # expected_path = self.build_path('testjob', '1', 'my_data_str', 'foobar.txt') # self.assertFalse(os.path.exists(expected_path)) # annotation = models.MyDataStrTestFile(job=test_job, data=test_file) # annotation.save() # self.assertTrue(os.path.exists(expected_path)) # expected_path = self.build_path('testjob', '1', 'my_data_func', 'foobar.txt') self.assertFalse(os.path.exists(expected_path)) annotation = models.MyDataFuncTestFile(job=test_job, data=test_file) annotation.save() self.assertTrue(os.path.exists(expected_path)) # self.assertEqual(test_job.delete(), (3, {'tests.MyDataFuncTestFile': 1, 'tests.TestFile': 1, 'tests.TestJob': 1})) self.assertFalse(os.path.exists(expected_path)) self.assertFalse(os.path.exists(self.build_path('testjob', '1'))) self.assertTrue(os.path.exists(self.build_path('testjob'))) # # test_job = models.MyRootFuncTestJob() test_job.save() expected_path = self.build_path('my_root_func', '1', 'data', 'foobar.txt') self.assertFalse(os.path.exists(expected_path)) annotation = models.MyRootFuncTestFile(job=test_job, data=test_file) annotation.save() self.assertTrue(os.path.exists(expected_path)) # expected_path = self.build_path('my_root_func', '1', 'my_data_func', 'foobar.txt') self.assertFalse(os.path.exists(expected_path)) annotation = models.MyRootDataFuncTestFile(job=test_job, data=test_file) annotation.save() self.assertTrue(os.path.exists(expected_path))
def test_job(self): # Base case expected_path = self.build_path('testjob', '1', 'results') self.assertFalse(os.path.isdir(expected_path)) test_job = models.TestJob() test_job.save() self.assertTrue(os.path.isdir(expected_path)) # expected_path = self.build_path(models.MyRootStrTestJob.root_job, '1', 'results') self.assertFalse(os.path.isdir(expected_path)) test_job = models.MyRootStrTestJob() test_job.save() self.assertTrue(os.path.isdir(expected_path)) # expected_path = self.build_path('my_root_func', '1', 'results') self.assertFalse(os.path.isdir(expected_path)) test_job = models.MyRootFuncTestJob() test_job.save() self.assertTrue(os.path.isdir(expected_path)) # expected_path = self.build_path('myresultsstrtestjob', '1', 'my_results_str') self.assertFalse(os.path.isdir(expected_path)) test_job = models.MyResultsStrTestJob() test_job.save() self.assertTrue(os.path.isdir(expected_path)) # expected_path = self.build_path('myresultsfunctestjob', '1', 'my_results_func', ) self.assertFalse(os.path.isdir(expected_path)) test_job = models.MyResultsFuncTestJob() test_job.save() self.assertTrue(os.path.isdir(expected_path)) # expected_path = self.build_path('jobresultsfunctestjob', '1', 'results') self.assertFalse(os.path.isdir(expected_path)) test_job = models.JobResultsFuncTestJob() test_job.save() self.assertTrue(os.path.isdir(expected_path)) # expected_path = self.build_path('my_root_func', '1', 'my_results_func') self.assertFalse(os.path.isdir(expected_path)) test_job = models.MyRootResultsFuncTestJob() test_job.save() self.assertTrue(os.path.isdir(expected_path))
def test_unprepared_job(self): job = models.TestJob() mt = JobHolder(job) job.save() # mt will still have no _job_pk workflow = chain(mt, tasks.identity_task.s()) self.assertRaises(job.DoesNotExist, workflow.apply_async, debug=True)
def test_unsaved_job(self): job = models.TestJob() mt = JobHolder(job) self.assertRaises(job.DoesNotExist, mt.pre_serialization)
def setUp(self): self.job = models.TestJob() self.job.save() self.holder = JobHolder(self.job) self.app_root = os.path.join(settings.django_settings.MEDIA_ROOT, settings.APP_MEDIA_ROOT)