def test_remove_record_from_local_tracking(self):
try:
# Remove
remove_record_from_local_tracking(self.fake_job_dict.keys()[0],
config=self.config)
dict_copy = self.fake_job_dict.copy()
dict_copy.pop("job_name")
# Can I use this in a test? I guess I'm double-testing
with get_shelve_database(config=self.config) as db:
self.assertEqual(dict_copy, db)
finally:
with get_shelve_database(config=self.config) as db:
db = {}
db.update(self.fake_job_dict)
def test_get_shelve_database(self):
with get_shelve_database(config=self.config) as db:
assert(db)