Coverage for tests/test_field_sa_args_kwargs.py: 100%

1from typing import Optional 10 ctx1uvwxyzABCD

2 

3from sqlalchemy import ForeignKey 10 ctx1uvwxyzABCD

4from sqlmodel import Field, SQLModel, create_engine 10 ctx1uvwxyzABCD

5 

6 

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

8 class Team(SQLModel, table=True): 10 ctx1abcdefghij

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

10 name: str 10 ctx1abcdefghij

11 

12 class Hero(SQLModel, table=True): 10 ctx1abcdefghij

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

14 team_id: Optional[int] = Field( 10 ctx1abcdefghij

15 default=None, 

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

17 ) 

18 

19 engine = create_engine("sqlite://", echo=True) 10 ctx1abcdefghij

20 SQLModel.metadata.create_all(engine) 10 ctx1abcdefghij

21 create_table_log = [ 10 ctx1abcdefghij

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 10 ctx1abcdefghij

25 

26 

27def test_sa_column_kargs(clear_sqlmodel, caplog) -> None: 10 ctx1uvwxyzABCD

28 class Item(SQLModel, table=True): 10 ctx1klmnopqrst

29 id: Optional[int] = Field( 10 ctx1klmnopqrst

30 default=None, 

31 sa_column_kwargs={"primary_key": True}, 

32 ) 

33 

34 engine = create_engine("sqlite://", echo=True) 10 ctx1klmnopqrst

35 SQLModel.metadata.create_all(engine) 10 ctx1klmnopqrst

36 create_table_log = [ 10 ctx1klmnopqrst

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

38 ][0] 

39 assert "PRIMARY KEY (id)" in create_table_log 10 ctx1klmnopqrst