import asyncio

from typing import Optional

import pytest

from fastapi_users import models

class User(models.BaseUser):

first_name: Optional[str]

class UserCreate(models.BaseUserCreate):

first_name: Optional[str]

class UserUpdate(models.BaseUserUpdate):

pass

class UserDB(User, models.BaseUserDB):

pass

class UserOAuth(User, models.BaseOAuthAccountMixin):

pass

class UserDBOAuth(UserOAuth, UserDB):

pass

@pytest.fixture(scope="session")

def event_loop():

"""Force the pytest-asyncio loop to be the main one."""

loop = asyncio.new_event_loop()

yield loop

loop.close()

@pytest.fixture

def oauth_account1() -> models.BaseOAuthAccount:

return models.BaseOAuthAccount(

oauth_name="service1",

access_token="TOKEN",

expires_at=1579000751,

account_id="user_oauth1",

account_email="king.arthur@camelot.bt",

)

@pytest.fixture

def oauth_account2() -> models.BaseOAuthAccount:

return models.BaseOAuthAccount(

oauth_name="service2",

access_token="TOKEN",

expires_at=1579000751,

account_id="user_oauth2",

account_email="king.arthur@camelot.bt",

)