Adds tests for full text hash search.

This commit is contained in:
Simon Prickett 2022-01-05 18:08:49 +00:00
parent 800ec69565
commit 8d69b59916

View file

@ -47,6 +47,7 @@ async def m(key_prefix, redis):
email: str = Field(index=True) email: str = Field(index=True)
join_date: datetime.date join_date: datetime.date
age: int = Field(index=True) age: int = Field(index=True)
bio: str = Field(index=True, full_text_search=True)
class Meta: class Meta:
model_key_prefix = "member" model_key_prefix = "member"
@ -67,6 +68,7 @@ async def members(m):
email="a@example.com", email="a@example.com",
age=38, age=38,
join_date=today, join_date=today,
bio="This is member 1 whose greatness makes him the life and soul of any party he goes to.",
) )
member2 = m.Member( member2 = m.Member(
@ -75,6 +77,7 @@ async def members(m):
email="k@example.com", email="k@example.com",
age=34, age=34,
join_date=today, join_date=today,
bio="This is member 2 who can be quite anxious until you get to know them.",
) )
member3 = m.Member( member3 = m.Member(
@ -83,6 +86,7 @@ async def members(m):
email="as@example.com", email="as@example.com",
age=100, age=100,
join_date=today, join_date=today,
bio="This is member 3 who is a funny and lively sort of person.",
) )
await member1.save() await member1.save()
await member2.save() await member2.save()
@ -124,6 +128,21 @@ async def test_exact_match_queries(members, m):
).all() ).all()
assert actual == [member2] assert actual == [member2]
@pytest.mark.asyncio
async def test_full_text_search_queries(members, m):
member1, member2, member3 = members
actual = await (
m.Member.find(m.Member.bio % "great").all()
)
assert actual == [member1]
actual = await (
m.Member.find(~(m.Member.bio % "anxious")).all()
)
assert actual == [member1, member3]
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_recursive_query_resolution(members, m): async def test_recursive_query_resolution(members, m):
@ -163,6 +182,7 @@ async def test_tag_queries_punctuation(m):
email="a|b@example.com", # NOTE: This string uses the TAG field separator. email="a|b@example.com", # NOTE: This string uses the TAG field separator.
age=38, age=38,
join_date=today, join_date=today,
bio="This is a test user on our system.",
) )
await member1.save() await member1.save()
@ -172,6 +192,7 @@ async def test_tag_queries_punctuation(m):
email="a|villain@example.com", # NOTE: This string uses the TAG field separator. email="a|villain@example.com", # NOTE: This string uses the TAG field separator.
age=38, age=38,
join_date=today, join_date=today,
bio="This is a villain, they are a really bad person!",
) )
await member2.save() await member2.save()
@ -334,6 +355,7 @@ def test_validation_passes(m):
email="a@example.com", email="a@example.com",
join_date=today, join_date=today,
age=38, age=38,
bio="This is the bio field.",
) )
assert member.first_name == "Andrew" assert member.first_name == "Andrew"
@ -346,6 +368,7 @@ async def test_saves_model_and_creates_pk(m):
email="a@example.com", email="a@example.com",
join_date=today, join_date=today,
age=38, age=38,
bio="This is the bio field for this user.",
) )
# Save a model instance to Redis # Save a model instance to Redis
await member.save() await member.save()
@ -408,6 +431,7 @@ async def test_saves_many(m):
email="a@example.com", email="a@example.com",
join_date=today, join_date=today,
age=38, age=38,
bio="This is the user bio.",
) )
member2 = m.Member( member2 = m.Member(
first_name="Kim", first_name="Kim",
@ -415,6 +439,7 @@ async def test_saves_many(m):
email="k@example.com", email="k@example.com",
join_date=today, join_date=today,
age=34, age=34,
bio="This is the bio for Kim.",
) )
members = [member1, member2] members = [member1, member2]
result = await m.Member.add(members) result = await m.Member.add(members)