Coverage for tests/test_instance_no_args.py: 100%

1from typing import Optional 7 ctx1hijklmn

2 

3import pytest 7 ctx1hijklmn

4from pydantic import ValidationError 7 ctx1hijklmn

5from sqlmodel import Field, Session, SQLModel, create_engine, select 7 ctx1hijklmn

6 

7 

8def test_allow_instantiation_without_arguments(clear_sqlmodel): 7 ctx1hijklmn

9 class Item(SQLModel, table=True): 7 ctx1abcdefg

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

11 name: str 7 ctx1abcdefg

12 description: Optional[str] = None 7 ctx1abcdefg

13 

14 engine = create_engine("sqlite:///:memory:") 7 ctx1abcdefg

15 SQLModel.metadata.create_all(engine) 7 ctx1abcdefg

16 with Session(engine) as db: 7 ctx1abcdefg

17 item = Item() 7 ctx1abcdefg

18 item.name = "Rick" 7 ctx1abcdefg

19 db.add(item) 7 ctx1abcdefg

20 db.commit() 7 ctx1abcdefg

21 statement = select(Item) 7 ctx1abcdefg

22 result = db.exec(statement).all() 7 ctx1abcdefg

23 assert len(result) == 1 7 ctx1abcdefg

24 assert isinstance(item.id, int) 7 ctx1abcdefg

25 SQLModel.metadata.clear() 7 ctx1abcdefg

26 

27 

28def test_not_allow_instantiation_without_arguments_if_not_table(): 7 ctx1hijklmn

29 class Item(SQLModel): 7 ctx1opqrstu

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

31 name: str 7 ctx1opqrstu

32 description: Optional[str] = None 7 ctx1opqrstu

33 

34 with pytest.raises(ValidationError): 7 ctx1opqrstu

35 Item() 7 ctx1opqrstu