Coverage for tests/test_field_sa_args_kwargs.py: 100%

1from typing import Optional 7 ctx1nopqrst

2 

3from sqlalchemy import ForeignKey 7 ctx1nopqrst

4from sqlmodel import Field, SQLModel, create_engine 7 ctx1nopqrst

5 

6 

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

8 class Team(SQLModel, table=True): 7 ctx1gabcdef

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

10 name: str 7 ctx1gabcdef

11 

12 class Hero(SQLModel, table=True): 7 ctx1gabcdef

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

14 team_id: Optional[int] = Field( 7 ctx1gabcdef

15 default=None, 

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

17 ) 

18 

19 engine = create_engine("sqlite://", echo=True) 7 ctx1gabcdef

20 SQLModel.metadata.create_all(engine) 7 ctx1gabcdef

21 create_table_log = [ 6 ctx1abcdef

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 7 ctx1gabcdef

25 

26 

27def test_sa_column_kargs(clear_sqlmodel, caplog) -> None: 7 ctx1nopqrst

28 class Item(SQLModel, table=True): 7 ctx1uhijklm

29 id: Optional[int] = Field( 7 ctx1uhijklm

30 default=None, 

31 sa_column_kwargs={"primary_key": True}, 

32 ) 

33 

34 engine = create_engine("sqlite://", echo=True) 7 ctx1uhijklm

35 SQLModel.metadata.create_all(engine) 7 ctx1uhijklm

36 create_table_log = [ 6 ctx1hijklm

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

38 ][0] 

39 assert "PRIMARY KEY (id)" in create_table_log 7 ctx1uhijklm