11from faker import Faker
22from flask import json
33from flask .testing import FlaskClient
4+ from flask_restplus ._http import HTTPStatus
45from pytest_mock import MockFixture
56
67from time_tracker_api .projects .projects_model import PROJECT_TYPE
@@ -25,7 +26,7 @@ def test_create_project_should_succeed_with_valid_request(client: FlaskClient, m
2526
2627 response = client .post ("/projects" , json = valid_project_data , follow_redirects = True )
2728
28- assert 201 == response .status_code
29+ assert HTTPStatus . CREATED == response .status_code
2930 repository_create_mock .assert_called_once_with (valid_project_data )
3031
3132
@@ -40,7 +41,7 @@ def test_create_project_should_reject_bad_request(client: FlaskClient, mocker: M
4041
4142 response = client .post ("/projects" , json = invalid_project_data , follow_redirects = True )
4243
43- assert 400 == response .status_code
44+ assert HTTPStatus . BAD_REQUEST == response .status_code
4445 repository_create_mock .assert_not_called ()
4546
4647
@@ -52,24 +53,21 @@ def test_list_all_projects(client: FlaskClient, mocker: MockFixture):
5253
5354 response = client .get ("/projects" , follow_redirects = True )
5455
55- assert 200 == response .status_code
56- json_data = json .loads (response .data )
57- assert [] == json_data
56+ assert HTTPStatus .OK == response .status_code
57+ assert [] == json .loads (response .data )
5858 repository_find_all_mock .assert_called_once ()
5959
6060
6161def test_get_project_should_succeed_with_valid_id (client : FlaskClient , mocker : MockFixture ):
6262 from time_tracker_api .projects .projects_namespace import project_dao
63-
6463 valid_id = fake .random_int (1 , 9999 )
65-
6664 repository_find_mock = mocker .patch .object (project_dao .repository ,
6765 'find' ,
6866 return_value = fake_project )
6967
7068 response = client .get ("/projects/%s" % valid_id , follow_redirects = True )
7169
72- assert 200 == response .status_code
70+ assert HTTPStatus . OK == response .status_code
7371 fake_project == json .loads (response .data )
7472 repository_find_mock .assert_called_once_with (str (valid_id ))
7573
@@ -86,11 +84,12 @@ def test_get_project_should_return_not_found_with_invalid_id(client: FlaskClient
8684
8785 response = client .get ("/projects/%s" % invalid_id , follow_redirects = True )
8886
89- assert 404 == response .status_code
87+ assert HTTPStatus . NOT_FOUND == response .status_code
9088 repository_find_mock .assert_called_once_with (str (invalid_id ))
9189
9290
93- def test_get_project_should_return_422_for_invalid_id_format (client : FlaskClient , mocker : MockFixture ):
91+ def test_get_project_should_response_with_unprocessable_entity_for_invalid_id_format (client : FlaskClient ,
92+ mocker : MockFixture ):
9493 from time_tracker_api .projects .projects_namespace import project_dao
9594 from werkzeug .exceptions import UnprocessableEntity
9695
@@ -102,7 +101,7 @@ def test_get_project_should_return_422_for_invalid_id_format(client: FlaskClient
102101
103102 response = client .get ("/projects/%s" % invalid_id , follow_redirects = True )
104103
105- assert 422 == response .status_code
104+ assert HTTPStatus . UNPROCESSABLE_ENTITY == response .status_code
106105 repository_find_mock .assert_called_once_with (str (invalid_id ))
107106
108107
@@ -116,7 +115,7 @@ def test_update_project_should_succeed_with_valid_data(client: FlaskClient, mock
116115 valid_id = fake .random_int (1 , 9999 )
117116 response = client .put ("/projects/%s" % valid_id , json = valid_project_data , follow_redirects = True )
118117
119- assert 200 == response .status_code
118+ assert HTTPStatus . OK == response .status_code
120119 fake_project == json .loads (response .data )
121120 repository_update_mock .assert_called_once_with (str (valid_id ), valid_project_data )
122121
@@ -133,7 +132,7 @@ def test_update_project_should_reject_bad_request(client: FlaskClient, mocker: M
133132 valid_id = fake .random_int (1 , 9999 )
134133 response = client .put ("/projects/%s" % valid_id , json = invalid_project_data , follow_redirects = True )
135134
136- assert 400 == response .status_code
135+ assert HTTPStatus . BAD_REQUEST == response .status_code
137136 repository_update_mock .assert_not_called ()
138137
139138
@@ -151,7 +150,7 @@ def test_update_project_should_return_not_found_with_invalid_id(client: FlaskCli
151150 json = valid_project_data ,
152151 follow_redirects = True )
153152
154- assert 404 == response .status_code
153+ assert HTTPStatus . NOT_FOUND == response .status_code
155154 repository_update_mock .assert_called_once_with (str (invalid_id ), valid_project_data )
156155
157156
@@ -166,7 +165,7 @@ def test_delete_project_should_succeed_with_valid_id(client: FlaskClient, mocker
166165
167166 response = client .delete ("/projects/%s" % valid_id , follow_redirects = True )
168167
169- assert 204 == response .status_code
168+ assert HTTPStatus . NO_CONTENT == response .status_code
170169 assert b'' == response .data
171170 repository_remove_mock .assert_called_once_with (str (valid_id ))
172171
@@ -183,11 +182,12 @@ def test_delete_project_should_return_not_found_with_invalid_id(client: FlaskCli
183182
184183 response = client .delete ("/projects/%s" % invalid_id , follow_redirects = True )
185184
186- assert 404 == response .status_code
185+ assert HTTPStatus . NOT_FOUND == response .status_code
187186 repository_remove_mock .assert_called_once_with (str (invalid_id ))
188187
189188
190- def test_delete_project_should_return_422_for_invalid_id_format (client : FlaskClient , mocker : MockFixture ):
189+ def test_delete_project_should_return_unprocessable_entity_for_invalid_id_format (client : FlaskClient ,
190+ mocker : MockFixture ):
191191 from time_tracker_api .projects .projects_namespace import project_dao
192192 from werkzeug .exceptions import UnprocessableEntity
193193
@@ -199,5 +199,5 @@ def test_delete_project_should_return_422_for_invalid_id_format(client: FlaskCli
199199
200200 response = client .delete ("/projects/%s" % invalid_id , follow_redirects = True )
201201
202- assert 422 == response .status_code
202+ assert HTTPStatus . UNPROCESSABLE_ENTITY == response .status_code
203203 repository_remove_mock .assert_called_once_with (str (invalid_id ))
0 commit comments