Skip to content

Commit dd7762d

Browse files
committed
Now works with both 2.7 and 3.5
1 parent aef9bd4 commit dd7762d

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

test/tck/steps/bolt_compability_steps.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,15 +80,17 @@ def step_impl(context, size, type):
8080

8181
@when("the driver asks the server to echo this (?P<unused>.+) back")
8282
def step_impl(context, unused):
83-
context.results = [ send_string("RETURN " + as_cypher_text(context.expected)).stream(), send_parameters("RETURN {input}", {'input': context.expected}).stream()]
83+
context.results = [ send_string("RETURN " + as_cypher_text(context.expected)), send_parameters("RETURN {input}", {'input': context.expected})]
8484

8585

8686
@step("the value given in the result should be the same as what was sent")
8787
def step_impl(context):
8888
assert len(context.results) > 0
8989
for result in context.results:
90-
result_value = result.next().values()[0]
90+
assert result.next()
91+
result_value = result.record().values()[0]
9192
assert result_value == context.expected
93+
assert result.at_end()
9294

9395

9496
def as_cypher_text(expected):

test/tck/steps/driver_equality_steps.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def step_impl(context, key1, key2):
3737

3838
@step("saved values should all equal")
3939
def step_impl(context):
40-
values = context.values.values()
40+
values = list(context.values.values())
4141
assert len(values) > 1
4242
first_val = values.pop()
4343
for item in values:
@@ -46,7 +46,7 @@ def step_impl(context):
4646

4747
@step("none of the saved values should be equal")
4848
def step_impl(context):
49-
values = context.values.values()
49+
values = list(context.values.values())
5050
assert len(values) > 1
5151
first_val = values.pop()
5252
for item in values:

0 commit comments

Comments
 (0)