fix unit test error

mxsm · Nov 17, 2024 · 263db7f · 263db7f
1 parent d6ceb77
commit 263db7f
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/rocketmq-store/src/queue/queue_offset_operator.rs b/rocketmq-store/src/queue/queue_offset_operator.rs
@@ -184,22 +184,28 @@ mod tests {
     fn update_queue_offset_updates_existing_value() {
         let operator = QueueOffsetOperator::new();
         operator.increase_queue_offset("key".into(), 5);
-        operator.update_queue_offset("key".into(), 10);
+        operator.update_queue_offset(&CheetahString::from_static_str("key"), 10);
         assert_eq!(operator.get_queue_offset("key".into()), 10);
     }
 
     #[test]
     fn remove_clears_key_from_all_tables() {
         let operator = QueueOffsetOperator::new();
         operator.increase_queue_offset("topic-1".into(), 5);
-        operator.increase_batch_queue_offset("topic-1".into(), 5);
-        operator.increase_lmq_offset("topic-1".into(), 5);
+        operator.increase_batch_queue_offset(&CheetahString::from_static_str("topic-1"), 5);
+        operator.increase_lmq_offset(&CheetahString::from_static_str("topic-1"), 5);
 
-        operator.remove("topic".into(), 1);
+        operator.remove(&CheetahString::from_static_str("topic"), 1);
 
         assert_eq!(operator.get_queue_offset("topic-1".into()), 0);
-        assert_eq!(operator.get_batch_queue_offset("topic-1".into()), 0);
-        assert_eq!(operator.get_lmq_offset("topic-1".into()), 0);
+        assert_eq!(
+            operator.get_batch_queue_offset(&CheetahString::from_static_str("topic-1")),
+            0
+        );
+        assert_eq!(
+            operator.get_lmq_offset(&CheetahString::from_static_str("topic-1")),
+            0
+        );
     }
 
     #[test]