Skip to content

Commit 3da3d4f

Browse files
committed
fix: black linter
1 parent 2e01245 commit 3da3d4f

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

budget/tests.py

+13-13
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from budget.views import BudgetViewSet
77
from rest_framework.response import Response
88

9+
910
class BudgetViewSetTests(TestCase):
1011
def setUp(self):
1112
self.client = APIClient()
@@ -18,29 +19,28 @@ def test_create(self, mock_login_user, mock_create):
1819
"AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
1920
}
2021
mock_create.return_value = Response(status=status.HTTP_201_CREATED, data={"amount": 100})
21-
22+
2223
request = self.client.post("/budget/", {"amount": 100})
2324
request.headers["Authorization"] = "Bearer mock_access_token"
24-
25+
2526
response = self.client.post("/budget/", {"amount": 100})
26-
27+
2728
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
2829
self.assertEqual(response.data["amount"], 100)
2930

30-
3131
@patch.object(BudgetViewSet, "list")
3232
@patch.object(CognitoService, "login_user")
3333
def test_list(self, mock_login_user, mock_list):
3434
mock_login_user.return_value = {
3535
"AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
3636
}
3737
mock_list.return_value = Response(status=status.HTTP_200_OK, data={"amount": 100})
38-
38+
3939
request = self.client.get("/budget/")
4040
request.headers["Authorization"] = "Bearer mock_access_token"
41-
41+
4242
response = self.client.get("/budget/")
43-
43+
4444
self.assertEqual(response.status_code, status.HTTP_200_OK)
4545
self.assertEqual(response.data["amount"], 100)
4646

@@ -51,12 +51,12 @@ def test_destroy(self, mock_login_user, mock_destroy):
5151
"AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
5252
}
5353
mock_destroy.return_value = Response(status=status.HTTP_204_NO_CONTENT)
54-
54+
5555
request = self.client.delete("/budget/")
5656
request.headers["Authorization"] = "Bearer mock_access_token"
57-
57+
5858
response = self.client.delete("/budget/")
59-
59+
6060
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
6161

6262
@patch.object(BudgetViewSet, "partial_update")
@@ -66,11 +66,11 @@ def test_partial_update(self, mock_login_user, mock_partial_update):
6666
"AuthenticationResult": {"AccessToken": "mock_access_token", "IdToken": "mock_id_token"}
6767
}
6868
mock_partial_update.return_value = Response(status=status.HTTP_200_OK, data={"amount": 200})
69-
69+
7070
request = self.client.put("/budget/", {"amount": 200})
7171
request.headers["Authorization"] = "Bearer mock_access_token"
72-
72+
7373
response = self.client.put("/budget/", {"amount": 200})
74-
74+
7575
self.assertEqual(response.status_code, status.HTTP_200_OK)
7676
self.assertEqual(response.data["amount"], 200)

0 commit comments

Comments
 (0)