def test_key(self):
from gcloud.datastore.key import Key
key = Key('PATH', 1234, dataset_id='DATASET')
name, value = self._callFUT(key)
self.assertEqual(name, 'key_value')
self.assertEqual(value, key.to_protobuf())
def test_key(self):
from gcloud.datastore.key import Key
key = Key('PATH', 1234, dataset_id='DATASET')
name, value = self._callFUT(key)
self.assertEqual(name, 'key_value')
self.assertEqual(value, key.to_protobuf())
def test_key(self):
from gcloud.datastore.key import Key
pb = self._makePB()
key = Key('KIND', 1234, dataset_id='DATASET')
self._callFUT(pb, key)
value = pb.key_value
self.assertEqual(value, key.to_protobuf())
def test_key(self):
from gcloud.datastore.key import Key
pb = self._makePB()
key = Key('KIND', 1234, dataset_id='DATASET')
self._callFUT(pb, key)
value = pb.key_value
self.assertEqual(value, key.to_protobuf())
def test_entity_w_key(self):
from gcloud.datastore.entity import Entity
from gcloud.datastore.key import Key
pb = self._makePB()
key = Key('KIND', 123, dataset_id='DATASET')
entity = Entity(key=key)
entity['foo'] = u'Foo'
self._callFUT(pb, entity)
value = pb.entity_value
self.assertEqual(value.key, key.to_protobuf())
props = list(value.property)
self.assertEqual(len(props), 1)
self.assertEqual(props[0].name, 'foo')
self.assertEqual(props[0].value.string_value, u'Foo')
def test_entity_w_key(self):
from gcloud.datastore.entity import Entity
from gcloud.datastore.key import Key
pb = self._makePB()
key = Key('KIND', 123, dataset_id='DATASET')
entity = Entity(key=key)
entity['foo'] = u'Foo'
self._callFUT(pb, entity)
value = pb.entity_value
self.assertEqual(value.key, key.to_protobuf())
props = list(value.property)
self.assertEqual(len(props), 1)
self.assertEqual(props[0].name, 'foo')
self.assertEqual(props[0].value.string_value, u'Foo')
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 _make_key(kind, id, dataset_id):
from gcloud.datastore._datastore_v1_pb2 import Key
key = Key()
key.partition_id.dataset_id = dataset_id
elem = key.path_element.add()
elem.kind = kind
elem.id = id
return key
def _makePB(self, dataset_id=None, namespace=None, path=()):
from gcloud.datastore._datastore_v1_pb2 import Key
pb = Key()
if dataset_id is not None:
pb.partition_id.dataset_id = dataset_id
if namespace is not None:
pb.partition_id.namespace = namespace
for elem in path:
added = pb.path_element.add()
added.kind = elem['kind']
if 'id' in elem:
added.id = elem['id']
if 'name' in elem:
added.name = elem['name']
return pb
def __init__(self):
from gcloud.datastore.key import Key
self.key = Key('KIND', dataset_id='DATASET')