feat: fixed tests after multitype queue change

wazuh · Dec 12, 2024 · 4439615 · 4439615
1 parent db9643d
commit 4439615
Show file tree

Hide file tree

Showing 2 changed files with 555 additions and 531 deletions.
diff --git a/src/agent/multitype_queue/include/multitype_queue.hpp b/src/agent/multitype_queue/include/multitype_queue.hpp
@@ -35,11 +35,11 @@ class MultiTypeQueue : public IMultiTypeQueue
 
     // TODO: doc
     /// @brief
-    const size_t m_maxItems;
+    size_t m_maxItems;
 
     // TODO: doc
     /// @brief
-    const std::chrono::seconds m_timeout;
+    const std::chrono::milliseconds m_timeout;
 
     // TODO: doc
     /// @brief
@@ -64,8 +64,7 @@ class MultiTypeQueue : public IMultiTypeQueue
      */
     template<typename ConfigGetter>
     MultiTypeQueue(const ConfigGetter& getConfigValue)
-        : m_maxItems(config::agent::QUEUE_DEFAULT_SIZE)
-        , m_timeout(config::agent::QUEUE_STATUS_REFRESH_TIMER)
+        : m_timeout(config::agent::QUEUE_STATUS_REFRESH_TIMER)
     {
         auto dbFolderPath =
             getConfigValue.template operator()<std::string>("agent", "path.data").value_or(config::DEFAULT_DATA_PATH);
@@ -79,6 +78,9 @@ class MultiTypeQueue : public IMultiTypeQueue
             m_batchInterval = config::agent::DEFAULT_BATCH_INTERVAL;
         }
 
+        m_maxItems =
+            getConfigValue.template operator()<size_t>("agent", "queue_size").value_or(config::agent::QUEUE_DEFAULT_SIZE);
+
         const auto dbFilePath = dbFolderPath + "/" + config::agent::QUEUE_DB_NAME;
 
         try