File tree Expand file tree Collapse file tree 1 file changed +6
-17
lines changed Expand file tree Collapse file tree 1 file changed +6
-17
lines changed Original file line number Diff line number Diff line change 1- import re
21from typing import Optional
32
43from 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
3630def 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+ )
You can’t perform that action at this time.
0 commit comments