def test_w_listvalue(self): from google.protobuf.struct_pb2 import Value from google.cloud.spanner_v1._helpers import _make_list_value_pb list_value = _make_list_value_pb([1, 2, 3]) value_pb = self._callFUT(list_value) self.assertIsInstance(value_pb, Value) self.assertEqual(value_pb.list_value, list_value)
def _make_list_value(values=(), value_pbs=None): from google.protobuf.struct_pb2 import ListValue from google.protobuf.struct_pb2 import Value from google.cloud.spanner_v1._helpers import _make_list_value_pb if value_pbs is not None: return Value(list_value=ListValue(values=value_pbs)) return Value(list_value=_make_list_value_pb(values))
def _to_pb(self): """Construct a KeyRange protobuf. :rtype: :class:`~google.cloud.spanner_v1.KeyRange` :returns: protobuf corresponding to this instance. """ kwargs = {} if self.start_open is not None: kwargs["start_open"] = _make_list_value_pb(self.start_open) if self.start_closed is not None: kwargs["start_closed"] = _make_list_value_pb(self.start_closed) if self.end_open is not None: kwargs["end_open"] = _make_list_value_pb(self.end_open) if self.end_closed is not None: kwargs["end_closed"] = _make_list_value_pb(self.end_closed) return KeyRangePB(**kwargs)
def to_pb(self): """Construct a KeyRange protobuf. :rtype: :class:`~google.cloud.spanner_v1.proto.keys_pb2.KeyRange` :returns: protobuf corresponding to this instance. """ kwargs = {} if self.start_open: kwargs['start_open'] = _make_list_value_pb(self.start_open) if self.start_closed: kwargs['start_closed'] = _make_list_value_pb(self.start_closed) if self.end_open: kwargs['end_open'] = _make_list_value_pb(self.end_open) if self.end_closed: kwargs['end_closed'] = _make_list_value_pb(self.end_closed) return KeyRangePB(**kwargs)
def _to_pb(self): """Construct a KeyRange protobuf. :rtype: :class:`~google.cloud.spanner_v1.proto.keys_pb2.KeyRange` :returns: protobuf corresponding to this instance. """ kwargs = {} if self.start_open is not None: kwargs["start_open"] = _make_list_value_pb(self.start_open) if self.start_closed is not None: kwargs["start_closed"] = _make_list_value_pb(self.start_closed) if self.end_open is not None: kwargs["end_open"] = _make_list_value_pb(self.end_open) if self.end_closed is not None: kwargs["end_closed"] = _make_list_value_pb(self.end_closed) return KeyRangePB(**kwargs)
def test_w_struct(self): from google.protobuf.struct_pb2 import Value from google.cloud.spanner_v1.proto.type_pb2 import Type, StructType from google.cloud.spanner_v1.proto.type_pb2 import STRUCT, STRING, INT64 from google.cloud.spanner_v1._helpers import _make_list_value_pb VALUES = [u"phred", 32] struct_type_pb = StructType(fields=[ StructType.Field(name="name", type=Type(code=STRING)), StructType.Field(name="age", type=Type(code=INT64)), ]) field_type = Type(code=STRUCT, struct_type=struct_type_pb) value_pb = Value(list_value=_make_list_value_pb(VALUES)) self.assertEqual(self._callFUT(value_pb, field_type), VALUES)
def test_w_struct(self): from google.protobuf.struct_pb2 import Value from google.cloud.spanner_v1.proto.type_pb2 import Type, StructType from google.cloud.spanner_v1.proto.type_pb2 import ( STRUCT, STRING, INT64) from google.cloud.spanner_v1._helpers import _make_list_value_pb VALUES = [u'phred', 32] struct_type_pb = StructType(fields=[ StructType.Field(name='name', type=Type(code=STRING)), StructType.Field(name='age', type=Type(code=INT64)), ]) field_type = Type(code=STRUCT, struct_type=struct_type_pb) value_pb = Value(list_value=_make_list_value_pb(VALUES)) self.assertEqual(self._callFUT(value_pb, field_type), VALUES)
def _callFUT(self, *args, **kw): from google.cloud.spanner_v1._helpers import _make_list_value_pb return _make_list_value_pb(*args, **kw)