def _build_discover_versions_batch_item(self, protocol_versions=None):
operation = Operation(OperationEnum.DISCOVER_VERSIONS)
payload = discover_versions.DiscoverVersionsRequestPayload(
protocol_versions)
batch_item = messages.RequestBatchItem(
operation=operation, request_payload=payload)
return batch_item
def test_read_with_one_protocol_version(self):
stream = self.encoding_one
payload = discover_versions.DiscoverVersionsRequestPayload()
protocol_versions = self.protocol_versions_one
self._test_read(stream, payload, protocol_versions)
def test_read_with_empty_protocol_list(self):
stream = self.encoding_empty
payload = discover_versions.DiscoverVersionsRequestPayload()
protocol_versions = self.protocol_versions_empty
self._test_read(stream, payload, protocol_versions)
def test_init_with_args(self):
discover_versions.DiscoverVersionsRequestPayload(
self.protocol_versions_empty)
def test_init_with_none(self):
discover_versions.DiscoverVersionsRequestPayload()
def test_write_with_two_protocol_versions(self):
payload = discover_versions.DiscoverVersionsRequestPayload(
self.protocol_versions_two)
expected = self.encoding_two
self._test_write(payload, expected)
def test_write_with_empty_protocol_list(self):
payload = discover_versions.DiscoverVersionsRequestPayload(
self.protocol_versions_empty)
expected = self.encoding_empty
self._test_write(payload, expected)
def _create_discover_versions_payload(self):
return discover_versions.DiscoverVersionsRequestPayload()
