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)
