def test_find(configure_db, create_company, create_product):
ids = ["1234555-1", "1234567-8", "2131232-4", "4124432-4"]
for company_id in ids:
create_company(company_id=company_id)
c = Company.find({"company_id": "4124432-4"})
assert c[0].company_id == "4124432-4"
assert c[0].owner.first_name == "John"
d = Company.find({"owner.first_name": "John"})
assert len(d) == 4
for p in TEST_PRODUCTS:
create_product(**p)
assert len(Product.find({})) == 4
products = Product.find({"stock": {op.GTE: 1}})
assert len(products) == 3
products = Product.find({"stock": {op.GTE: 2, op.LT: 4}})
assert len(products) == 2
products = Product.find({"product_id": {op.IN: ["a", "d", "g"]}})
assert len(products) == 2
with pytest.raises(ValueError):
Product.find({"product_id": {"<>": "a"}})
def test_truncate_collection(configure_db, create_company):
create_company(company_id="1234567-8")
create_company(company_id="1234567-9")
companies = Company.find({})
assert len(companies) == 2
Company.truncate_collection()
new_companies = Company.find({})
assert len(new_companies) == 0
def test_delete_model(configure_db, create_company):
company: Company = create_company(company_id="11223344-5",
first_name="Jane",
last_name="Doe")
_id = company.id
company.delete()
with pytest.raises(ModelNotFoundError):
Company.get_by_id(_id)
def test_get_by_id(configure_db, create_company):
c: Company = create_company(company_id="1234567-8")
assert c.id is not None
assert c.company_id == "1234567-8"
assert c.owner.last_name == "Doe"
c_2 = Company.get_by_id(c.id)
assert c_2.id == c.id
assert c_2.company_id == "1234567-8"
assert c_2.owner.first_name == "John"
def test_find_not_in(configure_db, create_company):
ids = ["1234555-1", "1234567-8", "2131232-4", "4124432-4"]
for company_id in ids:
create_company(company_id=company_id)
found = Company.find(
{"company_id": {
op.NOT_IN: [
"1234555-1",
"1234567-8",
]
}})
assert len(found) == 2
for company in found:
assert company.company_id in ("2131232-4", "4124432-4")
def test_find_one(configure_db, create_company):
with pytest.raises(ModelNotFoundError):
Company.find_one()
company_a: Company = create_company(company_id="1234555-1",
first_name="Foo")
company_b: Company = create_company(company_id="1231231-2",
first_name="Bar")
a: Company = Company.find_one({"company_id": company_a.company_id})
b: Company = Company.find_one({"company_id": company_b.company_id})
assert a.company_id == company_a.company_id
assert b.company_id == company_b.company_id
assert a.owner.first_name == "Foo"
assert b.owner.first_name == "Bar"
with pytest.raises(ModelNotFoundError):
Company.find_one({"company_id": "Foo"})
random_company = Company.find_one()
assert random_company.company_id in {a.company_id, b.company_id}
def test_get_by_empty_str_id(configure_db):
with pytest.raises(ModelNotFoundError):
Company.get_by_id("")