Skip to content

Commit c3001f2

Browse files
test: avoid web log queue dependency
1 parent b46d062 commit c3001f2

File tree

1 file changed

+10
-26
lines changed

1 file changed

+10
-26
lines changed

tests/mem_scheduler/test_scheduler.py

Lines changed: 10 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)