test_collector_registry_integration.cpp

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2021-2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#include <gtest/gtest.h>
 
#include "kcenon/monitoring/factory/builtin_collectors.h"
#include "kcenon/monitoring/plugins/collector_registry.h"
 
using namespace kcenon::monitoring;
 
class CollectorRegistryIntegrationTest : public ::testing::Test {
protected:
    void SetUp() override {
        // Clean registry before each test
        auto& registry = collector_registry::instance();
        registry.clear();
    }
 
    void TearDown() override {
        // Clean up after test
        auto& registry = collector_registry::instance();
        registry.clear();
    }
};
 
// Test that all builtin collectors are registered with the registry
TEST_F(CollectorRegistryIntegrationTest, BuiltinCollectorsRegisteredWithRegistry) {
    auto& registry = collector_registry::instance();
 
    // Register all builtin collectors
    bool success = register_builtin_collectors();
    EXPECT_TRUE(success);
 
    // Check that we have 9 plugin-based collectors registered
    // (battery, uptime, interrupt, network_metrics, platform_metrics,
    //  process_metrics, security, smart, vm)
    auto total_plugins = registry.plugin_count();
    EXPECT_EQ(total_plugins, 9);
}
 
// Test that get_plugins returns all registered collectors
TEST_F(CollectorRegistryIntegrationTest, GetPluginsReturnsAllCollectors) {
    auto& registry = collector_registry::instance();
 
    // Register builtin collectors
    register_builtin_collectors();
 
    // Try to get a specific plugin first (safer)
    auto* battery = registry.get_plugin("battery_collector");
 
    // If battery collector is available on this platform
    if (battery != nullptr) {
        // Check that it has a valid name (the actual implementation uses "battery")
        EXPECT_FALSE(std::string(battery->name()).empty());
    }
 
    // Check plugin count
    auto total_plugins = registry.plugin_count();
    EXPECT_EQ(total_plugins, 9);
}
 
// Test registry statistics
TEST_F(CollectorRegistryIntegrationTest, RegistryStatistics) {
    auto& registry = collector_registry::instance();
 
    // Register builtin collectors
    register_builtin_collectors();
 
    // Get statistics (without triggering full instantiation)
    auto stats = registry.get_registry_stats();
 
    // Should have total_plugins stat
    EXPECT_TRUE(stats.find("total_plugins") != stats.end());
}
 
// Test that specific collectors can be retrieved
TEST_F(CollectorRegistryIntegrationTest, GetSpecificCollector) {
    auto& registry = collector_registry::instance();
 
    // Register builtin collectors
    register_builtin_collectors();
 
    // Try to get battery_collector
    auto* battery = registry.get_plugin("battery_collector");
 
    // Should either be valid or null (if not available on platform)
    if (battery != nullptr) {
        // Check that it has a valid name
        EXPECT_FALSE(std::string(battery->name()).empty());
    }
}
 
// Test initialization of all plugins
TEST_F(CollectorRegistryIntegrationTest, InitializeAllPlugins) {
    auto& registry = collector_registry::instance();
 
    // Register builtin collectors
    register_builtin_collectors();
 
    // Get a specific plugin
    auto* uptime = registry.get_plugin("uptime_collector");
 
    if (uptime != nullptr) {
        // Initialize this specific plugin
        config_map config;
        bool init_result = uptime->initialize(config);
        EXPECT_TRUE(init_result);
    }
}
 
// Test that has_plugin works correctly
TEST_F(CollectorRegistryIntegrationTest, HasPlugin) {
    auto& registry = collector_registry::instance();
 
    // Register builtin collectors
    register_builtin_collectors();
 
    // Should have battery_collector registered
    EXPECT_TRUE(registry.has_plugin("battery_collector"));
 
    // Should have uptime_collector registered
    EXPECT_TRUE(registry.has_plugin("uptime_collector"));
 
    // Should not have a non-existent collector
    EXPECT_FALSE(registry.has_plugin("non_existent_collector"));
}