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_delete_company_complicated_expression(self): """ Delete should support more complicated criterion with the `fetch` synchronization strategy enabled. """ with transaction(): company = Company(name="name").create() Company(name="other-name").create() with transaction(): self.company_store._delete(Company.name.in_(["name"]), synchronize_session="fetch") assert_that( calling(Company.retrieve).with_args(company.id), raises(ModelNotFoundError, pattern="Company not found"), ) assert_that(Company.count(), is_(equal_to(1)))