From a6a9f79358d65464e473c964cb0eb4c47f3dcc73 Mon Sep 17 00:00:00 2001 From: Adam Holmberg Date: Fri, 7 Oct 2016 11:40:29 -0500 Subject: [PATCH] update custom payload integration tests for PR #655 --- tests/integration/standard/test_custom_protocol_handler.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/integration/standard/test_custom_protocol_handler.py b/tests/integration/standard/test_custom_protocol_handler.py index c6818f7f..89b1b2fd 100644 --- a/tests/integration/standard/test_custom_protocol_handler.py +++ b/tests/integration/standard/test_custom_protocol_handler.py @@ -130,8 +130,9 @@ class CustomResultMessageRaw(ResultMessage): paging_state, column_metadata = cls.recv_results_metadata(f, user_type_map) rowcount = read_int(f) 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] - return (paging_state, (coltypes, rows)) + return paging_state, coltypes, (colnames, rows) class CustomTestRawRowType(ProtocolHandler): @@ -166,7 +167,7 @@ class CustomResultMessageTracked(ResultMessage): tuple(ctype.from_binary(val, protocol_version) for ctype, val in zip(coltypes, row)) for row in rows] - return (paging_state, (colnames, parsed_rows)) + return paging_state, coltypes, (colnames, parsed_rows) class CustomProtocolHandlerResultMessageTracked(ProtocolHandler):