@@ -139,37 +139,21 @@ def test_submit_web_logs(self):
139139 },
140140 )
141141
142+ self .scheduler .rabbitmq_config = MagicMock ()
143+ self .scheduler .rabbitmq_publish_message = MagicMock ()
144+
142145 # Submit the log message
143146 self .scheduler ._submit_web_logs (messages = log_message )
144147
145- # Get the actual message from the queue
146- actual_message = self .scheduler ._web_log_message_queue .get ()
147-
148- # Verify core fields
149- self .assertEqual (actual_message .user_id , "test_user" )
150- self .assertEqual (actual_message .mem_cube_id , "test_cube" )
151- self .assertEqual (actual_message .label , QUERY_TASK_LABEL )
152- self .assertEqual (actual_message .from_memory_type , "WorkingMemory" )
153- self .assertEqual (actual_message .to_memory_type , "LongTermMemory" )
154- self .assertEqual (actual_message .log_content , "Test Content" )
155-
156- # Verify memory sizes
157- self .assertEqual (actual_message .current_memory_sizes ["long_term_memory_size" ], 0 )
158- self .assertEqual (actual_message .current_memory_sizes ["user_memory_size" ], 0 )
159- self .assertEqual (actual_message .current_memory_sizes ["working_memory_size" ], 0 )
160- self .assertEqual (actual_message .current_memory_sizes ["transformed_act_memory_size" ], 0 )
161-
162- # Verify memory capacities
163- self .assertEqual (actual_message .memory_capacities ["long_term_memory_capacity" ], 1000 )
164- self .assertEqual (actual_message .memory_capacities ["user_memory_capacity" ], 500 )
165- self .assertEqual (actual_message .memory_capacities ["working_memory_capacity" ], 100 )
166- self .assertEqual (actual_message .memory_capacities ["transformed_act_memory_capacity" ], 0 )
148+ self .scheduler .rabbitmq_publish_message .assert_called_once_with (
149+ message = log_message .to_dict ()
150+ )
167151
168152 # Verify auto-generated fields exist
169- self .assertTrue (hasattr (actual_message , "item_id" ))
170- self .assertTrue (isinstance (actual_message .item_id , str ))
171- self .assertTrue (hasattr (actual_message , "timestamp" ))
172- self .assertTrue (isinstance (actual_message .timestamp , datetime ))
153+ self .assertTrue (hasattr (log_message , "item_id" ))
154+ self .assertTrue (isinstance (log_message .item_id , str ))
155+ self .assertTrue (hasattr (log_message , "timestamp" ))
156+ self .assertTrue (isinstance (log_message .timestamp , datetime ))
173157
174158 def test_activation_memory_update (self ):
175159 """Test activation memory update functionality with DynamicCache handling."""
0 commit comments