Notes
Notes - notes.io |
from .main import app
from unittest.mock import patch, AsyncMock, MagicMock
import pytest
import pandas as pd
from datetime import date
client = TestClient(app)
@pytest.fixture
def mock_db_conn_pool(monkeypatch):
mock_connection = MagicMock()
mock_pool = AsyncMock()
mock_pool.return_value.__aenter__.return_value = mock_connection
async def mock_create_database_pool():
return mock_pool
async def mock_destroy_database_pool():
pass
monkeypatch.setattr("app.database.create_database_pool", mock_create_database_pool)
monkeypatch.setattr("app.database.destroy_database_pool", mock_destroy_database_pool)
monkeypatch.setattr("app.database.execute_named_query", AsyncMock())
app.state.db_conn_pool = mock_pool
return mock_pool
@pytest.mark.usefixtures("mock_db_conn_pool")
class TestInvestors:
@pytest.mark.asyncio
async def test_investors_api(self):
params = {
"asOf": "2024-08-04",
"initials": ["JRQ", "MOC", "MR"],
"investmentDivision": ["CFII", "CII", "CRGI"]
}
investors_mock = {
"date": "2024-01-31",
"investors": [
{
"cgPersonID": "02399",
"initials": "JRQ",
"name": "John Queen",
"investmentDivision": "CFII"
},
{
"cgPersonID": "10446",
"initials": "MOC",
"name": "Mark Casey",
"investmentDivision": "CII"
},
{
"cgPersonID": "03773",
"initials": "MR",
"name": "Martin Romo",
"investmentDivision": "CRGI"
}
]
}
with patch("app.database.execute_named_query", return_value=pd.DataFrame(investors_mock['investors'])):
response = client.get("/api/v1/investors", params=params)
print("Response status code:", response.status_code)
print("Response JSON:", response.json())
if response.json() == {'detail': 'No investors found.'}:
assert response.status_code == 404
else:
assert response.status_code == 200, f"Expected status code 200, but got {response.status_code}"
assert response.json() == investors_mock, f"Expected response {investors_mock}, but got {response.json()}"
if __name__ == "__main__":
pytest.main()
|
|
Notes.io is a web-based application for taking notes. You can take your notes and share with others people. If you like taking long notes, notes.io is designed for you. To date, over 8,000,000,000 notes created and continuing...
With notes.io;
- * You can take a note from anywhere and any device with internet connection.
- * You can share the notes in social platforms (YouTube, Facebook, Twitter, instagram etc.).
- * You can quickly share your contents without website, blog and e-mail.
- * You don't need to create any Account to share a note. As you wish you can use quick, easy and best shortened notes with sms, websites, e-mail, or messaging services (WhatsApp, iMessage, Telegram, Signal).
- * Notes.io has fabulous infrastructure design for a short link and allows you to share the note as an easy and understandable link.
Fast: Notes.io is built for speed and performance. You can take a notes quickly and browse your archive.
Easy: Notes.io doesn’t require installation. Just write and share note!
Short: Notes.io’s url just 8 character. You’ll get shorten link of your note when you want to share. (Ex: notes.io/q )
Free: Notes.io works for 12 years and has been free since the day it was started.
You immediately create your first note and start sharing with the ones you wish. If you want to contact us, you can use the following communication channels;
Email: [email protected]
Twitter: http://twitter.com/notesio
Instagram: http://instagram.com/notes.io
Facebook: http://facebook.com/notesio
Regards;
Notes.io Team
