6
6
from budget .views import BudgetViewSet
7
7
from rest_framework .response import Response
8
8
9
+
9
10
class BudgetViewSetTests (TestCase ):
10
11
def setUp (self ):
11
12
self .client = APIClient ()
@@ -18,29 +19,28 @@ def test_create(self, mock_login_user, mock_create):
18
19
"AuthenticationResult" : {"AccessToken" : "mock_access_token" , "IdToken" : "mock_id_token" }
19
20
}
20
21
mock_create .return_value = Response (status = status .HTTP_201_CREATED , data = {"amount" : 100 })
21
-
22
+
22
23
request = self .client .post ("/budget/" , {"amount" : 100 })
23
24
request .headers ["Authorization" ] = "Bearer mock_access_token"
24
-
25
+
25
26
response = self .client .post ("/budget/" , {"amount" : 100 })
26
-
27
+
27
28
self .assertEqual (response .status_code , status .HTTP_201_CREATED )
28
29
self .assertEqual (response .data ["amount" ], 100 )
29
30
30
-
31
31
@patch .object (BudgetViewSet , "list" )
32
32
@patch .object (CognitoService , "login_user" )
33
33
def test_list (self , mock_login_user , mock_list ):
34
34
mock_login_user .return_value = {
35
35
"AuthenticationResult" : {"AccessToken" : "mock_access_token" , "IdToken" : "mock_id_token" }
36
36
}
37
37
mock_list .return_value = Response (status = status .HTTP_200_OK , data = {"amount" : 100 })
38
-
38
+
39
39
request = self .client .get ("/budget/" )
40
40
request .headers ["Authorization" ] = "Bearer mock_access_token"
41
-
41
+
42
42
response = self .client .get ("/budget/" )
43
-
43
+
44
44
self .assertEqual (response .status_code , status .HTTP_200_OK )
45
45
self .assertEqual (response .data ["amount" ], 100 )
46
46
@@ -51,12 +51,12 @@ def test_destroy(self, mock_login_user, mock_destroy):
51
51
"AuthenticationResult" : {"AccessToken" : "mock_access_token" , "IdToken" : "mock_id_token" }
52
52
}
53
53
mock_destroy .return_value = Response (status = status .HTTP_204_NO_CONTENT )
54
-
54
+
55
55
request = self .client .delete ("/budget/" )
56
56
request .headers ["Authorization" ] = "Bearer mock_access_token"
57
-
57
+
58
58
response = self .client .delete ("/budget/" )
59
-
59
+
60
60
self .assertEqual (response .status_code , status .HTTP_204_NO_CONTENT )
61
61
62
62
@patch .object (BudgetViewSet , "partial_update" )
@@ -66,11 +66,11 @@ def test_partial_update(self, mock_login_user, mock_partial_update):
66
66
"AuthenticationResult" : {"AccessToken" : "mock_access_token" , "IdToken" : "mock_id_token" }
67
67
}
68
68
mock_partial_update .return_value = Response (status = status .HTTP_200_OK , data = {"amount" : 200 })
69
-
69
+
70
70
request = self .client .put ("/budget/" , {"amount" : 200 })
71
71
request .headers ["Authorization" ] = "Bearer mock_access_token"
72
-
72
+
73
73
response = self .client .put ("/budget/" , {"amount" : 200 })
74
-
74
+
75
75
self .assertEqual (response .status_code , status .HTTP_200_OK )
76
76
self .assertEqual (response .data ["amount" ], 200 )
0 commit comments