Coverage for tests/test_instance_no_args.py: 100%

1from typing import Optional 10 ctx1klmnopqrst

2 

3import pytest 10 ctx1klmnopqrst

4from pydantic import ValidationError 10 ctx1klmnopqrst

5from sqlmodel import Field, Session, SQLModel, create_engine, select 10 ctx1klmnopqrst

6 

7 

8def test_allow_instantiation_without_arguments(clear_sqlmodel): 10 ctx1klmnopqrst

9 class Item(SQLModel, table=True): 10 ctx1abcdefghij

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

11 name: str 10 ctx1abcdefghij

12 description: Optional[str] = None 10 ctx1abcdefghij

13 

14 engine = create_engine("sqlite:///:memory:") 10 ctx1abcdefghij

15 SQLModel.metadata.create_all(engine) 10 ctx1abcdefghij

16 with Session(engine) as db: 10 ctx1abcdefghij

17 item = Item() 10 ctx1abcdefghij

18 item.name = "Rick" 10 ctx1abcdefghij

19 db.add(item) 10 ctx1abcdefghij

20 db.commit() 10 ctx1abcdefghij

21 statement = select(Item) 10 ctx1abcdefghij

22 result = db.exec(statement).all() 10 ctx1abcdefghij

23 assert len(result) == 1 10 ctx1abcdefghij

24 assert isinstance(item.id, int) 10 ctx1abcdefghij

25 SQLModel.metadata.clear() 10 ctx1abcdefghij

26 

27 

28def test_not_allow_instantiation_without_arguments_if_not_table(): 10 ctx1klmnopqrst

29 class Item(SQLModel): 10 ctx1uvwxyzABCD

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

31 name: str 10 ctx1uvwxyzABCD

32 description: Optional[str] = None 10 ctx1uvwxyzABCD

33 

34 with pytest.raises(ValidationError): 10 ctx1uvwxyzABCD

35 Item() 10 ctx1uvwxyzABCD