ideas

tests/test_idea.py

@@ -27,14 +27,23 @@ async def test_idea_flow():
     print("Creating idea...")
     user_id = "test_user_123"
     project_id = "test_project_abc"
-    idea = await service.create_idea("My Test Idea", project_id, user_id)
+    idea = await service.create_idea("My Test Idea", "Initial Description", project_id, user_id)
     print(f"Idea created: {idea.id} - {idea.name}")
     
-    # 2. Add Generation linked to Idea
+    # 2. Update Idea
+    print("Updating idea...")
+    updated_idea = await service.update_idea(idea.id, description="Updated description")
+    print(f"Idea updated: {updated_idea.description}")
+    if updated_idea.description == "Updated description":
+        print("✅ Idea update successful")
+    else:
+        print("❌ Idea update FAILED")
+
+    # 3. Add Generation linked to Idea
     print("Creating generation linked to idea...")
     gen = Generation(
         prompt="idea generation 1",
-        idea_id=idea.id,
+        # idea_id=idea.id,  <-- Intentionally NOT linking initially to test linking method
         project_id=project_id,
         created_by=user_id,
         aspect_ratio=AspectRatios.NINESIXTEEN,
@@ -42,15 +51,23 @@ async def test_idea_flow():
         assets_list=[]
     )
     gen_id = await dao.generations.create_generation(gen)
-    print(f"Created linked generation: {gen_id}")
+    print(f"Created generation: {gen_id}")
+    
+    # Link generation to idea
+    print("Linking generation to idea...")
+    success = await service.add_generation_to_idea(idea.id, gen_id)
+    if success:
+         print("✅ Linking successful")
+    else:
+         print("❌ Linking FAILED")
     
     # Debug: Check if generation was saved with idea_id
     saved_gen = await dao.generations.collection.find_one({"_id": ObjectId(gen_id)})
     print(f"DEBUG: Saved Generation in DB idea_id: {saved_gen.get('idea_id')}")
     
-    # 3. Fetch Generations for Idea (Verify filtering and ordering)
+    # 4. Fetch Generations for Idea (Verify filtering and ordering)
     print("Fetching generations for idea...")
-    gens = await dao.generations.get_generations(idea_id=idea.id)
+    gens = await service.dao.generations.get_generations(idea_id=idea.id) # using repo directly as service might return wrapper
     print(f"Found {len(gens)} generations in idea")
     
     if len(gens) == 1 and gens[0].id == gen_id:
@@ -58,7 +75,7 @@ async def test_idea_flow():
     else:
         print("❌ Generation retrieval FAILED")
 
-    # 4. Fetch Ideas for Project
+    # 5. Fetch Ideas for Project
     ideas = await service.get_ideas(project_id)
     print(f"Found {len(ideas)} ideas for project")