def test_streaming_read(self):
# Setup Expected Response
chunked_value = True
resume_token = b'103'
expected_response = {
'chunked_value': chunked_value,
'resume_token': resume_token
}
expected_response = result_set_pb2.PartialResultSet(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[iter([expected_response])])
client = spanner_v1.SpannerClient(channel=channel)
# Setup Request
session = client.session_path('[PROJECT]', '[INSTANCE]', '[DATABASE]',
'[SESSION]')
table = 'table110115790'
columns = []
key_set = {}
response = client.streaming_read(session, table, columns, key_set)
resources = list(response)
assert len(resources) == 1
assert expected_response == resources[0]
assert len(channel.requests) == 1
expected_request = spanner_pb2.ReadRequest(session=session,
table=table,
columns=columns,
key_set=key_set)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_execute_streaming_sql(self):
# Setup Expected Response
chunked_value = True
resume_token = b'103'
expected_response = {
'chunked_value': chunked_value,
'resume_token': resume_token
}
expected_response = result_set_pb2.PartialResultSet(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[iter([expected_response])])
client = spanner_v1.SpannerClient(channel=channel)
# Setup Request
session = client.session_path('[PROJECT]', '[INSTANCE]', '[DATABASE]',
'[SESSION]')
sql = 'sql114126'
response = client.execute_streaming_sql(session, sql)
resources = list(response)
assert len(resources) == 1
assert expected_response == resources[0]
assert len(channel.requests) == 1
expected_request = spanner_pb2.ExecuteSqlRequest(session=session,
sql=sql)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_execute_streaming_sql(self):
# Setup Expected Response
chunked_value = True
resume_token = b"103"
expected_response = {
"chunked_value": chunked_value,
"resume_token": resume_token,
}
expected_response = result_set_pb2.PartialResultSet(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[iter([expected_response])])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = spanner_v1.SpannerClient()
# Setup Request
session = client.session_path("[PROJECT]", "[INSTANCE]", "[DATABASE]",
"[SESSION]")
sql = "sql114126"
response = client.execute_streaming_sql(session, sql)
resources = list(response)
assert len(resources) == 1
assert expected_response == resources[0]
assert len(channel.requests) == 1
expected_request = spanner_pb2.ExecuteSqlRequest(session=session,
sql=sql)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_streaming_read(self, mock_create_stub):
# Mock gRPC layer
grpc_stub = mock.Mock()
mock_create_stub.return_value = grpc_stub
client = spanner_v1.SpannerClient()
# Mock request
session = client.session_path('[PROJECT]', '[INSTANCE]', '[DATABASE]',
'[SESSION]')
table = 'table110115790'
columns = []
key_set = {}
# Mock response
chunked_value = True
resume_token = b'103'
expected_response = {
'chunked_value': chunked_value,
'resume_token': resume_token
}
expected_response = result_set_pb2.PartialResultSet(
**expected_response)
grpc_stub.StreamingRead.return_value = iter([expected_response])
response = client.streaming_read(session, table, columns, key_set)
resources = list(response)
self.assertEqual(1, len(resources))
self.assertEqual(expected_response, resources[0])
grpc_stub.StreamingRead.assert_called_once()
args, kwargs = grpc_stub.StreamingRead.call_args
self.assertEqual(len(args), 2)
self.assertEqual(len(kwargs), 1)
self.assertIn('metadata', kwargs)
actual_request = args[0]
expected_request = spanner_pb2.ReadRequest(session=session,
table=table,
columns=columns,
key_set=key_set)
self.assertEqual(expected_request, actual_request)
def test_execute_streaming_sql(self, mock_create_stub):
# Mock gRPC layer
grpc_stub = mock.Mock()
mock_create_stub.return_value = grpc_stub
client = spanner_v1.SpannerClient()
# Mock request
session = client.session_path('[PROJECT]', '[INSTANCE]', '[DATABASE]',
'[SESSION]')
sql = 'sql114126'
# Mock response
chunked_value = True
resume_token = b'103'
expected_response = {
'chunked_value': chunked_value,
'resume_token': resume_token
}
expected_response = result_set_pb2.PartialResultSet(
**expected_response)
grpc_stub.ExecuteStreamingSql.return_value = iter([expected_response])
response = client.execute_streaming_sql(session, sql)
resources = list(response)
self.assertEqual(1, len(resources))
self.assertEqual(expected_response, resources[0])
grpc_stub.ExecuteStreamingSql.assert_called_once()
args, kwargs = grpc_stub.ExecuteStreamingSql.call_args
self.assertEqual(len(args), 2)
self.assertEqual(len(kwargs), 1)
self.assertIn('metadata', kwargs)
actual_request = args[0]
expected_request = spanner_pb2.ExecuteSqlRequest(session=session,
sql=sql)
self.assertEqual(expected_request, actual_request)
def test_streaming_read(self):
# Setup Expected Response
chunked_value = True
resume_token = b"103"
expected_response = {
"chunked_value": chunked_value,
"resume_token": resume_token,
}
expected_response = result_set_pb2.PartialResultSet(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[iter([expected_response])])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = spanner_v1.SpannerClient()
# Setup Request
session = client.session_path("[PROJECT]", "[INSTANCE]", "[DATABASE]",
"[SESSION]")
table = "table110115790"
columns = []
key_set = {}
response = client.streaming_read(session, table, columns, key_set)
resources = list(response)
assert len(resources) == 1
assert expected_response == resources[0]
assert len(channel.requests) == 1
expected_request = spanner_pb2.ReadRequest(session=session,
table=table,
columns=columns,
key_set=key_set)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
