def test_key(self):
from gcloud.datastore._datastore_v1_pb2 import Value
from gcloud.datastore.key import Key
pb = Value()
expected = Key('KIND', 1234, dataset_id='DATASET').to_protobuf()
pb.key_value.CopyFrom(expected)
found = self._callFUT(pb)
self.assertEqual(found.to_protobuf(), expected)
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
entity_pb.key.path_element.add(kind='KIND')
entity_pb.key.partition_id.dataset_id = 'DATASET'
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 _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
