imacat/mia-accounting
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 41 Wiki Activity

Revised the AccountTestCase test case for simplicity.

Browse Source
This commit is contained in:
依瑪貓 2023-02-02 00:16:57 +08:00
parent 8364025668
commit d9624c7be6
1 changed files with 41 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
tests/test_account.py
View File

@ -109,24 +109,21 @@ class AccountTestCase(unittest.TestCase):
Account.query.delete() Account.query.delete()
db.session.commit() db.session.commit()
self.viewer: UserClient = get_user_client(self, self.app, "viewer") editor: UserClient = get_user_client(self, self.app, "editor")
self.editor: UserClient = get_user_client(self, self.app, "editor") self.client: httpx.Client = editor.client
self.nobody: UserClient = get_user_client(self, self.app, "nobody") self.csrf_token: str = editor.csrf_token
client: httpx.Client = self.editor.client
csrf_token: str = self.editor.csrf_token
response: httpx.Response response: httpx.Response
response = client.post("/accounting/accounts/store", response = self.client.post("/accounting/accounts/store",
data={"csrf_token": csrf_token, data={"csrf_token": self.csrf_token,
"base_code": "1111", "base_code": "1111",
"title": "1111 title"}) "title": "1111 title"})
self.assertEqual(response.status_code, 302) self.assertEqual(response.status_code, 302)
self.assertEqual(response.headers["Location"], self.assertEqual(response.headers["Location"],
"/accounting/accounts/1111-001") "/accounting/accounts/1111-001")
response = client.post("/accounting/accounts/store", response = self.client.post("/accounting/accounts/store",
data={"csrf_token": csrf_token, data={"csrf_token": self.csrf_token,
"base_code": "1112", "base_code": "1112",
"title": "1112 title"}) "title": "1112 title"})
self.assertEqual(response.status_code, 302) self.assertEqual(response.status_code, 302)
@ -139,35 +136,34 @@ class AccountTestCase(unittest.TestCase):
:return: None. :return: None.
""" """
response: httpx.Response response: httpx.Response
client: httpx.Client = self.nobody.client nobody: UserClient = get_user_client(self, self.app, "nobody")
csrf_token: str = self.nobody.csrf_token
response = client.get("/accounting/accounts") response = nobody.client.get("/accounting/accounts")
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.get("/accounting/accounts/1111-001") response = nobody.client.get("/accounting/accounts/1111-001")
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.get("/accounting/accounts/create") response = nobody.client.get("/accounting/accounts/create")
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.post("/accounting/accounts/store", response = nobody.client.post("/accounting/accounts/store",
data={"csrf_token": csrf_token, data={"csrf_token": nobody.csrf_token,
"base_code": "1113", "base_code": "1113",
"title": "1113 title"}) "title": "1113 title"})
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.get("/accounting/accounts/1111-001/edit") response = nobody.client.get("/accounting/accounts/1111-001/edit")
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.post("/accounting/accounts/1111-001/update", response = nobody.client.post("/accounting/accounts/1111-001/update",
data={"csrf_token": csrf_token, data={"csrf_token": nobody.csrf_token,
"base_code": "1111", "base_code": "1111",
"title": "1111 title #2"}) "title": "1111 title #2"})
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.post("/accounting/accounts/1111-001/delete", response = nobody.client.post("/accounting/accounts/1111-001/delete",
data={"csrf_token": csrf_token}) data={"csrf_token": nobody.csrf_token})
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
def test_viewer(self) -> None: def test_viewer(self) -> None:
@ -176,33 +172,32 @@ class AccountTestCase(unittest.TestCase):
:return: None. :return: None.
""" """
response: httpx.Response response: httpx.Response
client: httpx.Client = self.viewer.client viewer: UserClient = get_user_client(self, self.app, "viewer")
csrf_token: str = self.viewer.csrf_token
response = client.get("/accounting/accounts") response = viewer.client.get("/accounting/accounts")
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
response = client.get("/accounting/accounts/1111-001") response = viewer.client.get("/accounting/accounts/1111-001")
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
response = client.get("/accounting/accounts/create") response = viewer.client.get("/accounting/accounts/create")
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.post("/accounting/accounts/store", response = viewer.client.post("/accounting/accounts/store",
data={"csrf_token": csrf_token, data={"csrf_token": viewer.csrf_token,
"base_code": "1113", "base_code": "1113",
"title": "1113 title"}) "title": "1113 title"})
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.get("/accounting/accounts/1111-001/edit") response = viewer.client.get("/accounting/accounts/1111-001/edit")
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.post("/accounting/accounts/1111-001/update", response = viewer.client.post("/accounting/accounts/1111-001/update",
data={"csrf_token": csrf_token, data={"csrf_token": viewer.csrf_token,
"base_code": "1111", "base_code": "1111",
"title": "1111 title #2"}) "title": "1111 title #2"})
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)
response = client.post("/accounting/accounts/1111-001/delete", response = viewer.client.post("/accounting/accounts/1111-001/delete",
data={"csrf_token": csrf_token}) data={"csrf_token": viewer.csrf_token})
self.assertEqual(response.status_code, 403) self.assertEqual(response.status_code, 403)