Coverage for tests/test_instance_no_args.py: 100%

1from typing import Optional 6 ctx1abcdef

2 

3import pytest 6 ctx1abcdef

4from pydantic import ValidationError 6 ctx1abcdef

5from sqlmodel import Field, Session, SQLModel, create_engine, select 6 ctx1abcdef

6 

7 

8def test_allow_instantiation_without_arguments(clear_sqlmodel): 6 ctx1abcdef

9 class Item(SQLModel, table=True): 6 ctx1abcdef

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

11 name: str 6 ctx1abcdef

12 description: Optional[str] = None 6 ctx1abcdef

13 

14 engine = create_engine("sqlite:///:memory:") 6 ctx1abcdef

15 SQLModel.metadata.create_all(engine) 6 ctx1abcdef

16 with Session(engine) as db: 6 ctx1abcdef

17 item = Item() 6 ctx1abcdef

18 item.name = "Rick" 6 ctx1abcdef

19 db.add(item) 6 ctx1abcdef

20 db.commit() 6 ctx1abcdef

21 statement = select(Item) 6 ctx1abcdef

22 result = db.exec(statement).all() 6 ctx1abcdef

23 assert len(result) == 1 6 ctx1abcdef

24 assert isinstance(item.id, int) 6 ctx1abcdef

25 SQLModel.metadata.clear() 6 ctx1abcdef

26 

27 

28def test_not_allow_instantiation_without_arguments_if_not_table(): 6 ctx1abcdef

29 class Item(SQLModel): 6 ctx1abcdef

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

31 name: str 6 ctx1abcdef

32 description: Optional[str] = None 6 ctx1abcdef

33 

34 with pytest.raises(ValidationError): 6 ctx1abcdef

35 Item() 6 ctx1abcdef