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