def testEmbeddedClientEntityWithoutKey(self):
client_entity = entity.Entity(key.Key('foo', project='bar'))
entity_without_key = entity.Entity()
entity_without_key['test'] = True
client_entity['embedded'] = entity_without_key
e = Entity.from_client_entity(client_entity)
self.assertIsInstance(e.properties['embedded'], dict)
def make_join(kind, order, type, field):
first_join = ds_entity.Entity()
first_join['kind'] = kind
first_join['field'] = field
first_join['order'] = order
first_join['type'] = type
return first_join
def test_datastore_api_get_version(self):
"""Test the get_version API method for the Datastore API."""
test_response = entity.Entity(
key=key.Key('Config', 'datastore_version', project='TEST_PROJECT'))
test_response['integer_value'] = 1
mock_client = mock.Mock()
mock_client.get.return_value = test_response
test_datastore_api = datastore.DatastoreAPI(self.config, mock_client)
self.assertEqual(1, test_datastore_api.get_version())
def to_client_entity(self):
"""
Returns a :class:`google.cloud.datastore.entity.Entity` instance that
represents this entity.
"""
key = self.key.to_client_key()
res = entity.Entity(key=key,
exclude_from_indexes=tuple(
self.exclude_from_indexes))
res.update(self.properties)
return res
def to_client_entity(self):
"""
Returns a :class:`google.cloud.datastore.entity.Entity` instance that
represents this entity.
"""
res = entity.Entity(key=self.key.to_client_key(),
exclude_from_indexes=tuple(self.exclude_from_indexes))
for name, value in self.properties.items():
if isinstance(value, Key):
if not value.project:
value.project = self.key.project
value = value.to_client_key()
res[name] = value
return res
def make_step(kind,
order,
where,
ref_time_type=None,
ref_time_value=None,
duration=None):
first_step = ds_entity.Entity()
first_step['kind'] = kind
first_step['order'] = order
first_step['where'] = where
if ref_time_type and ref_time_value:
first_step['referenceTime'] = make_reference_time(
ref_time_type, ref_time_value)
if duration:
first_step['duration'] = duration
return first_step
def MockEntity(*path):
key = ds_key_module.Key(*path, project="testing")
return entity.Entity(key=key)
def make_reference_time(rt_type, value):
entity = ds_entity.Entity()
entity['type'] = rt_type
entity['value'] = value
return entity
def make_threshold(operator, value):
threshold = ds_entity.Entity()
threshold['operator'] = operator
threshold['value'] = value
return threshold
def make_query(qname):
two_step_query = ds_entity.Entity()
two_step_query['name'] = qname
return two_step_query
def make_query(qname):
key = ds_key.Key('EntityKind', 1234, project='project')
two_step_query = ds_entity.Entity(key=key)
two_step_query['name'] = qname
return two_step_query
