Revised the test_delete test in the AccountTestCase test case.

This commit is contained in:
依瑪貓 2023-03-22 00:33:25 +08:00
parent 623313b58a
commit b60cc7902d

View File

@ -643,8 +643,8 @@ class AccountTestCase(unittest.TestCase):
:return: None. :return: None.
""" """
from accounting.models import Account from accounting.models import Account
detail_uri: str = f"{PREFIX}/{BANK.code}" detail_uri: str = f"{PREFIX}/{PETTY.code}"
delete_uri: str = f"{PREFIX}/{BANK.code}/delete" delete_uri: str = f"{PREFIX}/{PETTY.code}/delete"
list_uri: str = PREFIX list_uri: str = PREFIX
response: httpx.Response response: httpx.Response
@ -653,8 +653,7 @@ class AccountTestCase(unittest.TestCase):
"base_code": PETTY.base_code, "base_code": PETTY.base_code,
"title": PETTY.title}) "title": PETTY.title})
self.assertEqual(response.status_code, 302) self.assertEqual(response.status_code, 302)
self.assertEqual(response.headers["Location"], self.assertEqual(response.headers["Location"], detail_uri)
f"{PREFIX}/{PETTY.code}")
response = self.client.post("/accounting/currencies/store", response = self.client.post("/accounting/currencies/store",
data={"csrf_token": self.csrf_token, data={"csrf_token": self.csrf_token,
@ -669,7 +668,7 @@ class AccountTestCase(unittest.TestCase):
"next": NEXT_URI, "next": NEXT_URI,
"date": date.today().isoformat(), "date": date.today().isoformat(),
"currency-1-code": "USD", "currency-1-code": "USD",
"currency-1-credit-1-account_code": PETTY.code, "currency-1-credit-1-account_code": BANK.code,
"currency-1-credit-1-amount": "20"}) "currency-1-credit-1-amount": "20"})
with self.app.app_context(): with self.app.app_context():
@ -680,15 +679,13 @@ class AccountTestCase(unittest.TestCase):
response = self.client.post(f"{PREFIX}/{CASH.code}/delete", response = self.client.post(f"{PREFIX}/{CASH.code}/delete",
data={"csrf_token": self.csrf_token}) data={"csrf_token": self.csrf_token})
self.assertEqual(response.status_code, 302) self.assertEqual(response.status_code, 302)
self.assertEqual(response.headers["Location"], self.assertEqual(response.headers["Location"], f"{PREFIX}/{CASH.code}")
f"{PREFIX}/{CASH.code}")
# Cannot delete the account that is in use # Cannot delete the account that is in use
response = self.client.post(f"{PREFIX}/{PETTY.code}/delete", response = self.client.post(f"{PREFIX}/{BANK.code}/delete",
data={"csrf_token": self.csrf_token}) data={"csrf_token": self.csrf_token})
self.assertEqual(response.status_code, 302) self.assertEqual(response.status_code, 302)
self.assertEqual(response.headers["Location"], self.assertEqual(response.headers["Location"], f"{PREFIX}/{BANK.code}")
f"{PREFIX}/{PETTY.code}")
# Success # Success
response = self.client.get(detail_uri) response = self.client.get(detail_uri)
@ -700,7 +697,7 @@ class AccountTestCase(unittest.TestCase):
with self.app.app_context(): with self.app.app_context():
self.assertEqual({x.code for x in Account.query.all()}, self.assertEqual({x.code for x in Account.query.all()},
{CASH.code, PETTY.code}) {CASH.code, BANK.code})
response = self.client.get(detail_uri) response = self.client.get(detail_uri)
self.assertEqual(response.status_code, 404) self.assertEqual(response.status_code, 404)