def test_search_company(self):
"""
Should be able to search for companies.
"""
with transaction():
company = Company(
name="name",
type=CompanyType.private,
).create()
assert_that(Company.search(), contains(company))
assert_that(Company.search(name="whatever"), is_(empty()))
assert_that(Company.search(name=company.name), contains(company))
# NB: filtering is skipped if None
assert_that(Company.search(name=None), contains(company))
def test_search_company(self):
"""
Should be able to search for companies.
"""
with transaction():
company = Company(
name="name",
type=CompanyType.private,
).create()
assert_that(Company.search(), contains(company))
assert_that(Company.search(name="whatever"), is_(empty()))
assert_that(Company.search(name=company.name), contains(company))
# NB: filtering is skipped if None
assert_that(Company.search(name=None), contains(company))
def test_create_search_count_company(self):
"""
Should be able to search and count companies after creation.
"""
with transaction():
company1 = Company(name="name1").create()
company2 = Company(name="name2").create()
assert_that(Company.count(), is_(equal_to(2)))
# Pagination fields do not affect count calculations
assert_that(self.company_store.count(offset=1, limit=1), is_(equal_to(2)))
assert_that([company.id for company in Company.search()], contains_inanyorder(company1.id, company2.id))
def test_create_search_count_company(self):
"""
Should be able to search and count companies after creation.
"""
with transaction():
company1 = Company(name="name1").create()
company2 = Company(name="name2").create()
assert_that(Company.count(), is_(equal_to(2)))
# Pagination fields do not affect count calculations
assert_that(self.company_store.count(offset=1, limit=1),
is_(equal_to(2)))
assert_that([company.id for company in Company.search()],
contains_inanyorder(company1.id, company2.id))