Coverage for tests/test_field_sa_args_kwargs.py: 100%

1from typing import Optional 6 ctx1mnopqr

2 

3from sqlalchemy import ForeignKey 6 ctx1mnopqr

4from sqlmodel import Field, SQLModel, create_engine 6 ctx1mnopqr

5 

6 

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

8 class Team(SQLModel, table=True): 6 ctx1abcdef

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

10 name: str 6 ctx1abcdef

11 

12 class Hero(SQLModel, table=True): 6 ctx1abcdef

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

14 team_id: Optional[int] = Field( 6 ctx1abcdef

15 default=None, 

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

17 ) 

18 

19 engine = create_engine("sqlite://", echo=True) 6 ctx1abcdef

20 SQLModel.metadata.create_all(engine) 6 ctx1abcdef

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 6 ctx1abcdef

25 

26 

27def test_sa_column_kargs(clear_sqlmodel, caplog) -> None: 6 ctx1mnopqr

28 class Item(SQLModel, table=True): 6 ctx1ghijkl

29 id: Optional[int] = Field( 6 ctx1ghijkl

30 default=None, 

31 sa_column_kwargs={"primary_key": True}, 

32 ) 

33 

34 engine = create_engine("sqlite://", echo=True) 6 ctx1ghijkl

35 SQLModel.metadata.create_all(engine) 6 ctx1ghijkl

36 create_table_log = [ 6 ctx1ghijkl

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

38 ][0] 

39 assert "PRIMARY KEY (id)" in create_table_log 6 ctx1ghijkl