add unit tests

Signed-off-by: wiseaidev <[email protected]>

Author: wiseaidev

tests/test_hash_model.py

@@ -43,6 +43,7 @@ class Order(BaseHashModel):
         created_on: datetime.datetime
 
     class Member(BaseHashModel):
+        id: int = Field(index=True)
         first_name: str = Field(index=True)
         last_name: str = Field(index=True)
         email: str = Field(index=True)
@@ -64,6 +65,7 @@ class Meta:
 @pytest_asyncio.fixture
 async def members(m):
     member1 = m.Member(
+        id=0,
         first_name="Andrew",
         last_name="Brookins",
         email="[email protected]",
@@ -73,6 +75,7 @@ async def members(m):
     )
 
     member2 = m.Member(
+        id=1,
         first_name="Kim",
         last_name="Brookins",
         email="[email protected]",
@@ -82,6 +85,7 @@ async def members(m):
     )
 
     member3 = m.Member(
+        id=2,
         first_name="Andrew",
         last_name="Smith",
         email="[email protected]",
@@ -129,6 +133,9 @@ async def test_exact_match_queries(members, m):
     ).all()
     assert actual == [member2]
 
+    actual = await m.Member.find(m.Member.id == 0).all()
+    assert actual == [member1]
+
 
 @py_test_mark_asyncio
 async def test_full_text_search_queries(members, m):
@@ -143,6 +150,19 @@ async def test_full_text_search_queries(members, m):
     assert actual == [member1, member3]
 
 
+@py_test_mark_asyncio
+async def test_pagination_queries(members, m):
+    member1, member2, member3 = members
+
+    actual = await m.Member.find(m.Member.last_name == "Brookins").page()
+
+    assert actual == [member1, member2]
+
+    actual = await m.Member.find().page(1,2)
+
+    assert actual == [member1]
+
+
 @py_test_mark_asyncio
 async def test_recursive_query_resolution(members, m):
     member1, member2, member3 = members
@@ -611,4 +631,3 @@ class Address(m.BaseHashModel):
     assert (
         Address.redisearch_schema()
         == f"ON HASH PREFIX 1 {key_prefix} SCHEMA pk TAG SEPARATOR | a_string TAG SEPARATOR | a_full_text_string TAG SEPARATOR | a_full_text_string AS a_full_text_string_fts TEXT an_integer NUMERIC SORTABLE a_float NUMERIC"
-    )