add extra to Model.Meta, update docs and bump version

This commit is contained in:
collerek
2021-09-26 14:07:47 +02:00
parent 67487ebf9c
commit 1da9111dbd
10 changed files with 158 additions and 4 deletions

View File

@ -0,0 +1,63 @@
import json
import databases
import pytest
import sqlalchemy
from fastapi import FastAPI
from starlette.testclient import TestClient
import ormar
from ormar import Extra
from tests.settings import DATABASE_URL
app = FastAPI()
metadata = sqlalchemy.MetaData()
database = databases.Database(DATABASE_URL, force_rollback=True)
app.state.database = database
@app.on_event("startup")
async def startup() -> None:
database_ = app.state.database
if not database_.is_connected:
await database_.connect()
@app.on_event("shutdown")
async def shutdown() -> None:
database_ = app.state.database
if database_.is_connected:
await database_.disconnect()
class Item(ormar.Model):
class Meta:
database = database
metadata = metadata
extra = Extra.ignore
id: int = ormar.Integer(primary_key=True)
name: str = ormar.String(max_length=100)
@pytest.fixture(autouse=True, scope="module")
def create_test_database():
engine = sqlalchemy.create_engine(DATABASE_URL)
metadata.create_all(engine)
yield
metadata.drop_all(engine)
@app.post("/item/", response_model=Item)
async def create_item(item: Item):
return await item.save()
def test_extra_parameters_in_request():
client = TestClient(app)
with client as client:
data = {"name": "Name", "extraname": "to ignore"}
resp = client.post("item/", data=json.dumps(data))
assert resp.status_code == 200
assert "name" in resp.json()
assert resp.json().get("name") == "Name"

View File

@ -0,0 +1,28 @@
import databases
import sqlalchemy
import ormar
from ormar import Extra
from tests.settings import DATABASE_URL
database = databases.Database(DATABASE_URL, force_rollback=True)
metadata = sqlalchemy.MetaData()
class Child(ormar.Model):
class Meta(ormar.ModelMeta):
tablename = "children"
metadata = metadata
database = database
extra = Extra.ignore
id: int = ormar.Integer(name="child_id", primary_key=True)
first_name: str = ormar.String(name="fname", max_length=100)
last_name: str = ormar.String(name="lname", max_length=100)
def test_allow_extra_parameter():
child = Child(first_name="Test", last_name="Name", extra_param="Unexpected")
assert child.first_name == "Test"
assert child.last_name == "Name"
assert not hasattr(child, "extra_param")