redis-om-python/tests/test_hash_model.py

411 lines
11 KiB
Python
Raw Normal View History

import abc
2021-09-01 21:56:06 +02:00
import datetime
import decimal
from typing import Optional
2021-09-30 05:23:39 +02:00
from unittest import mock
2021-09-01 21:56:06 +02:00
import pytest
import redis
from pydantic import ValidationError
from redis_developer.model import Field, HashModel
from redis_developer.model.model import (
NotFoundError,
QueryNotSupportedError,
RedisModelError,
2021-09-01 21:56:06 +02:00
)
2021-09-01 21:56:06 +02:00
r = redis.Redis()
today = datetime.date.today()
class BaseHashModel(HashModel, abc.ABC):
class Meta:
2021-09-01 21:56:06 +02:00
global_key_prefix = "redis-developer"
class Order(BaseHashModel):
total: decimal.Decimal
currency: str
created_on: datetime.datetime
class Member(BaseHashModel):
2021-09-21 01:06:04 +02:00
first_name: str = Field(index=True)
last_name: str = Field(index=True)
email: str = Field(index=True)
2021-09-01 21:56:06 +02:00
join_date: datetime.date
2021-09-21 01:06:04 +02:00
age: int = Field(index=True)
2021-09-01 21:56:06 +02:00
class Meta:
2021-09-01 21:56:06 +02:00
model_key_prefix = "member"
primary_key_pattern = ""
@pytest.fixture()
def members():
member1 = Member(
first_name="Andrew",
last_name="Brookins",
email="a@example.com",
age=38,
join_date=today,
)
member2 = Member(
first_name="Kim",
last_name="Brookins",
email="k@example.com",
age=34,
join_date=today,
)
member3 = Member(
first_name="Andrew",
last_name="Smith",
email="as@example.com",
age=100,
join_date=today,
)
member1.save()
member2.save()
member3.save()
yield member1, member2, member3
2021-09-01 21:56:06 +02:00
def test_validates_required_fields():
# Raises ValidationError: last_name is required
2021-09-01 21:56:06 +02:00
with pytest.raises(ValidationError):
Member(first_name="Andrew", zipcode="97086", join_date=today)
2021-09-01 21:56:06 +02:00
def test_validates_field():
# Raises ValidationError: join_date is not a date
with pytest.raises(ValidationError):
Member(first_name="Andrew", last_name="Brookins", join_date="yesterday")
2021-09-01 21:56:06 +02:00
# Passes validation
def test_validation_passes():
member = Member(
first_name="Andrew",
last_name="Brookins",
email="a@example.com",
join_date=today,
age=38,
2021-09-01 21:56:06 +02:00
)
assert member.first_name == "Andrew"
def test_saves_model_and_creates_pk():
member = Member(
first_name="Andrew",
last_name="Brookins",
email="a@example.com",
join_date=today,
age=38,
2021-09-01 21:56:06 +02:00
)
# Save a model instance to Redis
member.save()
member2 = Member.get(member.pk)
assert member2 == member
def test_raises_error_with_embedded_models():
class Address(BaseHashModel):
address_line_1: str
address_line_2: Optional[str]
city: str
country: str
postal_code: str
2021-09-01 21:56:06 +02:00
with pytest.raises(RedisModelError):
2021-09-01 21:56:06 +02:00
class InvalidMember(BaseHashModel):
address: Address
@pytest.mark.skip("Not implemented yet")
def test_saves_many():
members = [
Member(
first_name="Andrew",
last_name="Brookins",
email="a@example.com",
join_date=today,
2021-09-01 21:56:06 +02:00
),
Member(
first_name="Kim",
last_name="Brookins",
email="k@example.com",
join_date=today,
),
2021-09-01 21:56:06 +02:00
]
Member.add(members)
@pytest.mark.skip("Not ready yet")
def test_updates_a_model(members):
member1, member2, member3 = members
2021-09-01 21:56:06 +02:00
# Or, with an implicit save:
member1.update(last_name="Smith")
assert Member.find(Member.pk == member1.pk).first() == member1
2021-09-01 21:56:06 +02:00
# Or, affecting multiple model instances with an implicit save:
2021-09-16 21:03:03 +02:00
Member.find(Member.last_name == "Brookins").update(last_name="Smith")
results = Member.find(Member.last_name == "Smith")
assert results == members
def test_paginate_query(members):
member1, member2, member3 = members
actual = Member.find().all(batch_size=1)
assert actual == [member1, member2, member3]
def test_access_result_by_index_cached(members):
2021-09-30 05:23:39 +02:00
member1, member2, member3 = members
query = Member.find().sort_by("age")
# Load the cache, throw away the result.
2021-09-30 05:23:39 +02:00
assert query._model_cache == []
query.execute()
2021-09-30 05:23:39 +02:00
assert query._model_cache == [member2, member1, member3]
# Access an item that should be in the cache.
with mock.patch.object(query.model, "db") as mock_db:
2021-09-30 05:23:39 +02:00
assert query[0] == member2
assert not mock_db.called
def test_access_result_by_index_not_cached(members):
member1, member2, member3 = members
query = Member.find().sort_by("age")
# Assert that we don't have any models in the cache yet -- we
# haven't made any requests of Redis.
assert query._model_cache == []
assert query[0] == member2
assert query[1] == member1
assert query[2] == member3
def test_exact_match_queries(members):
member1, member2, member3 = members
2021-09-16 21:03:03 +02:00
actual = Member.find(Member.last_name == "Brookins").all()
assert actual == [member1, member2]
2021-09-16 21:03:03 +02:00
actual = Member.find(
(Member.last_name == "Brookins") & ~(Member.first_name == "Andrew")
).all()
2021-09-16 21:03:03 +02:00
assert actual == [member2]
actual = Member.find(~(Member.last_name == "Brookins")).all()
2021-09-16 21:03:03 +02:00
assert actual == [member3]
actual = Member.find(Member.last_name != "Brookins").all()
2021-09-16 21:03:03 +02:00
assert actual == [member3]
2021-09-16 23:35:25 +02:00
actual = Member.find(
(Member.last_name == "Brookins") & (Member.first_name == "Andrew")
| (Member.first_name == "Kim")
).all()
assert actual == [member1, member2]
2021-09-16 21:03:03 +02:00
actual = Member.find(
Member.first_name == "Kim", Member.last_name == "Brookins"
).all()
assert actual == [member2]
2021-09-21 01:06:04 +02:00
def test_recursive_query_resolution(members):
member1, member2, member3 = members
actual = Member.find(
(Member.last_name == "Brookins")
| (Member.age == 100) & (Member.last_name == "Smith")
).all()
assert actual == [member1, member2, member3]
2021-09-21 01:06:04 +02:00
def test_tag_queries_boolean_logic(members):
member1, member2, member3 = members
actual = Member.find(
(Member.first_name == "Andrew") & (Member.last_name == "Brookins")
| (Member.last_name == "Smith")
).all()
assert actual == [member1, member3]
2021-09-21 01:06:04 +02:00
def test_tag_queries_punctuation():
member1 = Member(
first_name="Andrew, the Michael",
last_name="St. Brookins-on-Pier",
email="a|b@example.com", # NOTE: This string uses the TAG field separator.
age=38,
join_date=today,
)
member1.save()
member2 = Member(
first_name="Bob",
last_name="the Villain",
email="a|villain@example.com", # NOTE: This string uses the TAG field separator.
age=38,
join_date=today,
)
member2.save()
assert Member.find(Member.first_name == "Andrew, the Michael").first() == member1
assert Member.find(Member.last_name == "St. Brookins-on-Pier").first() == member1
# Notice that when we index and query multiple values that use the internal
# TAG separator for single-value exact-match fields, like an indexed string,
# the queries will succeed. We apply a workaround that queries for the union
# of the two values separated by the tag separator.
assert Member.find(Member.email == "a|b@example.com").all() == [member1]
assert Member.find(Member.email == "a|villain@example.com").all() == [member2]
2021-09-21 01:06:04 +02:00
def test_tag_queries_negation(members):
member1, member2, member3 = members
"""
first_name
NOT EQ
Andrew
"""
query = Member.find(~(Member.first_name == "Andrew"))
assert query.all() == [member2]
"""
first_name
NOT EQ
| Andrew
AND
| last_name
EQ
Brookins
"""
query = Member.find(
~(Member.first_name == "Andrew") & (Member.last_name == "Brookins")
)
assert query.all() == [member2]
"""
first_name
NOT EQ
| Andrew
AND
| last_name
| EQ
| | Brookins
OR
| last_name
EQ
Smith
"""
query = Member.find(
~(Member.first_name == "Andrew")
& ((Member.last_name == "Brookins") | (Member.last_name == "Smith"))
)
assert query.all() == [member2]
"""
first_name
NOT EQ
| Andrew
AND
| | last_name
| EQ
| Brookins
OR
| last_name
EQ
Smith
"""
query = Member.find(
~(Member.first_name == "Andrew") & (Member.last_name == "Brookins")
| (Member.last_name == "Smith")
)
assert query.all() == [member2, member3]
2021-09-21 01:06:04 +02:00
actual = Member.find(
(Member.first_name == "Andrew") & ~(Member.last_name == "Brookins")
).all()
assert actual == [member3]
2021-09-21 01:06:04 +02:00
def test_numeric_queries(members):
member1, member2, member3 = members
actual = Member.find(Member.age == 34).all()
assert actual == [member2]
actual = Member.find(Member.age > 34).all()
assert actual == [member1, member3]
actual = Member.find(Member.age < 35).all()
assert actual == [member2]
actual = Member.find(Member.age <= 34).all()
assert actual == [member2]
actual = Member.find(Member.age >= 100).all()
assert actual == [member3]
2021-10-14 02:16:20 +02:00
actual = Member.find(Member.age != 34).all()
assert actual == [member1, member3]
actual = Member.find(~(Member.age == 100)).all()
assert actual == [member1, member2]
2021-10-14 02:16:20 +02:00
actual = Member.find(Member.age > 30, Member.age < 40).all()
assert actual == [member1, member2]
2021-09-30 05:23:39 +02:00
def test_sorting(members):
member1, member2, member3 = members
actual = Member.find(Member.age > 34).sort_by("age").all()
assert actual == [member1, member3]
2021-09-30 05:23:39 +02:00
actual = Member.find(Member.age > 34).sort_by("-age").all()
assert actual == [member3, member1]
2021-09-30 05:23:39 +02:00
with pytest.raises(QueryNotSupportedError):
# This field does not exist.
Member.find().sort_by("not-a-real-field").all()
2021-09-30 05:23:39 +02:00
with pytest.raises(QueryNotSupportedError):
# This field is not sortable.
Member.find().sort_by("join_date").all()
2021-09-30 05:23:39 +02:00
2021-09-30 05:32:44 +02:00
def test_not_found():
with pytest.raises(NotFoundError):
2021-09-30 05:32:44 +02:00
# This ID does not exist.
Member.get(1000)
def test_schema():
class Address(BaseHashModel):
a_string: str = Field(index=True)
a_full_text_string: str = Field(index=True, full_text_search=True)
an_integer: int = Field(index=True, sortable=True)
a_float: float = Field(index=True)
another_integer: int
another_float: float
assert (
Address.redisearch_schema()
== "ON HASH PREFIX 1 redis-developer:tests.test_hash_model.Address: SCHEMA pk TAG SEPARATOR | a_string TAG SEPARATOR | a_full_text_string TAG SEPARATOR | a_full_text_string_fts TEXT an_integer NUMERIC SORTABLE a_float NUMERIC"
)