1  from api.database import db, db_wrapper, select
 2  from api.models import OAuthUserConnection
 3  
 4  
 5  async def test__serialize() -> None:
 6      obj = OAuthUserConnection(id="connection_id_123", provider_id="my_oauth_provider", display_name="Foo Bar")
 7      assert obj.serialize == {"id": "connection_id_123", "provider_id": "my_oauth_provider", "display_name": "Foo Bar"}
 8  
 9  
10  @db_wrapper
11  async def test__create() -> None:
12      obj = await OAuthUserConnection.create("user_id_123", "my_oauth_provider", "remote_user_id_123", "Foo Bar")
13      connections = await db.all(select(OAuthUserConnection))
14      assert connections == [obj]
15  
16      assert obj.user_id == "user_id_123"
17      assert obj.provider_id == "my_oauth_provider"
18      assert obj.remote_user_id == "remote_user_id_123"
19      assert obj.display_name == "Foo Bar"