class PersistenceTest(TestCase):
def setUp(self):
temp_dir = tempfile.mkdtemp(prefix='persistence_test')
self.persistence_file_name = os.path.join(temp_dir, 'name')
self.persistence = Persistence(self.persistence_file_name)
def tearDown(self):
self.persistence.close()
os.unlink(self.persistence_file_name)
os.rmdir(os.path.dirname(self.persistence_file_name))
def test_with_block(self):
report = self.persistence.report
assert_that(report).is_none()
report_object = "foo"
self.persistence.close()
with Persistence(self.persistence_file_name) as persistence:
persistence.report = report_object
with Persistence(self.persistence_file_name) as persistence:
report = persistence.report
assert_that(report).is_equal_to(report_object)
def test_with_block_with_non_existent_path(self):
self.assertRaises(Exception, self.with_block_with_non_existent_path)
def with_block_with_non_existent_path(self):
with Persistence('/non/existent/path') as _:
pass
def test_add_job_info(self):
self.persistence.add_job_info('job_name', 4234)
assert_that(self.persistence.job_info).contains_entry({'job_name': 4234})
def setUp(self):
temp_dir = tempfile.mkdtemp(prefix='persistence_test')
self.persistence_file_name = os.path.join(temp_dir, 'name')
self.persistence = Persistence(self.persistence_file_name)
