def test_task_entity_get_files(self):
core_snapshot_entity = CoreSnapshot(self.input_dict)
snapshot_entity = Snapshot(core_snapshot_entity)
# Test failure because entity has not been created by controller
failed = False
try:
snapshot_entity.get_files()
except DoesNotExist:
failed = True
assert failed
# Test success
snapshot_entity = self.__setup()
result = snapshot_entity.get_files()
assert len(result) == 1
assert isinstance(result[0], TextIOWrapper)
assert result[0].mode == "r"
assert "script.py" in result[0].name
snapshot_entity = self.__setup()
result = snapshot_entity.get_files(mode="a")
assert len(result) == 1
assert isinstance(result[0], TextIOWrapper)
assert result[0].mode == "a"
assert "script.py" in result[0].name
def test_snapshot_entity_instantiate(self):
core_snapshot_entity = CoreSnapshot(self.input_dict)
snapshot_entity = Snapshot(core_snapshot_entity)
for k, v in self.input_dict.items():
if k != "file_collection_id":
assert getattr(snapshot_entity, k) == v
assert snapshot_entity.task_id == None
assert snapshot_entity.label == None
assert snapshot_entity.created_at
def test_snapshot_entity_instantiate(self):
input_dict = {
"id": "test",
"model_id": "my_model",
"session_id": "my_session",
"message": "my test snapshot",
"code_id": "my_code",
"environment_id": "my_environment",
"file_collection_id": "my_files",
"config": {},
"stats": {}
}
core_snapshot_entity = CoreSnapshot(input_dict)
snapshot_entity = Snapshot(core_snapshot_entity, home=self.temp_dir)
for k, v in input_dict.items():
assert getattr(snapshot_entity, k) == v
assert snapshot_entity.task_id == None
assert snapshot_entity.label == None
assert snapshot_entity.created_at