Coverage for tests / test_instance_no_args.py: 100%

1from typing import Optional 9 ctx1jklmnopqr

2 

3import pytest 9 ctx1jklmnopqr

4from pydantic import ValidationError 9 ctx1jklmnopqr

5from sqlmodel import Field, Session, SQLModel, create_engine, select 9 ctx1jklmnopqr

6 

7 

8def test_allow_instantiation_without_arguments(clear_sqlmodel): 9 ctx1jklmnopqr

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

10 id: Optional[int] = Field(default=None, primary_key=True) 9 ctx1abcdefghi

11 name: str 9 ctx1abcdefghi

12 description: Optional[str] = None 9 ctx1abcdefghi

13 

14 engine = create_engine("sqlite:///:memory:") 9 ctx1abcdefghi

15 SQLModel.metadata.create_all(engine) 9 ctx1abcdefghi

16 with Session(engine) as db: 9 ctx1abcdefghi

17 item = Item() 9 ctx1abcdefghi

18 item.name = "Rick" 9 ctx1abcdefghi

19 db.add(item) 9 ctx1abcdefghi

20 db.commit() 9 ctx1abcdefghi

21 statement = select(Item) 9 ctx1abcdefghi

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

23 assert len(result) == 1 9 ctx1abcdefghi

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

25 SQLModel.metadata.clear() 9 ctx1abcdefghi

26 

27 

28def test_not_allow_instantiation_without_arguments_if_not_table(): 9 ctx1jklmnopqr

29 class Item(SQLModel): 9 ctx1stuvwxyzA

30 id: Optional[int] = Field(default=None, primary_key=True) 9 ctx1stuvwxyzA

31 name: str 9 ctx1stuvwxyzA

32 description: Optional[str] = None 9 ctx1stuvwxyzA

33 

34 with pytest.raises(ValidationError): 9 ctx1stuvwxyzA

35 Item() 9 ctx1stuvwxyzA