def test_list(self):
from gcloud.datastore.datastore_v1_pb2 import Value
pb = Value()
list_pb = pb.list_value
item_pb = list_pb.add()
item_pb.string_value = 'Foo'
item_pb = list_pb.add()
item_pb.string_value = 'Bar'
items = self._callFUT(pb)
self.assertEqual(items, ['Foo', 'Bar'])
def test_entity(self):
from gcloud.datastore.datastore_v1_pb2 import Value
from gcloud.datastore.entity import Entity
pb = Value()
entity_pb = pb.entity_value
prop_pb = entity_pb.property.add()
prop_pb.name = 'foo'
prop_pb.value.string_value = 'Foo'
entity = self._callFUT(pb)
self.assertTrue(isinstance(entity, Entity))
self.assertEqual(entity['foo'], 'Foo')
def test_key(self):
from gcloud.datastore.datastore_v1_pb2 import Value
from gcloud.datastore.key import Key
_DATASET = 'DATASET'
_KIND = 'KIND'
_ID = 1234
_PATH = [{'kind': _KIND, 'id': _ID}]
pb = Value()
expected = Key(dataset_id=_DATASET, path=_PATH).to_protobuf()
pb.key_value.CopyFrom(expected)
found = self._callFUT(pb)
self.assertEqual(found.to_protobuf(), expected)
def _makePB(self):
from gcloud.datastore.datastore_v1_pb2 import Value
return Value()
def test_unknown(self):
from gcloud.datastore.datastore_v1_pb2 import Value
pb = Value()
self.assertEqual(self._callFUT(pb), None)
def _makePB(self, attr_name, value):
from gcloud.datastore.datastore_v1_pb2 import Value
pb = Value()
setattr(pb, attr_name, value)
return pb