add basic all test and count assertions

Author: tomcounsell

tests/test_hash_model.py

@@ -96,6 +96,18 @@ async def members(m):
     yield member1, member2, member3
 
 
+@py_test_mark_asyncio
+async def test_all_query(members, m):
+
+    actual = await m.Member.find().all()
+    assert all([member in actual for member in members])
+
+    actual_count = await m.Member.find().count()
+    assert actual_count == len(members)
+
+    assert False  # pls just fail
+
+
 @py_test_mark_asyncio
 async def test_exact_match_queries(members, m):
     member1, member2, member3 = members
@@ -129,6 +141,11 @@ async def test_exact_match_queries(members, m):
     ).all()
     assert actual == [member2]
 
+    actual_count = await m.Member.find(
+        m.Member.first_name == "Kim", m.Member.last_name == "Brookins"
+    ).count()
+    assert actual_count == 1
+
 
 @py_test_mark_asyncio
 async def test_full_text_search_queries(members, m):
@@ -162,16 +179,17 @@ async def test_recursive_query_resolution(members, m):
 async def test_tag_queries_boolean_logic(members, m):
     member1, member2, member3 = members
 
-    actual = await (
-        m.Member.find(
-            (m.Member.first_name == "Andrew") & (m.Member.last_name == "Brookins")
-            | (m.Member.last_name == "Smith")
-        )
-        .sort_by("age")
-        .all()
-    )
+    find_query = m.Member.find(
+        (m.Member.first_name == "Andrew") & (m.Member.last_name == "Brookins")
+        | (m.Member.last_name == "Smith")
+    ).sort_by("age")
+
+    actual = await find_query.all()
     assert actual == [member1, member3]
 
+    actual_count = await find_query.count()
+    assert actual_count == 2
+
 
 @py_test_mark_asyncio
 async def test_tag_queries_punctuation(m):