Coverage for tests / test_field_sa_args_kwargs.py: 100%

1from typing import Optional 9 ctx1stuvwxyzA

2 

3from sqlalchemy import ForeignKey 9 ctx1stuvwxyzA

4from sqlmodel import Field, SQLModel, create_engine 9 ctx1stuvwxyzA

5 

6 

7def test_sa_column_args(clear_sqlmodel, caplog) -> None: 9 ctx1stuvwxyzA

8 class Team(SQLModel, table=True): 9 ctx1abcdefghi

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

10 name: str 9 ctx1abcdefghi

11 

12 class Hero(SQLModel, table=True): 9 ctx1abcdefghi

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

14 team_id: Optional[int] = Field( 9 ctx1abcdefghi

15 default=None, 

16 sa_column_args=[ForeignKey("team.id")], 

17 ) 

18 

19 engine = create_engine("sqlite://", echo=True) 9 ctx1abcdefghi

20 SQLModel.metadata.create_all(engine) 9 ctx1abcdefghi

21 create_table_log = [ 9 ctx1abcdefghi

22 message for message in caplog.messages if "CREATE TABLE hero" in message 

23 ][0] 

24 assert "FOREIGN KEY(team_id) REFERENCES team (id)" in create_table_log 9 ctx1abcdefghi

25 

26 

27def test_sa_column_kargs(clear_sqlmodel, caplog) -> None: 9 ctx1stuvwxyzA

28 class Item(SQLModel, table=True): 9 ctx1jklmnopqr

29 id: Optional[int] = Field( 9 ctx1jklmnopqr

30 default=None, 

31 sa_column_kwargs={"primary_key": True}, 

32 ) 

33 

34 engine = create_engine("sqlite://", echo=True) 9 ctx1jklmnopqr

35 SQLModel.metadata.create_all(engine) 9 ctx1jklmnopqr

36 create_table_log = [ 9 ctx1jklmnopqr

37 message for message in caplog.messages if "CREATE TABLE item" in message 

38 ][0] 

39 assert "PRIMARY KEY (id)" in create_table_log 9 ctx1jklmnopqr