From 91d040211a4ed6e4157896256a762d3854777b5e Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Fri, 6 Mar 2026 16:49:46 -0500
Subject: Initial commit

---
 tests/managers/thl/test_buyer.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 tests/managers/thl/test_buyer.py

(limited to 'tests/managers/thl/test_buyer.py')

diff --git a/tests/managers/thl/test_buyer.py b/tests/managers/thl/test_buyer.py
new file mode 100644
index 0000000..69ea105
--- /dev/null
+++ b/tests/managers/thl/test_buyer.py
@@ -0,0 +1,25 @@
+from generalresearch.models import Source
+
+
+class TestBuyer:
+
+    def test(
+        self,
+        delete_buyers_surveys,
+        buyer_manager,
+    ):
+
+        bs = buyer_manager.bulk_get_or_create(source=Source.TESTING, codes=["a", "b"])
+        assert len(bs) == 2
+        buyer_a = bs[0]
+        assert buyer_a.id is not None
+        bs2 = buyer_manager.bulk_get_or_create(source=Source.TESTING, codes=["a", "c"])
+        assert len(bs2) == 2
+        buyer_a2 = bs2[0]
+        buyer_c = bs2[1]
+        # a isn't created again
+        assert buyer_a == buyer_a2
+        assert bs2[0].id is not None
+
+        # and its cached
+        assert buyer_c.id == buyer_manager.source_code_pk[f"{Source.TESTING.value}:c"]
-- 
cgit v1.2.3