From 631486bce051a8ef3a8cb5ba40e975ddae8786b2 Mon Sep 17 00:00:00 2001
From: Allan Stockman Rugano <kuristockyrugano@gmail.com>
Date: Wed, 12 Jun 2024 17:38:07 +0200
Subject: [PATCH] Test fixes

---
 tests/admin/test_admin_rt.py                    |  4 ++--
 tests/extras/testutils/factories/power_query.py | 10 ++++++++--
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/tests/admin/test_admin_rt.py b/tests/admin/test_admin_rt.py
index 56b18283..ce463c26 100644
--- a/tests/admin/test_admin_rt.py
+++ b/tests/admin/test_admin_rt.py
@@ -9,9 +9,9 @@
 
 @pytest.fixture()
 def report_template():
-    from testutils.factories import ReportTemplate
+    from testutils.factories import ReportTemplateFactory
 
-    return ReportTemplate.objects.first()
+    return ReportTemplateFactory()
 
 
 @pytest.fixture()
diff --git a/tests/extras/testutils/factories/power_query.py b/tests/extras/testutils/factories/power_query.py
index 278b416e..94dfe0f9 100644
--- a/tests/extras/testutils/factories/power_query.py
+++ b/tests/extras/testutils/factories/power_query.py
@@ -2,10 +2,9 @@
 
 from django.apps import apps
 from django.core.files.base import ContentFile
-
+from django.core.files.uploadedfile import SimpleUploadedFile
 import factory
 from strategy_field.utils import fqn
-
 from hope_country_report.apps.power_query.models import (
     ChartPage,
     Dataset,
@@ -49,6 +48,13 @@ class Meta:
         model = ReportTemplate
         django_get_or_create = ("name",)
 
+    country_office = factory.SubFactory(CountryOfficeFactory)
+    name = factory.Faker("word")
+    file_suffix = ".pdf"
+    doc = factory.LazyAttribute(
+        lambda _: SimpleUploadedFile("test_template.pdf", b"Test file content", content_type="application/pdf")
+    )
+
 
 class FormatterFactory(AutoRegisterModelFactory):
     name = "Queryset To HTML"