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