Skip to content

Commit 573bd10

Browse files
committed
Simplify tests
1 parent 5ff875e commit 573bd10

File tree

1 file changed

+6
-17
lines changed

1 file changed

+6
-17
lines changed

tests/test_field_sa_fk_args_kwargs.py

Lines changed: 6 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import re
21
from typing import Optional
32

43
from sqlalchemy import ForeignKey, create_engine
@@ -23,14 +22,9 @@ class Document(Base, table=True):
2322
engine = create_engine("sqlite://", echo=True)
2423
SQLModel.metadata.create_all(engine)
2524

26-
fk_log = [
27-
message
28-
for message in caplog.messages
29-
if re.search(
30-
r"FOREIGN KEY\s*\(owner_id\)\s*REFERENCES\s*user\s*\(id\)", message
31-
)
32-
][0]
33-
assert "ON DELETE SET NULL" in fk_log
25+
assert (
26+
"FOREIGN KEY(owner_id) REFERENCES user (id) ON DELETE SET NULL" in caplog.text
27+
)
3428

3529

3630
def test_base_model_fk_args(clear_sqlmodel, caplog) -> None:
@@ -52,11 +46,6 @@ class Document(Base, table=True):
5246
engine = create_engine("sqlite://", echo=True)
5347
SQLModel.metadata.create_all(engine)
5448

55-
fk_log = [
56-
message
57-
for message in caplog.messages
58-
if re.search(
59-
r"FOREIGN KEY\s*\(owner_id\)\s*REFERENCES\s*user\s*\(id\)", message
60-
)
61-
][0]
62-
assert "ON DELETE SET NULL" in fk_log
49+
assert (
50+
"FOREIGN KEY(owner_id) REFERENCES user (id) ON DELETE SET NULL" in caplog.text
51+
)

0 commit comments

Comments
 (0)