def test_serdes(self):
param_dict = {
'health': 'health',
'mesos_task_state': 'mesos_task_state',
'is_draining': True,
'is_healthy': True,
'offer': 'offer',
'resources': 'resources',
}
assert json.loads(MesosTaskParameters(**param_dict).serialize()) == param_dict
assert MesosTaskParameters.deserialize(json.dumps(param_dict)) == MesosTaskParameters(**param_dict)
def test_serdes(self):
param_dict = {
"health": "health",
"mesos_task_state": "mesos_task_state",
"is_draining": True,
"is_healthy": True,
"offer": "offer",
"resources": "resources",
}
assert json.loads(
MesosTaskParameters(**param_dict).serialize()) == param_dict
assert MesosTaskParameters.deserialize(
json.dumps(param_dict)) == MesosTaskParameters(**param_dict)
def then_get_all_tasks_should_return(context, return_json):
all_tasks = context.task_store.get_all_tasks()
expected_tasks = {
k: MesosTaskParameters(**v)
for k, v in json.loads(return_json).items()
}
assert all_tasks == expected_tasks
def test_DictTaskStore():
task_store = DictTaskStore(service_name="foo", instance_name="bar", framework_id='foo', system_paasta_config=None)
task_store.add_task_if_doesnt_exist("task_id", mesos_task_state="foo")
task_store.update_task("task_id", is_draining=True)
assert task_store.get_all_tasks() == {
"task_id": MesosTaskParameters(
mesos_task_state="foo",
is_draining=True,
),
}
task_store.update_task("task_id", mesos_task_state="bar")
assert task_store.get_all_tasks() == {
"task_id": MesosTaskParameters(
mesos_task_state="bar",
is_draining=True,
),
}
def when_we_overwrite_task(context, task_id, params_json):
context.task_store.overwrite_task(
task_id, MesosTaskParameters(**json.loads(params_json)))