def test_to_pb_w_empty_list(self):
from google.protobuf.struct_pb2 import ListValue
from google.protobuf.struct_pb2 import Value
from google.cloud.spanner_v1.proto.keys_pb2 import KeyRange
key = u'key'
key_range = self._make_one(start_closed=[], end_closed=[key])
key_range_pb = key_range.to_pb()
expected = KeyRange(
start_closed=ListValue(values=[]),
end_closed=ListValue(values=[Value(string_value=key)]),
)
self.assertEqual(key_range_pb, expected)
def test_to_pb_w_start_closed_and_end_open(self):
from google.protobuf.struct_pb2 import ListValue
from google.protobuf.struct_pb2 import Value
from google.cloud.spanner_v1.proto.keys_pb2 import KeyRange
key1 = u'key_1'
key2 = u'key_2'
key_range = self._make_one(start_closed=[key1], end_open=[key2])
key_range_pb = key_range._to_pb()
expected = KeyRange(
start_closed=ListValue(values=[Value(string_value=key1)]),
end_open=ListValue(values=[Value(string_value=key2)]),
)
self.assertEqual(key_range_pb, expected)
def test_to_pb_w_only_ranges(self):
from google.cloud.spanner_v1.proto.keys_pb2 import KeySet
from google.cloud.spanner_v1.keyset import KeyRange
KEY_1 = u'KEY_1'
KEY_2 = u'KEY_2'
KEY_3 = u'KEY_3'
KEY_4 = u'KEY_4'
RANGES = [
KeyRange(start_open=KEY_1, end_closed=KEY_2),
KeyRange(start_closed=KEY_3, end_open=KEY_4),
]
keyset = self._make_one(ranges=RANGES)
result = keyset._to_pb()
self.assertIsInstance(result, KeySet)
self.assertFalse(result.all)
self.assertEqual(len(result.keys), 0)
self.assertEqual(len(result.ranges), len(RANGES))
for found, expected in zip(result.ranges, RANGES):
self.assertEqual(found, expected._to_pb())
def test_from_dict_w_ranges(self):
from google.cloud.spanner_v1.keyset import KeyRange
klass = self._get_target_class()
key_1 = u'KEY_1'
key_2 = u'KEY_2'
key_3 = u'KEY_3'
key_4 = u'KEY_4'
mapping = {
'ranges': [
{'start_open': [key_1], 'end_closed': [key_2]},
{'start_closed': [key_3], 'end_open': [key_4]},
],
}
keyset = klass._from_dict(mapping)
range_1 = KeyRange(start_open=[key_1], end_closed=[key_2])
range_2 = KeyRange(start_closed=[key_3], end_open=[key_4])
self.assertFalse(keyset.all_)
self.assertEqual(keyset.keys, [])
self.assertEqual(keyset.ranges, [range_1, range_2])
def test_from_dict_w_ranges(self):
from google.cloud.spanner_v1.keyset import KeyRange
klass = self._get_target_class()
key_1 = "KEY_1"
key_2 = "KEY_2"
key_3 = "KEY_3"
key_4 = "KEY_4"
mapping = {
"ranges": [
{"start_open": [key_1], "end_closed": [key_2]},
{"start_closed": [key_3], "end_open": [key_4]},
]
}
keyset = klass._from_dict(mapping)
range_1 = KeyRange(start_open=[key_1], end_closed=[key_2])
range_2 = KeyRange(start_closed=[key_3], end_open=[key_4])
self.assertFalse(keyset.all_)
self.assertEqual(keyset.keys, [])
self.assertEqual(keyset.ranges, [range_1, range_2])