-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
13 changed files
with
54 additions
and
32 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,2 +1,12 @@ | ||
import warnings | ||
|
||
|
||
SA_DEFAULT_KEY = 'sa_main' | ||
DEFAULT_KEY = SA_DEFAULT_KEY | ||
|
||
|
||
def __getattr__(name): | ||
if name == 'DEFAULT_KEY': | ||
msg = "'DEFAULT_KEY' has been deprecated, use 'SA_DEFAULT_KEY'" | ||
warnings.warn(msg, UserWarning, stacklevel=2) | ||
return SA_DEFAULT_KEY | ||
raise AttributeError(f"module {__name__} has no attribute {name}") |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,31 +1,31 @@ | ||
import pytest | ||
from sqlalchemy.ext.asyncio import AsyncSession | ||
|
||
from aiohttp_sqlalchemy import DEFAULT_KEY, DuplicateRequestKeyError, sa_decorator | ||
from aiohttp_sqlalchemy import SA_DEFAULT_KEY, DuplicateRequestKeyError, sa_decorator | ||
from tests.conftest import ClassBasedView, ClassHandler, function_handler | ||
|
||
|
||
async def test_duplicate_request_key_error(mocked_request, orm_session): | ||
assert mocked_request.get(DEFAULT_KEY) is None | ||
mocked_request[DEFAULT_KEY] = orm_session | ||
assert mocked_request.get(DEFAULT_KEY) is orm_session | ||
assert mocked_request.get(SA_DEFAULT_KEY) is None | ||
mocked_request[SA_DEFAULT_KEY] = orm_session | ||
assert mocked_request.get(SA_DEFAULT_KEY) is orm_session | ||
with pytest.raises(DuplicateRequestKeyError): | ||
await sa_decorator()(function_handler)(mocked_request) | ||
|
||
|
||
async def test_decorated_class_based_view(mocked_request): | ||
assert mocked_request.get(DEFAULT_KEY) is None | ||
assert mocked_request.get(SA_DEFAULT_KEY) is None | ||
await sa_decorator()(ClassBasedView.get)(mocked_request) | ||
assert isinstance(mocked_request.get(DEFAULT_KEY), AsyncSession) | ||
assert isinstance(mocked_request.get(SA_DEFAULT_KEY), AsyncSession) | ||
|
||
|
||
async def test_decorated_class_handler(mocked_request): | ||
assert mocked_request.get(DEFAULT_KEY) is None | ||
assert mocked_request.get(SA_DEFAULT_KEY) is None | ||
await sa_decorator()(ClassHandler.get)(mocked_request) | ||
assert isinstance(mocked_request.get(DEFAULT_KEY), AsyncSession) | ||
assert isinstance(mocked_request.get(SA_DEFAULT_KEY), AsyncSession) | ||
|
||
|
||
async def test_decorated_function_handler(mocked_request): | ||
assert mocked_request.get(DEFAULT_KEY) is None | ||
assert mocked_request.get(SA_DEFAULT_KEY) is None | ||
await sa_decorator()(function_handler)(mocked_request) | ||
assert isinstance(mocked_request.get(DEFAULT_KEY), AsyncSession) | ||
assert isinstance(mocked_request.get(SA_DEFAULT_KEY), AsyncSession) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,20 +1,20 @@ | ||
import pytest | ||
from sqlalchemy.ext.asyncio import AsyncSession | ||
|
||
from aiohttp_sqlalchemy import DEFAULT_KEY, DuplicateRequestKeyError | ||
from aiohttp_sqlalchemy import SA_DEFAULT_KEY, DuplicateRequestKeyError | ||
from tests.conftest import function_handler | ||
|
||
|
||
async def test_duplicate_request_key_error(sa_main_middleware, mocked_request, orm_session): | ||
assert mocked_request.get(DEFAULT_KEY) is None | ||
mocked_request[DEFAULT_KEY] = orm_session | ||
assert mocked_request.get(DEFAULT_KEY) is orm_session | ||
assert mocked_request.get(SA_DEFAULT_KEY) is None | ||
mocked_request[SA_DEFAULT_KEY] = orm_session | ||
assert mocked_request.get(SA_DEFAULT_KEY) is orm_session | ||
|
||
with pytest.raises(DuplicateRequestKeyError): | ||
await sa_main_middleware(mocked_request, function_handler) | ||
|
||
|
||
async def test_sa_middleware(sa_main_middleware, mocked_request): | ||
assert mocked_request.get(DEFAULT_KEY) is None | ||
assert mocked_request.get(SA_DEFAULT_KEY) is None | ||
await sa_main_middleware(mocked_request, function_handler) | ||
assert isinstance(mocked_request.get(DEFAULT_KEY), AsyncSession) | ||
assert isinstance(mocked_request.get(SA_DEFAULT_KEY), AsyncSession) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,7 +1,7 @@ | ||
from aiohttp_sqlalchemy import DEFAULT_KEY, SABaseView | ||
from aiohttp_sqlalchemy import SA_DEFAULT_KEY, SABaseView | ||
|
||
|
||
def test_sa_session(mocked_request, orm_session): | ||
mocked_request[DEFAULT_KEY] = orm_session | ||
mocked_request[SA_DEFAULT_KEY] = orm_session | ||
view = SABaseView(mocked_request) | ||
assert view.sa_session() is orm_session |