def setUp(self):
self.region = Region(name="region")
self.region.save()
self.company = Company(code="1001", name="company", region=self.region)
self.company.save()
self.store = Store(code="10011001", name="store", company=self.company)
self.store.save()
self.emp_region = Employee(username="******", organization=self.region)
self.emp_region.save()
self.emp_company = Employee(username="******", organization=self.company)
self.emp_company.save()
self.emp_store = Employee(username="******", organization=self.store)
self.emp_store.save()
self.empid_not_exists = 31415926
package = "com.tiantian.ttclock"
appid = 10000
d = date(year=2013, month=12, day=1)
self.log1 = LogMeta(date=d, uid=self.emp_region.pk, appID=appid, appPkg=package)
self.log1.save()
self.log2 = LogMeta(date=d, uid=self.emp_company.pk, appID=appid, appPkg=package)
self.log2.save()
self.log3 = LogMeta(date=d, uid=self.emp_store.pk, appID=appid, appPkg=package)
self.log3.save()
self.logs = LogMeta.objects.all()
class TestUserFilter(TestCase):
def setUp(self):
self.region = Region(name="region")
self.region.save()
self.company = Company(code="1001", name="company", region=self.region)
self.company.save()
self.store = Store(code="10011001", name="store", company=self.company)
self.store.save()
self.emp_region = Employee(username="******", organization=self.region)
self.emp_region.save()
self.emp_company = Employee(username="******", organization=self.company)
self.emp_company.save()
self.emp_store = Employee(username="******", organization=self.store)
self.emp_store.save()
self.empid_not_exists = 31415926
package = "com.tiantian.ttclock"
appid = 10000
d = date(year=2013, month=12, day=1)
self.log1 = LogMeta(date=d, uid=self.emp_region.pk, appID=appid, appPkg=package)
self.log1.save()
self.log2 = LogMeta(date=d, uid=self.emp_company.pk, appID=appid, appPkg=package)
self.log2.save()
self.log3 = LogMeta(date=d, uid=self.emp_store.pk, appID=appid, appPkg=package)
self.log3.save()
self.logs = LogMeta.objects.all()
def test_admin_filter(self):
filter = AdminFilter(self.logs, None, self.company.pk, None, None)
logs = filter.filter()
self.assertItemsEqual([self.log2, self.log3], logs)
filter = AdminFilter(self.logs, None, None, None, self.emp_store.pk)
logs = filter.filter()
self.assertItemsEqual([self.log3], logs)
filter = AdminFilter(self.logs, None, None, None, None)
logs = filter.filter()
self.assertItemsEqual(logs, logs)
def test_user_permitted_filter(self):
filter = UserPermittedFilter(self.emp_company, self.logs, None, None, None, self.emp_company.pk)
logs = filter.filter()
self.assertItemsEqual([self.log2], logs)
filter = UserPermittedFilter(self.emp_company, self.logs, None, None, None, self.emp_region.pk)
logs = filter.filter()
self.assertTrue(len(logs) == 0)
filter = UserPermittedFilter(self.emp_company, self.logs, None, None, self.store.pk, None)
logs = filter.filter()
self.assertItemsEqual([self.log3], logs)
filter = UserPermittedFilter(self.emp_company, self.logs, self.emp_region.pk, None, None, None)
logs = filter.filter()
self.assertTrue(len(logs) == 0)
filter = UserPermittedFilter(self.emp_company, self.logs, None, None, None, None)
logs = filter.filter()
self.assertItemsEqual([self.log2, self.log3], logs)
def test_user_unpermitted_filter(self):
filter = UserUnpermittedFilter(self.logs, self.emp_store.pk)
logs = filter.filter()
self.assertItemsEqual([self.log3], logs)
