Adds unit test for SetFields (fixes #341).

This commit is contained in:
Romain Dorgueil
2019-06-02 16:43:11 +02:00
parent eb83d0327e
commit 29d07aa5f1

View File

@ -142,3 +142,14 @@ def test_map_fields_error():
context.write_sync(tuple())
assert context.status == "!"
assert context.defunct
def test_set_fields():
with BufferingNodeExecutionContext(bonobo.SetFields(["x", "y"])) as context:
context.write_sync((1, 2))
output = context.get_buffer()
assert len(output) == 1
assert output[0]._fields == ("x", "y")
assert output[0].x == 1
assert output[0].y == 2