Coverage for tests / test_instance_no_args.py: 100%

1import pytest 8 ctx1qrstuvwx

2from pydantic import ValidationError 8 ctx1qrstuvwx

3from sqlmodel import Field, Session, SQLModel, create_engine, select 8 ctx1qrstuvwx

4 

5 

6def test_allow_instantiation_without_arguments(clear_sqlmodel): 8 ctx1qrstuvwx

7 class Item(SQLModel, table=True): 8 ctx1abcdefgh

8 id: int | None = Field(default=None, primary_key=True) 8 ctx1abcdefgh

9 name: str 8 ctx1abcdefgh

10 description: str | None = None 8 ctx1abcdefgh

11 

12 engine = create_engine("sqlite:///:memory:") 8 ctx1abcdefgh

13 SQLModel.metadata.create_all(engine) 8 ctx1abcdefgh

14 with Session(engine) as db: 8 ctx1abcdefgh

15 item = Item() 8 ctx1abcdefgh

16 item.name = "Rick" 8 ctx1abcdefgh

17 db.add(item) 8 ctx1abcdefgh

18 db.commit() 8 ctx1abcdefgh

19 statement = select(Item) 8 ctx1abcdefgh

20 result = db.exec(statement).all() 8 ctx1abcdefgh

21 assert len(result) == 1 8 ctx1abcdefgh

22 assert isinstance(item.id, int) 8 ctx1abcdefgh

23 SQLModel.metadata.clear() 8 ctx1abcdefgh

24 

25 

26def test_not_allow_instantiation_without_arguments_if_not_table(): 8 ctx1qrstuvwx

27 class Item(SQLModel): 8 ctx1ijklmnop

28 id: int | None = Field(default=None, primary_key=True) 8 ctx1ijklmnop

29 name: str 8 ctx1ijklmnop

30 description: str | None = None 8 ctx1ijklmnop

31 

32 with pytest.raises(ValidationError): 8 ctx1ijklmnop

33 Item() 8 ctx1ijklmnop