Add nested model and relationship test

Author: LonelyVikingMichael

tests/test_fastapi_users_db_ormar.py

@@ -20,12 +20,23 @@
 database = databases.Database(DATABASE_URL)
 
 
+class Role(ormar.Model):
+ class Meta:
+ tablename = "roles"
+ metadata = metadata
+ database = database
+
+ id = ormar.UUID(primary_key=True, uuid_format="string")
+ name = ormar.String(nullable=False, max_length=255)
+
+
 class User(OrmarBaseUserModel):
  class Meta:
  metadata = metadata
  database = database
 
  first_name = ormar.String(nullable=True, max_length=255)
+ roles = ormar.ManyToMany(Role, skip_reverse=True)
 
 
 class OAuthAccount(OrmarBaseOAuthAccountModel):
@@ -196,3 +207,27 @@ async def test_queries_oauth(
  # Unknown OAuth account
  unknown_oauth_user = await ormar_user_db_oauth.get_by_oauth_account("foo", "bar")
  assert unknown_oauth_user is None
+
+
+@pytest.mark.asyncio
+@pytest.mark.db
+async def test_queries_custom_fields_relations(ormar_user_db: OrmarUserDatabase[UserDB]):
+ # Create role to pair with
+ role = await Role.objects.create(
+ id=uuid.uuid4(),
+ name='editor'
+ )
+
+ assert role.id is not None
+
+ user = UserDB(
+ email="lancelot@camelot.bt",
+ hashed_password="guinevere",
+ roles=[role]
+ )
+
+ # Create with relationship
+ user_db = await ormar_user_db.create(user)
+ assert user_db.roles is not None
+ assert len(user_db.roles) is not 0
+ assert user_db.roles[0].id == role.id