Fix agency code genderation

HHS · Dec 3, 2024 · 62996c7 · 62996c7
1 parent bf36855
commit 62996c7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/api/tests/src/db/models/factories.py b/api/tests/src/db/models/factories.py
@@ -809,8 +809,7 @@ def _setup_next_sequence(cls):
     agency_id = factory.Sequence(lambda n: n)
     agency_name = factory.Faker("agency_name")
 
-    # Make agency_code unique by including the sequence number
-    agency_code = factory.Sequence(lambda n: f"AG-{n:04d}")
+    agency_code = factory.Iterator(CustomProvider.AGENCIES)
 
     sub_agency_code = factory.LazyAttribute(lambda a: a.agency_code.split("-")[0])