mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-12-02 14:34:02 +03:00
d45d98e300
* Add application * Dockerize the app * Fix 204 content-leength error * Rename database file * Add tests * Add README * Fix a typo in FastAPI example * Add docs on FastAPI + SQLAlchemy example * Update changelog * Add link to the example to README and other docs pages * Add EOF to the config.yml
108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
"""Tests module."""
|
|
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from .repositories import UserRepository, UserNotFoundError
|
|
from .models import User
|
|
from .application import app
|
|
|
|
|
|
@pytest.fixture
|
|
def client():
|
|
yield TestClient(app)
|
|
|
|
|
|
def test_get_list(client):
|
|
repository_mock = mock.Mock(spec=UserRepository)
|
|
repository_mock.get_all.return_value = [
|
|
User(id=1, email='test1@email.com', hashed_password='pwd', is_active=True),
|
|
User(id=2, email='test2@email.com', hashed_password='pwd', is_active=False),
|
|
]
|
|
|
|
with app.container.user_repository.override(repository_mock):
|
|
response = client.get('/users')
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data == [
|
|
{'id': 1, 'email': 'test1@email.com', 'hashed_password': 'pwd', 'is_active': True},
|
|
{'id': 2, 'email': 'test2@email.com', 'hashed_password': 'pwd', 'is_active': False},
|
|
]
|
|
|
|
|
|
def test_get_by_id(client):
|
|
repository_mock = mock.Mock(spec=UserRepository)
|
|
repository_mock.get_by_id.return_value = User(
|
|
id=1,
|
|
email='xyz@email.com',
|
|
hashed_password='pwd',
|
|
is_active=True,
|
|
)
|
|
|
|
with app.container.user_repository.override(repository_mock):
|
|
response = client.get('/users/1')
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data == {'id': 1, 'email': 'xyz@email.com', 'hashed_password': 'pwd', 'is_active': True}
|
|
repository_mock.get_by_id.assert_called_once_with(1)
|
|
|
|
|
|
def test_get_by_id_404(client):
|
|
repository_mock = mock.Mock(spec=UserRepository)
|
|
repository_mock.get_by_id.side_effect = UserNotFoundError(1)
|
|
|
|
with app.container.user_repository.override(repository_mock):
|
|
response = client.get('/users/1')
|
|
|
|
assert response.status_code == 404
|
|
|
|
|
|
@mock.patch('webapp.services.uuid4', return_value='xyz')
|
|
def test_add(_, client):
|
|
repository_mock = mock.Mock(spec=UserRepository)
|
|
repository_mock.add.return_value = User(
|
|
id=1,
|
|
email='xyz@email.com',
|
|
hashed_password='pwd',
|
|
is_active=True,
|
|
)
|
|
|
|
with app.container.user_repository.override(repository_mock):
|
|
response = client.post('/users')
|
|
|
|
assert response.status_code == 201
|
|
data = response.json()
|
|
assert data == {'id': 1, 'email': 'xyz@email.com', 'hashed_password': 'pwd', 'is_active': True}
|
|
repository_mock.add.assert_called_once_with(email='xyz@email.com', password='pwd')
|
|
|
|
|
|
def test_remove(client):
|
|
repository_mock = mock.Mock(spec=UserRepository)
|
|
|
|
with app.container.user_repository.override(repository_mock):
|
|
response = client.delete('/users/1')
|
|
|
|
assert response.status_code == 204
|
|
repository_mock.delete_by_id.assert_called_once_with(1)
|
|
|
|
|
|
def test_remove_404(client):
|
|
repository_mock = mock.Mock(spec=UserRepository)
|
|
repository_mock.delete_by_id.side_effect = UserNotFoundError(1)
|
|
|
|
with app.container.user_repository.override(repository_mock):
|
|
response = client.delete('/users/1')
|
|
|
|
assert response.status_code == 404
|
|
|
|
|
|
def test_status(client):
|
|
response = client.get('/status')
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data == {'status': 'OK'}
|