From 5552e698be67178b678806ca191bc15fd8c87ac2 Mon Sep 17 00:00:00 2001
From: "IGNY8 VPS (Salman)" <vps@igny8.com>
Date: Tue, 18 Nov 2025 02:23:01 +0000
Subject: [PATCH] Add credits to account in test setup for content generation,
 linking, and optimization tests

- Updated test cases in `test_universal_content_types.py`, `test_universal_content_linking.py`, and `test_universal_content_optimization.py` to initialize account credits for testing.
- Changed mock patch references to `run_ai_task` for consistency across tests.
---
 backend/celerybeat-schedule                   | Bin 16384 -> 16384 bytes
 .../tests/test_universal_content_types.py     |   9 ++++++---
 .../tests/test_universal_content_linking.py   |   3 +++
 .../test_universal_content_optimization.py    |   3 +++
 4 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/backend/celerybeat-schedule b/backend/celerybeat-schedule
index 9c1602177064b50319f5c99b0013643d544a8628..b71e2ea81643dcd01eb4a5be3a81d3c60165f16e 100644
GIT binary patch
delta 29
kcmZo@U~Fh$+@NH_$IrmPU^gX0v~5bz6z_t~1|}D{0e`{?A^-pY

delta 29
jcmZo@U~Fh$+@NH_#{dKireuh=O$nOfox9n<<N`MUgi#4{

diff --git a/backend/igny8_core/business/content/tests/test_universal_content_types.py b/backend/igny8_core/business/content/tests/test_universal_content_types.py
index bc0f75ab..12afa558 100644
--- a/backend/igny8_core/business/content/tests/test_universal_content_types.py
+++ b/backend/igny8_core/business/content/tests/test_universal_content_types.py
@@ -14,9 +14,12 @@ class UniversalContentTypesTests(IntegrationTestBase):
     
     def setUp(self):
         super().setUp()
+        # Add credits to account for testing
+        self.account.credits = 10000
+        self.account.save()
         self.service = ContentGenerationService()
     
-    @patch('igny8_core.business.content.services.content_generation_service.run_ai_task')
+    @patch('igny8_core.ai.tasks.run_ai_task')
     def test_product_content_generates_correctly(self, mock_run_ai_task):
         """
         Test: Product content generates correctly
@@ -55,7 +58,7 @@ class UniversalContentTypesTests(IntegrationTestBase):
         self.assertEqual(call_args[1]['function_name'], 'generate_product_content')
         self.assertEqual(call_args[1]['payload']['product_name'], 'Test Product')
     
-    @patch('igny8_core.business.content.services.content_generation_service.run_ai_task')
+    @patch('igny8_core.ai.tasks.run_ai_task')
     def test_service_pages_work_correctly(self, mock_run_ai_task):
         """
         Test: Service pages work correctly
@@ -94,7 +97,7 @@ class UniversalContentTypesTests(IntegrationTestBase):
         self.assertEqual(call_args[1]['function_name'], 'generate_service_page')
         self.assertEqual(call_args[1]['payload']['service_name'], 'Test Service')
     
-    @patch('igny8_core.business.content.services.content_generation_service.run_ai_task')
+    @patch('igny8_core.ai.tasks.run_ai_task')
     def test_taxonomy_pages_work_correctly(self, mock_run_ai_task):
         """
         Test: Taxonomy pages work correctly
diff --git a/backend/igny8_core/business/linking/tests/test_universal_content_linking.py b/backend/igny8_core/business/linking/tests/test_universal_content_linking.py
index fd79b5f8..38b2cd67 100644
--- a/backend/igny8_core/business/linking/tests/test_universal_content_linking.py
+++ b/backend/igny8_core/business/linking/tests/test_universal_content_linking.py
@@ -14,6 +14,9 @@ class UniversalContentLinkingTests(IntegrationTestBase):
     
     def setUp(self):
         super().setUp()
+        # Add credits to account for testing
+        self.account.credits = 10000
+        self.account.save()
         self.linker_service = LinkerService()
         
         # Create product content
diff --git a/backend/igny8_core/business/optimization/tests/test_universal_content_optimization.py b/backend/igny8_core/business/optimization/tests/test_universal_content_optimization.py
index c6025342..40acaa19 100644
--- a/backend/igny8_core/business/optimization/tests/test_universal_content_optimization.py
+++ b/backend/igny8_core/business/optimization/tests/test_universal_content_optimization.py
@@ -15,6 +15,9 @@ class UniversalContentOptimizationTests(IntegrationTestBase):
     
     def setUp(self):
         super().setUp()
+        # Add credits to account for testing
+        self.account.credits = 10000
+        self.account.save()
         self.optimizer_service = OptimizerService()
         
         # Create product content