Skip to content

Commit 8c12adc

Browse files
authored
Fix TSVectorField returned value conversion (#2237)
1 parent af34316 commit 8c12adc

2 files changed

Lines changed: 12 additions & 1 deletion

File tree

tests/contrib/postgres/test_tsvector_field.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,3 +22,13 @@ async def test_tsvector_generated_sql(db_tsvector):
2222
"GENERATED ALWAYS AS (SETWEIGHT(TO_TSVECTOR('english',COALESCE(\"title\", '')),'A')"
2323
" || SETWEIGHT(TO_TSVECTOR('english',COALESCE(\"body\", '')),'B')) STORED"
2424
)
25+
26+
27+
@pytest.mark.asyncio
28+
async def test_tsvector_generated_field_save(db_tsvector):
29+
"""Test TSVector generated fields can be returned after insert."""
30+
skip_if_not_postgres()
31+
entry = await TSVectorEntry.create(title="Tortoise ORM", body="Async Python ORM")
32+
33+
assert isinstance(entry.search_vector, str)
34+
assert entry.search_vector

tortoise/contrib/postgres/fields.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,9 @@
77
from tortoise.fields import Field
88

99

10-
class TSVectorField(Field):
10+
class TSVectorField(Field[str]):
1111
SQL_TYPE = "TSVECTOR"
12+
field_type = str
1213
allows_generated = True
1314

1415
def __init__(

0 commit comments

Comments
 (0)