def test_init_with_args(self): """ Test that a QueryRequestPayload object can be constructed with valid values. """ query.QueryRequestPayload(self.query_functions_a) query.QueryRequestPayload(self.query_functions_b) query.QueryRequestPayload(self.query_functions_c)
def _test_write(self, encoding, query_functions): stream = utils.BytearrayStream() payload = query.QueryRequestPayload(query_functions) payload.write(stream) length_expected = len(encoding) length_received = len(stream) msg = "encoding lengths not equal" msg += "; expected {0}, received {1}".format(length_expected, length_received) self.assertEqual(length_expected, length_received, msg) msg = "encoding mismatch" msg += ";\nexpected:\n{0}\nreceived:\n{1}".format(encoding, stream) self.assertEqual(encoding, stream, msg)
def _test_read(self, stream, query_functions): payload = query.QueryRequestPayload() payload.read(stream) expected = len(query_functions) observed = len(payload.query_functions) msg = "query functions list decoding mismatch" msg += "; expected {0} results, received {1}".format( expected, observed) self.assertEqual(expected, observed, msg) for i in xrange(len(query_functions)): expected = query_functions[i] observed = payload.query_functions[i] msg = "query function decoding mismatch" msg += "; expected {0}, received {1}".format(expected, observed) self.assertEqual(expected, observed, msg)
def _build_query_batch_item(self, query_functions=None): operation = Operation(OperationEnum.QUERY) payload = query.QueryRequestPayload(query_functions) batch_item = messages.RequestBatchItem( operation=operation, request_payload=payload) return batch_item
def _create_query_payload(self): return query.QueryRequestPayload()
def test_init_with_none(self): """ Test that a QueryRequestPayload object can be constructed with no specified value. """ query.QueryRequestPayload()