Skip to content

add some features, change depricated methods #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
assigdev wants to merge 6 commits into
base: master
Choose a base branch
from
Open

add some features, change depricated methods #1

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
9e4ad73
add start_session method for client
assigdev Oct 4, 2018
4c246b8
add async methods in ASYNC_METHODS for AsyncCollection
assigdev Oct 4, 2018
6fbc9cb
change sync load fixture to async
assigdev Oct 4, 2018
5fb492b
add async mongo client fixture
assigdev Oct 4, 2018
d18ad2e
fix tests, changed deprecated methods count to count_documents and in…
assigdev Oct 4, 2018
d6fc34b
start_session method for client
assigdev Oct 4, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
add start_session method for client
  • Loading branch information
@assigdev
assigdev committed Oct 4, 2018
commit 9e4ad7389f1d5f1b6fbc09d28b286902f8ed1923
13 changes: 12 additions & 1 deletion pytest_async_mongodb/plugin.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,6 @@ def pytest_addoption(parser):
help='Try loading fixtures from this directory')



def wrapper(func):
@functools.wraps(func)
async def wrapped(*args, **kwargs):
Expand All @@ -51,6 +50,14 @@ def __getattribute__(self, name):
return attr


class Session:
async def __aenter__(self):
await asyncio.sleep(0)

async def __aexit__(self, exc_type, exc, tb):
await asyncio.sleep(0)


class AsyncCollection(AsyncClassMethod, mongomock.Collection):

ASYNC_METHODS = [
Expand All @@ -74,6 +81,10 @@ async def find_one(self, filter=None, *args, **kwargs):
except StopIteration:
return None

async def start_session(self, **kwargs):
await asyncio.sleep(0)
return Session()


class AsyncDatabase(AsyncClassMethod, mongomock.Database):

Expand Down