def test_get_existing_by_multi_field(
ndb_stub: datastore_stub.LocalDatastoreStub, ) -> None:
model = SimpleModel(id="test", str_prop="asdf", int_prop=42)
ndb_stub._insert_model(model)
query_res = SimpleModel.query(SimpleModel.str_prop == "asdf",
SimpleModel.int_prop == 42).get()
assert query_res == model
def test_get_existing_by_field_not_found(
ndb_stub: datastore_stub.LocalDatastoreStub, ) -> None:
model = SimpleModel(
id="test",
str_prop="asdf",
)
ndb_stub._insert_model(model)
query_res = SimpleModel.query(SimpleModel.str_prop == "foo").get()
assert query_res is None
def test_count_existing_by_field(
ndb_stub: datastore_stub.LocalDatastoreStub) -> None:
model = SimpleModel(
id="test",
str_prop="asdf",
)
ndb_stub._insert_model(model)
count = SimpleModel.query(SimpleModel.str_prop == "asdf").count()
assert count == 1
def test_get_existing_by_id(
ndb_stub: datastore_stub.LocalDatastoreStub) -> None:
model = SimpleModel(
id="test",
str_prop="asdf",
)
ndb_stub._insert_model(model)
model_res = SimpleModel.get_by_id("test")
assert model_res == model
def test_delete_model(ndb_stub: datastore_stub.LocalDatastoreStub) -> None:
model = SimpleModel(
id="test",
str_prop="asdf",
)
ndb_stub._insert_model(model)
key = ndb.Key(SimpleModel, "test")
key.delete()
assert SimpleModel.get_by_id("test") is None
def test_fetch_existing_by_field_multiple_with_order(
ndb_stub: datastore_stub.LocalDatastoreStub, ) -> None:
model1 = SimpleModel(id="test", str_prop="asdf", int_prop=10)
model2 = SimpleModel(id="test2", str_prop="asdf", int_prop=20)
ndb_stub._insert_model(model1)
ndb_stub._insert_model(model2)
# We don't pass an ordering here, so the order is not deterministic
query_res = (SimpleModel.query(SimpleModel.str_prop == "asdf").order(
SimpleModel.int_prop).fetch(limit=2))
assert len(query_res) == 2
assert query_res == [model1, model2]
def test_fetch_existing_by_le(
ndb_stub: datastore_stub.LocalDatastoreStub) -> None:
model1 = SimpleModel(
id="test",
int_prop=42,
)
model2 = SimpleModel(
id="test2",
int_prop=43,
)
ndb_stub._insert_model(model1)
ndb_stub._insert_model(model2)
query_res = SimpleModel.query(SimpleModel.int_prop <= 43).fetch(limit=5)
assert len(query_res) == 2
def test_fetch_existing_by_field_with_limit(
ndb_stub: datastore_stub.LocalDatastoreStub, ) -> None:
model1 = SimpleModel(
id="test",
str_prop="asdf",
)
model2 = SimpleModel(
id="test2",
str_prop="asdf",
)
ndb_stub._insert_model(model1)
ndb_stub._insert_model(model2)
# Since we don't pass an order by, it's non deterministic which we get
query_res = SimpleModel.query(SimpleModel.str_prop == "asdf").fetch(
limit=1)
assert len(query_res) == 1
assert query_res[0].str_prop == "asdf"
def test_fetch_existing_by_field_multiple(
ndb_stub: datastore_stub.LocalDatastoreStub, ) -> None:
model1 = SimpleModel(
id="test",
str_prop="asdf",
)
model2 = SimpleModel(
id="test2",
str_prop="asdf",
)
ndb_stub._insert_model(model1)
ndb_stub._insert_model(model2)
# We don't pass an ordering here, so the order is not deterministic
query_res = SimpleModel.query(SimpleModel.str_prop == "asdf").fetch(
limit=2)
assert len(query_res) == 2
query_res.sort(key=lambda m: m.key.id())
assert query_res == [model1, model2]