openstack/deb-python-cassandra-driver
Code Issues Proposed changes

update custom payload integration tests for PR #655

Browse Source
This commit is contained in:
Adam Holmberg
2016-10-07 11:40:29 -05:00
parent ba6f0e8ef4
commit a6a9f79358
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/integration/standard/test_custom_protocol_handler.py
View File

@@ -130,8 +130,9 @@ class CustomResultMessageRaw(ResultMessage):
paging_state, column_metadata = cls.recv_results_metadata(f, user_type_map) paging_state, column_metadata = cls.recv_results_metadata(f, user_type_map)
rowcount = read_int(f) rowcount = read_int(f)
rows = [cls.recv_row(f, len(column_metadata)) for _ in range(rowcount)] rows = [cls.recv_row(f, len(column_metadata)) for _ in range(rowcount)]
colnames = [c[2] for c in column_metadata]
coltypes = [c[3] for c in column_metadata] coltypes = [c[3] for c in column_metadata]
return (paging_state, (coltypes, rows)) return paging_state, coltypes, (colnames, rows)
class CustomTestRawRowType(ProtocolHandler): class CustomTestRawRowType(ProtocolHandler):
@@ -166,7 +167,7 @@ class CustomResultMessageTracked(ResultMessage):
tuple(ctype.from_binary(val, protocol_version) tuple(ctype.from_binary(val, protocol_version)
for ctype, val in zip(coltypes, row)) for ctype, val in zip(coltypes, row))
for row in rows] for row in rows]
return (paging_state, (colnames, parsed_rows)) return paging_state, coltypes, (colnames, parsed_rows)
class CustomProtocolHandlerResultMessageTracked(ProtocolHandler): class CustomProtocolHandlerResultMessageTracked(ProtocolHandler):