Add test

lordgamez · Dec 11, 2024 · 1679a59 · 1679a59
1 parent ec5d63d
commit 1679a59
Show file tree

Hide file tree

Showing 4 changed files with 263 additions and 205 deletions.
diff --git a/libminifi/src/core/ProcessSession.cpp b/libminifi/src/core/ProcessSession.cpp
@@ -841,7 +841,7 @@ void ProcessSession::commit() {
   const auto commit_start_time = std::chrono::steady_clock::now();
   try {
     std::unordered_map<std::string, TransferMetrics> transfers;
-      auto increaseTransferMetrics = [&](const FlowFile& record, const Relationship& relationship) {
+    auto increaseTransferMetrics = [&](const FlowFile& record, const Relationship& relationship) {
       ++transfers[relationship.getName()].transfer_count;
       transfers[relationship.getName()].transfer_size += record.getSize();
     };

diff --git a/libminifi/test/libtest/unit/SingleProcessorTestController.h b/libminifi/test/libtest/unit/SingleProcessorTestController.h
@@ -39,7 +39,7 @@ class SingleProcessorTestController : public TestController {
  public:
   explicit SingleProcessorTestController(std::unique_ptr<core::Processor> processor) {
     auto name = processor->getName();
-    processor_ = plan->addProcessor(std::move(processor), name);
+    processor_ = plan->addProcessor(std::move(processor), name, {});
     input_ = plan->addConnection(nullptr, core::Relationship{"success", "success"}, processor_);
     outgoing_connections_ = [this] {
       std::unordered_map<core::Relationship, Connection*> result;

diff --git a/libminifi/test/libtest/unit/TestBase.cpp b/libminifi/test/libtest/unit/TestBase.cpp
@@ -539,11 +539,13 @@ bool TestPlan::runProcessor(size_t target_location, const PreTriggerVerifier& ve
 
   if (verify) {
     auto current_session = std::make_shared<minifi::core::ProcessSession>(context);
+    current_session->setMetrics(processor->getMetrics());
     process_sessions_.push_back(current_session);
     verify(context, current_session);
     current_session->commit();
   } else {
     auto session_factory = std::make_shared<TestSessionFactory>(context, [&] (auto current_session) {
+      current_session->setMetrics(processor->getMetrics());
       process_sessions_.push_back(current_session);
     });
     logger_->log_info("Running {}", processor->getName());