More ExternalIPsConfig unit tests

Co-authored-by: JoukoVirtanen <jouko@stackrox.com>

Author: ovalenti

collector/lib/ExternalIPsConfig.h

@@ -41,4 +41,4 @@ class ExternalIPsConfig {
 
 std::ostream& operator<<(std::ostream& os, const ExternalIPsConfig& config);
 
-}  // end namespace collector
+}  // end namespace collector

collector/test/CollectorConfigTest.cpp

@@ -4,6 +4,7 @@
 
 #include "CollectorArgs.h"
 #include "CollectorConfig.h"
+#include "ConfigLoader.h"
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
 
@@ -155,4 +156,91 @@ TEST(CollectorConfigTest, TestEnableExternalIpsRuntimeConfig) {
   EXPECT_TRUE(config.GetExternalIPsConf().IsEnabled(Direction::BOTH));
 }
 
+TEST(CollectorConfigTest, TestIsEnabledIngress) {
+  std::string yamlStr = R"(
+                networking:
+                  externalIps:
+                    enabled: enabled
+                    direction: ingress
+             )";
+
+  YAML::Node yamlNode = YAML::Load(yamlStr);
+  CollectorConfig config;
+  ASSERT_EQ(ConfigLoader(config).LoadConfiguration(yamlNode), ConfigLoader::SUCCESS) << "Input: " << yamlStr;
+
+  auto runtime_config = config.GetRuntimeConfig();
+
+  EXPECT_TRUE(runtime_config.has_value());
+
+  EXPECT_EQ(Direction::INGRESS, config.GetExternalIPsConf().GetDirection());
+  EXPECT_TRUE(config.GetExternalIPsConf().IsEnabled(Direction::INGRESS));
+  EXPECT_FALSE(config.GetExternalIPsConf().IsEnabled(Direction::EGRESS));
+  EXPECT_FALSE(config.GetExternalIPsConf().IsEnabled(Direction::BOTH));
+}
+
+TEST(CollectorConfigTest, TestIsEnabledEgress) {
+  std::string yamlStr = R"(
+                networking:
+                  externalIps:
+                    enabled: enabled
+                    direction: egress
+             )";
+
+  YAML::Node yamlNode = YAML::Load(yamlStr);
+  CollectorConfig config;
+  ASSERT_EQ(ConfigLoader(config).LoadConfiguration(yamlNode), ConfigLoader::SUCCESS) << "Input: " << yamlStr;
+
+  auto runtime_config = config.GetRuntimeConfig();
+
+  EXPECT_TRUE(runtime_config.has_value());
+
+  EXPECT_EQ(Direction::EGRESS, config.GetExternalIPsConf().GetDirection());
+  EXPECT_FALSE(config.GetExternalIPsConf().IsEnabled(Direction::INGRESS));
+  EXPECT_TRUE(config.GetExternalIPsConf().IsEnabled(Direction::EGRESS));
+  EXPECT_FALSE(config.GetExternalIPsConf().IsEnabled(Direction::BOTH));
+}
+
+TEST(CollectorConfigTest, TestIsEnabledBoth) {
+  std::string yamlStr = R"(
+                networking:
+                  externalIps:
+                    enabled: enabled
+                    direction: both
+             )";
+
+  YAML::Node yamlNode = YAML::Load(yamlStr);
+  CollectorConfig config;
+  ASSERT_EQ(ConfigLoader(config).LoadConfiguration(yamlNode), ConfigLoader::SUCCESS) << "Input: " << yamlStr;
+
+  auto runtime_config = config.GetRuntimeConfig();
+
+  EXPECT_TRUE(runtime_config.has_value());
+
+  EXPECT_EQ(Direction::BOTH, config.GetExternalIPsConf().GetDirection());
+  EXPECT_TRUE(config.GetExternalIPsConf().IsEnabled(Direction::INGRESS));
+  EXPECT_TRUE(config.GetExternalIPsConf().IsEnabled(Direction::EGRESS));
+  EXPECT_TRUE(config.GetExternalIPsConf().IsEnabled(Direction::BOTH));
+}
+
+TEST(CollectorConfigTest, TestIsEnabledNone) {
+  std::string yamlStr = R"(
+                networking:
+                  externalIps:
+                    enabled: disabled
+             )";
+
+  YAML::Node yamlNode = YAML::Load(yamlStr);
+  CollectorConfig config;
+  ASSERT_EQ(ConfigLoader(config).LoadConfiguration(yamlNode), ConfigLoader::SUCCESS) << "Input: " << yamlStr;
+
+  auto runtime_config = config.GetRuntimeConfig();
+
+  EXPECT_TRUE(runtime_config.has_value());
+
+  EXPECT_EQ(Direction::NONE, config.GetExternalIPsConf().GetDirection());
+  EXPECT_FALSE(config.GetExternalIPsConf().IsEnabled(Direction::INGRESS));
+  EXPECT_FALSE(config.GetExternalIPsConf().IsEnabled(Direction::EGRESS));
+  EXPECT_FALSE(config.GetExternalIPsConf().IsEnabled(Direction::BOTH));
+}
+
 }  // namespace collector