Skip to content

test_requests

tests.test_requests

Classes

MockModel

Bases: Base

Source code in tests/test_requests.py 
18
19
20
21
22
23
24
25
26
27
28
29
class MockModel(Base):
    __tablename__ = 'mockmodel'
    id = Column(Integer, primary_key=True, autoincrement=True)
    field1 = Column(String)
    field2 = Column(String)

    def serialize(self):
        return {
            "id": self.id,
            "field1": self.field1,
            "field2": self.field2,
        }
Attributes
__tablename__ class-attribute instance-attribute 
__tablename__ = 'mockmodel'
id class-attribute instance-attribute 
id = Column(Integer, primary_key=True, autoincrement=True)
field1 class-attribute instance-attribute 
field1 = Column(String)
field2 class-attribute instance-attribute 
field2 = Column(String)
Functions
serialize 
serialize()
Source code in tests/test_requests.py 
24
25
26
27
28
29
def serialize(self):
    return {
        "id": self.id,
        "field1": self.field1,
        "field2": self.field2,
    }

Functions

mock_session 

mock_session()
Source code in tests/test_requests.py 
31
32
33
34
35
@pytest.fixture
def mock_session():
    with patch('lightapi.handlers.SessionLocal') as session_mock:
        session = session_mock.return_value
        yield session

test_data 

test_data()
Source code in tests/test_requests.py 
37
38
39
@pytest.fixture
def test_data():
    return {"field1": "value1", "field2": "value2"}

test_create_handler async 

test_create_handler(mock_session, test_data)
Source code in tests/test_requests.py 
42
43
44
45
46
47
48
49
50
@pytest.mark.asyncio
async def test_create_handler(mock_session, test_data):
    handler = CreateHandler(model=MockModel)
    request = make_mocked_request('POST', '/', headers={'Content-Type': 'application/json'}, payload=str(test_data))

    mock_session.query(MockModel).filter.return_value.first.return_value = None
    response = await handler.handle(mock_session, request)

    assert response.status == 201

test_read_handler_existing_item async 

test_read_handler_existing_item(mock_session)
Source code in tests/test_requests.py 
52
53
54
55
56
57
58
59
60
61
@pytest.mark.asyncio
async def test_read_handler_existing_item(mock_session):
    handler = ReadHandler(model=MockModel)
    mock_item = MagicMock(spec=MockModel)
    mock_session.query(MockModel).filter.return_value.first.return_value = mock_item

    request = make_mocked_request('GET', '/1', match_info={'id': '1'})
    response = await handler.handle(mock_session, request)

    assert response.status == 200

test_read_handler_missing_item async 

test_read_handler_missing_item(mock_session)
Source code in tests/test_requests.py 
63
64
65
66
67
68
69
70
71
@pytest.mark.asyncio
async def test_read_handler_missing_item(mock_session):
    handler = ReadHandler(model=MockModel)
    mock_session.query(MockModel).filter.return_value.first.return_value = None

    request = make_mocked_request('GET', '/1', match_info={'id': '1'})
    response = await handler.handle(mock_session, request)

    assert response.status == 404

test_update_handler_existing_item async 

test_update_handler_existing_item(mock_session, test_data)
Source code in tests/test_requests.py 
73
74
75
76
77
78
79
80
81
82
@pytest.mark.asyncio
async def test_update_handler_existing_item(mock_session, test_data):
    handler = UpdateHandler(model=MockModel)
    mock_item = MagicMock(spec=MockModel)
    mock_session.query(MockModel).filter.return_value.first.return_value = mock_item

    request = make_mocked_request('PUT', '/1', headers={'Content-Type': 'application/json'}, payload=str(test_data), match_info={'id': '1'})
    response = await handler.handle(mock_session, request)

    assert response.status == 200

test_update_handler_missing_item async 

test_update_handler_missing_item(mock_session, test_data)
Source code in tests/test_requests.py 
84
85
86
87
88
89
90
91
92
@pytest.mark.asyncio
async def test_update_handler_missing_item(mock_session, test_data):
    handler = UpdateHandler(model=MockModel)
    mock_session.query(MockModel).filter.return_value.first.return_value = None

    request = make_mocked_request('PUT', '/1', headers={'Content-Type': 'application/json'}, payload=str(test_data), match_info={'id': '1'})
    response = await handler.handle(mock_session, request)

    assert response.status == 404

test_delete_handler_existing_item async 

test_delete_handler_existing_item(mock_session)
Source code in tests/test_requests.py 
 94
 95
 96
 97
 98
 99
100
101
102
103
@pytest.mark.asyncio
async def test_delete_handler_existing_item(mock_session):
    handler = DeleteHandler(model=MockModel)
    mock_item = MagicMock(spec=MockModel)
    mock_session.query(MockModel).filter.return_value.first.return_value = mock_item

    request = make_mocked_request('DELETE', '/1', match_info={'id': '1'})
    response = await handler.handle(mock_session, request)

    assert response.status == 204

test_delete_handler_missing_item async 

test_delete_handler_missing_item(mock_session)
Source code in tests/test_requests.py 
105
106
107
108
109
110
111
112
113
@pytest.mark.asyncio
async def test_delete_handler_missing_item(mock_session):
    handler = DeleteHandler(model=MockModel)
    mock_session.query(MockModel).filter.return_value.first.return_value = None

    request = make_mocked_request('DELETE', '/1', match_info={'id': '1'})
    response = await handler.handle(mock_session, request)

    assert response.status == 404

test_retrieve_all_handler async 

test_retrieve_all_handler(mock_session)
Source code in tests/test_requests.py 
115
116
117
118
119
120
121
122
123
124
@pytest.mark.asyncio
async def test_retrieve_all_handler(mock_session):
    handler = RetrieveAllHandler(model=MockModel)
    mock_items = [MagicMock(spec=MockModel), MagicMock(spec=MockModel)]
    mock_session.query(MockModel).all.return_value = mock_items

    request = make_mocked_request('GET', '/')
    response = await handler.handle(mock_session, request)

    assert response.status == 200

test_options_handler async 

test_options_handler(mock_session)
Source code in tests/test_requests.py 
126
127
128
129
130
131
132
133
@pytest.mark.asyncio
async def test_options_handler(mock_session):
    handler = OptionsHandler(model=MockModel)
    request = make_mocked_request('OPTIONS', '/')

    response = await handler.handle(mock_session, request)

    assert response.status == 200

test_head_handler async 

test_head_handler(mock_session)
Source code in tests/test_requests.py 
135
136
137
138
139
140
141
142
@pytest.mark.asyncio
async def test_head_handler(mock_session):
    handler = HeadHandler(model=MockModel)
    request = make_mocked_request('HEAD', '/')

    response = await handler.handle(mock_session, request)

    assert response.status == 200