Coverage for tests/test_field_sa_args_kwargs.py: 100%

1from typing import Optional 6 ctx1lmnopq

2 

3from sqlalchemy import ForeignKey 6 ctx1lmnopq

4from sqlmodel import Field, SQLModel, create_engine 6 ctx1lmnopq

5 

6 

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

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

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

10 name: str 6 ctx1fabcde

11 

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

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

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

15 default=None, 

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

17 ) 

18 

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

20 SQLModel.metadata.create_all(engine) 6 ctx1fabcde

21 create_table_log = [ 5 ctx1abcde

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 ctx1fabcde

25 

26 

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

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

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

30 default=None, 

31 sa_column_kwargs={"primary_key": True}, 

32 ) 

33 

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

35 SQLModel.metadata.create_all(engine) 6 ctx1rghijk

36 create_table_log = [ 5 ctx1ghijk

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 ctx1rghijk