|
| 1 | +from faker import Faker |
1 | 2 | from flask import json
|
2 | 3 | from flask.testing import FlaskClient
|
3 | 4 | from pytest_mock import MockFixture
|
4 | 5 |
|
| 6 | +from time_tracker_api.projects.projects_model import PROJECT_TYPE |
5 | 7 |
|
6 |
| -def test_list_all_elements(client: FlaskClient, mocker: MockFixture): |
| 8 | +fake = Faker() |
| 9 | + |
| 10 | +valid_project_data = { |
| 11 | + "name": fake.company(), |
| 12 | + "description": fake.paragraph(), |
| 13 | + "type": fake.word(PROJECT_TYPE.valid_type_values()), |
| 14 | +} |
| 15 | +fake_project = ({ |
| 16 | + "id": fake.random_int(1, 9999) |
| 17 | +}).update(valid_project_data) |
| 18 | + |
| 19 | + |
| 20 | +def test_create_project_should_succeed_with_valid_request(client: FlaskClient, mocker: MockFixture): |
| 21 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 22 | + repository_create_mock = mocker.patch.object(project_dao.repository, |
| 23 | + 'create', |
| 24 | + return_value=fake_project) |
| 25 | + |
| 26 | + response = client.post("/projects", json=valid_project_data, follow_redirects=True) |
| 27 | + |
| 28 | + assert 201 == response.status_code |
| 29 | + repository_create_mock.assert_called_once_with(valid_project_data) |
| 30 | + |
| 31 | + |
| 32 | +def test_create_project_should_reject_bad_request(client: FlaskClient, mocker: MockFixture): |
7 | 33 | from time_tracker_api.projects.projects_namespace import project_dao
|
8 |
| - repository_find_all_mock = mocker.patch.object(project_dao.repository, 'find_all', return_value=[]) |
| 34 | + invalid_project_data = valid_project_data.copy().update({ |
| 35 | + "type": 'anything', |
| 36 | + }) |
| 37 | + repository_create_mock = mocker.patch.object(project_dao.repository, |
| 38 | + 'create', |
| 39 | + return_value=fake_project) |
| 40 | + |
| 41 | + response = client.post("/projects", json=invalid_project_data, follow_redirects=True) |
| 42 | + |
| 43 | + assert 400 == response.status_code |
| 44 | + repository_create_mock.assert_not_called() |
| 45 | + |
| 46 | + |
| 47 | +def test_list_all_projects(client: FlaskClient, mocker: MockFixture): |
| 48 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 49 | + repository_find_all_mock = mocker.patch.object(project_dao.repository, |
| 50 | + 'find_all', |
| 51 | + return_value=[]) |
9 | 52 |
|
10 | 53 | response = client.get("/projects", follow_redirects=True)
|
11 | 54 |
|
12 | 55 | assert 200 == response.status_code
|
13 |
| - |
14 | 56 | json_data = json.loads(response.data)
|
15 | 57 | assert [] == json_data
|
16 | 58 | repository_find_all_mock.assert_called_once()
|
| 59 | + |
| 60 | + |
| 61 | +def test_get_project_should_succeed_with_valid_id(client: FlaskClient, mocker: MockFixture): |
| 62 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 63 | + |
| 64 | + valid_id = fake.random_int(1, 9999) |
| 65 | + |
| 66 | + repository_find_mock = mocker.patch.object(project_dao.repository, |
| 67 | + 'find', |
| 68 | + return_value=fake_project) |
| 69 | + |
| 70 | + response = client.get("/projects/%s" % valid_id, follow_redirects=True) |
| 71 | + |
| 72 | + assert 200 == response.status_code |
| 73 | + fake_project == json.loads(response.data) |
| 74 | + repository_find_mock.assert_called_once_with(str(valid_id)) |
| 75 | + |
| 76 | + |
| 77 | +def test_get_project_should_return_not_found_with_invalid_id(client: FlaskClient, mocker: MockFixture): |
| 78 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 79 | + from werkzeug.exceptions import NotFound |
| 80 | + |
| 81 | + invalid_id = fake.random_int(1, 9999) |
| 82 | + |
| 83 | + repository_find_mock = mocker.patch.object(project_dao.repository, |
| 84 | + 'find', |
| 85 | + side_effect=NotFound) |
| 86 | + |
| 87 | + response = client.get("/projects/%s" % invalid_id, follow_redirects=True) |
| 88 | + |
| 89 | + assert 404 == response.status_code |
| 90 | + repository_find_mock.assert_called_once_with(str(invalid_id)) |
| 91 | + |
| 92 | + |
| 93 | +def test_get_project_should_return_422_for_invalid_id_format(client: FlaskClient, mocker: MockFixture): |
| 94 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 95 | + from werkzeug.exceptions import UnprocessableEntity |
| 96 | + |
| 97 | + invalid_id = fake.company() |
| 98 | + |
| 99 | + repository_find_mock = mocker.patch.object(project_dao.repository, |
| 100 | + 'find', |
| 101 | + side_effect=UnprocessableEntity) |
| 102 | + |
| 103 | + response = client.get("/projects/%s" % invalid_id, follow_redirects=True) |
| 104 | + |
| 105 | + assert 422 == response.status_code |
| 106 | + repository_find_mock.assert_called_once_with(str(invalid_id)) |
| 107 | + |
| 108 | + |
| 109 | +def test_update_project_should_succeed_with_valid_data(client: FlaskClient, mocker: MockFixture): |
| 110 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 111 | + |
| 112 | + repository_update_mock = mocker.patch.object(project_dao.repository, |
| 113 | + 'update', |
| 114 | + return_value=fake_project) |
| 115 | + |
| 116 | + valid_id = fake.random_int(1, 9999) |
| 117 | + response = client.put("/projects/%s" % valid_id, json=valid_project_data, follow_redirects=True) |
| 118 | + |
| 119 | + assert 200 == response.status_code |
| 120 | + fake_project == json.loads(response.data) |
| 121 | + repository_update_mock.assert_called_once_with(str(valid_id), valid_project_data) |
| 122 | + |
| 123 | + |
| 124 | +def test_update_project_should_reject_bad_request(client: FlaskClient, mocker: MockFixture): |
| 125 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 126 | + invalid_project_data = valid_project_data.copy().update({ |
| 127 | + "type": 'anything', |
| 128 | + }) |
| 129 | + repository_update_mock = mocker.patch.object(project_dao.repository, |
| 130 | + 'update', |
| 131 | + return_value=fake_project) |
| 132 | + |
| 133 | + valid_id = fake.random_int(1, 9999) |
| 134 | + response = client.put("/projects/%s" % valid_id, json=invalid_project_data, follow_redirects=True) |
| 135 | + |
| 136 | + assert 400 == response.status_code |
| 137 | + repository_update_mock.assert_not_called() |
| 138 | + |
| 139 | + |
| 140 | +def test_update_project_should_return_not_found_with_invalid_id(client: FlaskClient, mocker: MockFixture): |
| 141 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 142 | + from werkzeug.exceptions import NotFound |
| 143 | + |
| 144 | + invalid_id = fake.random_int(1, 9999) |
| 145 | + |
| 146 | + repository_update_mock = mocker.patch.object(project_dao.repository, |
| 147 | + 'update', |
| 148 | + side_effect=NotFound) |
| 149 | + |
| 150 | + response = client.put("/projects/%s" % invalid_id, |
| 151 | + json=valid_project_data, |
| 152 | + follow_redirects=True) |
| 153 | + |
| 154 | + assert 404 == response.status_code |
| 155 | + repository_update_mock.assert_called_once_with(str(invalid_id), valid_project_data) |
| 156 | + |
| 157 | + |
| 158 | +def test_delete_project_should_succeed_with_valid_id(client: FlaskClient, mocker: MockFixture): |
| 159 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 160 | + |
| 161 | + valid_id = fake.random_int(1, 9999) |
| 162 | + |
| 163 | + repository_remove_mock = mocker.patch.object(project_dao.repository, |
| 164 | + 'remove', |
| 165 | + return_value=None) |
| 166 | + |
| 167 | + response = client.delete("/projects/%s" % valid_id, follow_redirects=True) |
| 168 | + |
| 169 | + assert 204 == response.status_code |
| 170 | + assert b'' == response.data |
| 171 | + repository_remove_mock.assert_called_once_with(str(valid_id)) |
| 172 | + |
| 173 | + |
| 174 | +def test_delete_project_should_return_not_found_with_invalid_id(client: FlaskClient, mocker: MockFixture): |
| 175 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 176 | + from werkzeug.exceptions import NotFound |
| 177 | + |
| 178 | + invalid_id = fake.random_int(1, 9999) |
| 179 | + |
| 180 | + repository_remove_mock = mocker.patch.object(project_dao.repository, |
| 181 | + 'remove', |
| 182 | + side_effect=NotFound) |
| 183 | + |
| 184 | + response = client.delete("/projects/%s" % invalid_id, follow_redirects=True) |
| 185 | + |
| 186 | + assert 404 == response.status_code |
| 187 | + repository_remove_mock.assert_called_once_with(str(invalid_id)) |
| 188 | + |
| 189 | + |
| 190 | +def test_delete_project_should_return_422_for_invalid_id_format(client: FlaskClient, mocker: MockFixture): |
| 191 | + from time_tracker_api.projects.projects_namespace import project_dao |
| 192 | + from werkzeug.exceptions import UnprocessableEntity |
| 193 | + |
| 194 | + invalid_id = fake.company() |
| 195 | + |
| 196 | + repository_remove_mock = mocker.patch.object(project_dao.repository, |
| 197 | + 'remove', |
| 198 | + side_effect=UnprocessableEntity) |
| 199 | + |
| 200 | + response = client.delete("/projects/%s" % invalid_id, follow_redirects=True) |
| 201 | + |
| 202 | + assert 422 == response.status_code |
| 203 | + repository_remove_mock.assert_called_once_with(str(invalid_id)) |
0 commit comments