def test_file_empty(self):
test_dir = tempfile.mkdtemp()
file_name = Constants.CORE_STATE_FILE
# test on empty file
self.runtime.create_temp_file(test_dir, file_name, content=None)
core_state_handler = CoreStateHandler(os.path.join(test_dir),
self.json_file_handler)
core_state_json = core_state_handler.read_file()
self.assertTrue(core_state_json is None)
self.assertFalse(
hasattr(core_state_json, self.core_state_fields.number))
# test on file with empty JSON
self.runtime.create_temp_file(test_dir, file_name, "{}")
core_state_json = core_state_handler.read_file()
self.assertTrue(hasattr(core_state_json,
self.core_state_fields.number))
self.assertTrue(hasattr(core_state_json,
self.core_state_fields.action))
self.assertTrue(core_state_json.number is None)
self.assertTrue(core_state_json.action is None)
# Remove the directory after the test
shutil.rmtree(test_dir)
def test_file_does_not_exists(self):
core_state_handler = CoreStateHandler(
os.path.join(os.path.pardir, "tests", "helper"),
self.json_file_handler)
core_state_json = core_state_handler.read_file()
self.assertTrue(core_state_handler.read_file() is None)
self.assertFalse(
hasattr(core_state_json, self.core_state_fields.number))
self.assertFalse(
hasattr(core_state_json, self.core_state_fields.last_heartbeat))
def test_key_not_in_file(self):
parent_key = self.core_state_fields.parent_key
core_state_json = {parent_key: {"test_no": 1}}
test_dir = tempfile.mkdtemp()
file_name = Constants.CORE_STATE_FILE
self.runtime.create_temp_file(test_dir, file_name, "{}")
core_state_handler = CoreStateHandler(os.path.join(test_dir),
self.json_file_handler)
core_state_handler.read_file()
seq_no = self.core_state_fields.number
self.assertTrue(
core_state_handler.json_file_handler.get_json_config_value_safely(
core_state_json, seq_no, parent_key, False) is None)
# Remove the directory after the test
shutil.rmtree(test_dir)
def test_success_file_read(self):
core_state_handler = CoreStateHandler(
os.path.join(os.path.pardir, "tests", "helpers"),
self.json_file_handler)
core_state_json = core_state_handler.read_file()
self.assertTrue(core_state_json.number is not None)
self.assertTrue(core_state_json.completed is not None)
self.assertEqual(core_state_json.action, "Assessment")
self.assertEqual(core_state_json.number, 1234)
def test_file_exists(self):
core_state_handler = CoreStateHandler(
os.path.join(os.path.pardir, "tests", "helpers"),
self.json_file_handler)
self.assertTrue(core_state_handler.read_file() is not None)