From 1a9b4194fb19d4df3cbb8e601452466c85b42d93 Mon Sep 17 00:00:00 2001
From: Oluwanifemi Daramola <76186151+nifedara@users.noreply.github.com>
Date: Tue, 2 Apr 2024 09:08:10 +0100
Subject: [PATCH] added factory boy

---
 backend/tests/factories.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 backend/tests/factories.py

diff --git a/backend/tests/factories.py b/backend/tests/factories.py
new file mode 100644
index 00000000..eca34494
--- /dev/null
+++ b/backend/tests/factories.py
@@ -0,0 +1,21 @@
+import factory
+
+from login.models import OsmUser
+from core.models import Dataset
+
+
+class OsmUserFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = OsmUser
+
+    username = "Test User"
+    osm_id = 123456
+
+
+class DatasetFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Dataset
+
+    name = "Test Dataset"
+    created_by = factory.SubFactory(OsmUserFactory)
+    status = -1