1 files changed, 48 insertions, 0 deletions
diff --git a/tests/managers/thl/test_user_manager/test_user_fetch.py b/tests/managers/thl/test_user_manager/test_user_fetch.py
new file mode 100644
index 0000000..a4b3d57
--- /dev/null
+++ b/tests/managers/thl/test_user_manager/test_user_fetch.py
@@ -0,0 +1,48 @@
+from uuid import uuid4
+
+import pytest
+
+from generalresearch.models.thl.user import User
+from test_utils.models.conftest import product, user_manager, user_factory
+
+
+class TestUserManagerFetch:
+
+    def test_fetch(self, user_factory, product, user_manager):
+        user1: User = user_factory(product=product)
+        user2: User = user_factory(product=product)
+        res = user_manager.fetch_by_bpuids(
+            product_id=product.uuid,
+            product_user_ids=[user1.product_user_id, user2.product_user_id],
+        )
+        assert len(res) == 2
+
+        res = user_manager.fetch(user_ids=[user1.user_id, user2.user_id])
+        assert len(res) == 2
+
+        res = user_manager.fetch(user_uuids=[user1.uuid, user2.uuid])
+        assert len(res) == 2
+
+        # filter including bogus values
+        res = user_manager.fetch(user_uuids=[user1.uuid, uuid4().hex])
+        assert len(res) == 1
+
+        res = user_manager.fetch(user_uuids=[uuid4().hex])
+        assert len(res) == 0
+
+    def test_fetch_invalid(self, user_manager):
+        with pytest.raises(AssertionError) as e:
+            user_manager.fetch(user_uuids=[], user_ids=None)
+        assert "Must pass ONE of user_ids, user_uuids" in str(e.value)
+
+        with pytest.raises(AssertionError) as e:
+            user_manager.fetch(user_uuids=uuid4().hex)
+        assert "must pass a collection of user_uuids" in str(e.value)
+
+        with pytest.raises(AssertionError) as e:
+            user_manager.fetch(user_uuids=[uuid4().hex], user_ids=[1, 2, 3])
+        assert "Must pass ONE of user_ids, user_uuids" in str(e.value)
+
+        with pytest.raises(AssertionError) as e:
+            user_manager.fetch(user_ids=list(range(501)))
+        assert "limit 500 user_ids" in str(e.value)