bismark populator improvement...

jmchilton · Jan 31, 2023 · 6e4c029 · 6e4c029
1 parent 9c6ce31
commit 6e4c029
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/lib/tool_shed/test/base/populators.py b/lib/tool_shed/test/base/populators.py
@@ -92,8 +92,16 @@ def __init__(self, admin_api_interactor: ShedApiInteractor, api_interactor: Shed
         self._admin_api_interactor = admin_api_interactor
         self._api_interactor = api_interactor
 
-    def setup_bismark_repo(self, repository_id: HasRepositoryId, end: Optional[int] = None):
-        self.setup_test_data_repo_by_id("bismark", repository_id, assert_ok=False, end=end)
+    def setup_bismark_repo(
+        self,
+        repository_id: Optional[HasRepositoryId] = None,
+        end: Optional[int] = None,
+        category_id: Optional[str] = None,
+    ) -> HasRepositoryId:
+        if repository_id is None:
+            category_id = category_id or self.new_category(prefix="testbismark").id
+            repository_id = self.new_repository(category_id, prefix="testbismark")
+        return self.setup_test_data_repo_by_id("bismark", repository_id, assert_ok=False, end=end)
 
     def setup_test_data_repo_by_id(
         self,