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)