def test_query_companies(self): companies = query_companies(self.admin, "") self.assertTrue(len(companies) == 0) companies = query_companies(self.emp_no_perm, self.region1.pk) self.assertTrue(len(companies) == 0) companies = query_companies(self.admin, self.region1.pk) self.assertItemsEqual(companies, [self.company]) companies = query_companies(self.emp_region1_1, self.region1.pk) self.assertItemsEqual(companies, [self.company]) companies = query_companies(self.emp_store, self.region1.pk) self.assertTrue(len(companies) == 0)
def test_query_stores(self): stores = query_stores(self.admin, "") self.assertTrue(len(stores) == 0) stores = query_stores(self.admin, self.company.pk) self.assertItemsEqual(stores, [self.store]) stores = query_companies(self.emp_no_perm, self.company.pk) self.assertTrue(len(stores) == 0) stores = query_stores(self.emp_company, self.company.pk) self.assertItemsEqual(stores, [self.store]) stores = query_stores(self.emp_store, self.company.pk) self.assertTrue(len(stores) == 0)