def property_filters() -> List[PropertyFilter]: return [ PropertyFilter(prop='prop1', operator=PropertyFilterOperator.LESS_THAN, value=Value('value1')), PropertyFilter(prop='prop2', operator=PropertyFilterOperator.GREATER_THAN, value=Value(1234)) ]
async def test_query(creds: str, kind: str, project: str) -> None: async with Session() as s: ds = Datastore(project=project, service_file=creds, session=s) property_filter = PropertyFilter(prop='value', operator=PropertyFilterOperator.EQUAL, value=Value(42)) query = Query(kind=kind, query_filter=Filter(property_filter)) before = await ds.runQuery(query, session=s) num_results = len(before.entity_results) transaction = await ds.beginTransaction(session=s) mutations = [ ds.make_mutation(Operation.INSERT, Key(project, [PathElement(kind)]), properties={'value': 42}), ds.make_mutation(Operation.INSERT, Key(project, [PathElement(kind)]), properties={'value': 42}), ] await ds.commit(mutations, transaction=transaction, session=s) after = await ds.runQuery(query, session=s) assert len(after.entity_results) == num_results + 2
def test_make_mutation_from_value_object(key): value = Value(30, exclude_from_indexes=True) properties = {'value': value} results = Datastore.make_mutation(Operation.INSERT, key, properties) assert results['insert']['properties']['value'] == value.to_repr()
def test_to_repr(v, expected_json_key): value = Value(v) r = value.to_repr() assert len(r) == 2 # Value + excludeFromIndexes assert r['excludeFromIndexes'] is False assert r[str(expected_json_key)] == v
def test_to_repr_with_datetime_value(): dt = datetime(year=2018, month=7, day=15, hour=11, minute=22, second=33, microsecond=456789) value = Value(dt) r = value.to_repr() assert r['timestampValue'] == '2018-07-15T11:22:33.456789000Z'
async def test_insert_value_object(creds: str, kind: str, project: str) -> None: key = Key(project, [PathElement(kind)]) async with Session() as s: ds = Datastore(project=project, service_file=creds, session=s) properties = {'value': Value(30, exclude_from_indexes=True)} insert_result = await ds.insert(key, properties) assert len(insert_result['mutationResults']) == 1
def test_to_repr_non_supported_type(): class NonSupportedType: pass value = Value(NonSupportedType()) with pytest.raises(Exception) as ex_info: value.to_repr() assert NonSupportedType.__name__ in ex_info.value.args[0]
async def test_query_with_key_projection(creds: str, kind: str, project: str) -> None: async with Session() as s: ds = Datastore(project=project, service_file=creds, session=s) # setup test data await ds.insert(Key(project, [PathElement(kind)]), {'value': 30}, s) property_filter = PropertyFilter( prop='value', operator=PropertyFilterOperator.EQUAL, value=Value(30)) projection = [Projection.from_repr({'property': {'name': '__key__'}})] query = Query(kind=kind, query_filter=Filter(property_filter), limit=1, projection=projection) result = await ds.runQuery(query, session=s) assert result.entity_results[0].entity.properties == {} assert result.entity_result_type.value == 'KEY_ONLY' # clean up test data await ds.delete(result.entity_results[0].entity.key, s)
def query_filter() -> Filter: inner_filter = PropertyFilter( prop='property_name', operator=PropertyFilterOperator.EQUAL, value=Value(123)) return Filter(inner_filter)
def value() -> Value: return Value(value='foobar', exclude_from_indexes=False)
def test_to_repr_exclude_from_indexes(): value = Value(123, exclude_from_indexes=True) r = value.to_repr() assert r['excludeFromIndexes']
def test_to_repr_with_geo_point_value(lat_lng): value = Value(lat_lng) r = value.to_repr() assert r['geoPointValue'] == lat_lng.to_repr()
def test_to_repr_with_key_value(key): value = Value(key) r = value.to_repr() assert r['keyValue'] == key.to_repr()
def test_to_repr_with_null_value(): value = Value(None) r = value.to_repr() assert r['nullValue'] == 'NULL_VALUE'
def test_to_repr_with_blob_value(): value = Value(b'foobar') r = value.to_repr() assert r['blobValue'] == 'Zm9vYmFy'
def value() -> Value: return Value('value')