feat: add additional test to check invalid distribution strategy

argilla-io · Jun 14, 2024 · 2082e12 · 2082e12
1 parent 6953fb7
commit 2082e12
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/argilla-server/tests/unit/api/handlers/v1/datasets/test_create_dataset.py b/argilla-server/tests/unit/api/handlers/v1/datasets/test_create_dataset.py
@@ -117,3 +117,23 @@ async def test_create_dataset_with_overlap_distribution_using_invalid_min_submit
 
         assert response.status_code == 422
         assert (await db.execute(select(func.count(Dataset.id)))).scalar_one() == 0
+
+    async def test_create_dataset_with_invalid_distribution_stategry(self, db: AsyncSession, async_client: AsyncClient, owner_auth_header: dict):
+        workspace = await WorkspaceFactory.create()
+
+        response = await async_client.post(
+            self.url(),
+            headers=owner_auth_header,
+            json={
+                "name": "Dataset Name",
+                "distribution": {
+                    "strategy": "invalid_strategy",
+                },
+                "workspace_id": str(workspace.id),
+            },
+        )
+
+        import pdb; pdb.set_trace()
+
+        assert response.status_code == 422
+        assert (await db.execute(select(func.count(Dataset.id)))).scalar_one() == 0