def test_from_dict(self):
data_dict = {
'id': 'task-7',
'signature': {
'options': {
'task_id': 'task-7'
},
'task': None,
'args': (),
'kwargs': {},
},
'dependencies': {
'task-4': None
},
}
w1 = WorkflowNode.from_dict(data_dict)
assert w1.id == 'task-7'
assert w1.signature.id == 'task-7'
assert w1.dependencies == {'task-4': None}
def test_from_dict_with_custom_payload(self):
data_dict = {
'id': 'task-7',
'signature': {
'options': {
'task_id': 'task-7'
},
'task': None,
'args': (),
'kwargs': {},
},
'dependencies': {
'task-4': None
},
'custom_payload': {
'foo': 42
}
}
w1 = WorkflowNode.from_dict(data_dict)
assert w1.id == 'task-7'
assert w1.signature.id == 'task-7'
assert w1.dependencies == {'task-4': None}
assert w1.custom_payload['foo'] == 42