Coverage for tests / test_instance_no_args.py: 100%

1import pytest 9 ctx1jklmnopqr

2from pydantic import ValidationError 9 ctx1jklmnopqr

3from sqlalchemy import Engine 9 ctx1jklmnopqr

4from sqlmodel import Field, Session, SQLModel, select 9 ctx1jklmnopqr

5 

6 

7def test_allow_instantiation_without_arguments(database_engine: Engine): 9 ctx1jklmnopqr

8 class Item(SQLModel, table=True): 9 ctx1abcdefghi

9 id: int | None = Field(default=None, primary_key=True) 9 ctx1abcdefghi

10 name: str 9 ctx1abcdefghi

11 description: str | None = None 9 ctx1abcdefghi

12 

13 engine = database_engine 9 ctx1abcdefghi

14 SQLModel.metadata.create_all(engine) 9 ctx1abcdefghi

15 with Session(engine) as db: 9 ctx1abcdefghi

16 item = Item() 9 ctx1abcdefghi

17 item.name = "Rick" 9 ctx1abcdefghi

18 db.add(item) 9 ctx1abcdefghi

19 db.commit() 9 ctx1abcdefghi

20 statement = select(Item) 9 ctx1abcdefghi

21 result = db.exec(statement).all() 9 ctx1abcdefghi

22 assert len(result) == 1 9 ctx1abcdefghi

23 assert isinstance(item.id, int) 9 ctx1abcdefghi

24 

25 

26def test_not_allow_instantiation_without_arguments_if_not_table(): 9 ctx1jklmnopqr

27 class Item(SQLModel): 9 ctx1stuvwxyzA

28 id: int | None = Field(default=None, primary_key=True) 9 ctx1stuvwxyzA

29 name: str 9 ctx1stuvwxyzA

30 description: str | None = None 9 ctx1stuvwxyzA

31 

32 with pytest.raises(ValidationError): 9 ctx1stuvwxyzA

33 Item() 9 ctx1stuvwxyzA