class TestDashboardSite(TestCase): def setUp(self): self.requests = map(Request, ('GET', 'POST', 'PUT', 'PATCH', 'DELETE')) self.ma = ReimbursementModelAdmin(Reimbursement, 'dashboard') def test_has_add_permission(self): permissions = map(self.ma.has_add_permission, self.requests) self.assertNotIn(True, tuple(permissions)) def test_has_change_permission(self): permissions = map(self.ma.has_change_permission, self.requests) expected = (True, False, False, False, False) self.assertEqual(expected, tuple(permissions)) def test_has_delete_permission(self): permissions = map(self.ma.has_delete_permission, self.requests) self.assertNotIn(True, tuple(permissions)) def test_format_document_with_cnpj(self): obj = ReimbursementMock('12345678901234') self.assertEqual('12.345.678/9012-34', self.ma._format_document(obj)) def test_format_document_with_cpf(self): obj = ReimbursementMock('12345678901') self.assertEqual('123.456.789-01', self.ma._format_document(obj)) def test_format_document_with_unknown(self): obj = ReimbursementMock('2345678') self.assertEqual('2345678', self.ma._format_document(obj))
def setUp(self): self.requests = map(Request, ('GET', 'POST', 'PUT', 'PATCH', 'DELETE')) self.ma = ReimbursementModelAdmin(Reimbursement, 'dashboard')