ppf/new: Add asignment of *_id values returned by add_* methods.

diff --git a/ppf/new/top.py b/ppf/new/top.py
index b74aace..17c0804 100644 (file)
--- a/ppf/new/top.py
+++ b/ppf/new/top.py
@@ -36,11 +36,13 @@ def enhance_swarm_description(swarm_description, swarm_writer):
     swarm_writer is an instance of storage.SwarmWriter.
     """
     swarm = swarm_description.get_swarm()
-    swarm_writer.add_swarm(swarm)
+    swarm_id = swarm_writer.add_swarm(swarm)
 
     for session_entry in swarm_description.get_session_entries():
-        session = session_entry.get_sssion()
-        session_id = swarm_writer.add_session(session)
+        session = session_entry.get_session()
+        # Add swarm_id to session
+        session.swarm_id = swarm_id
+        session_id = swarm_writer.add_client_session(session)
         # Update session_id in swarm_description list.
         swarm_description.update_session_entry_id(session_entry, session_id)