Add proxy tests

rcaelers · Sep 12, 2024 · e59c5a9 · e59c5a9
1 parent d90591a
commit e59c5a9
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/libs/http/include/http/Options.hh b/libs/http/include/http/Options.hh
@@ -72,6 +72,24 @@ namespace unfold::http
     std::chrono::seconds timeout = std::chrono::seconds(30);
     std::shared_ptr<spdlog::logger> logger{unfold::utils::Logging::create("unfold:http")};
   };
+
+  inline std::ostream &operator<<(std::ostream &os, Options::ProxyType type)
+  {
+    switch (type)
+      {
+      case Options::ProxyType::None:
+        os << "None";
+        break;
+      case Options::ProxyType::System:
+        os << "System";
+        break;
+      case Options::ProxyType::Custom:
+        os << "Custom";
+        break;
+      }
+    return os;
+  }
+
 } // namespace unfold::http
 
 #endif // NET_HTTP_OPTIONS_HH
diff --git a/test/UpgradeControlTests.cc b/test/UpgradeControlTests.cc
@@ -782,4 +782,18 @@ BOOST_AUTO_TEST_CASE(upgrade_control_callback_install_failed)
   BOOST_CHECK_EQUAL(status->error(), unfold::UnfoldErrc::InternalError);
 }
 
+BOOST_AUTO_TEST_CASE(upgrade_control_proxy)
+{
+  control->set_proxy(unfold::ProxyType::None);
+  BOOST_CHECK_EQUAL(http->options().get_proxy(), unfold::http::Options::ProxyType::None);
+  control->set_proxy(unfold::ProxyType::System);
+  BOOST_CHECK_EQUAL(http->options().get_proxy(), unfold::http::Options::ProxyType::System);
+  control->set_proxy(unfold::ProxyType::Custom);
+  BOOST_CHECK_EQUAL(http->options().get_proxy(), unfold::http::Options::ProxyType::Custom);
+  control->set_custom_proxy("http://proxy:8080");
+  BOOST_CHECK_EQUAL(http->options().get_custom_proxy(), "http://proxy:8080");
+  control->set_proxy(unfold::ProxyType::None);
+  BOOST_CHECK_EQUAL(http->options().get_proxy(), unfold::http::Options::ProxyType::None);
+}
+
 BOOST_AUTO_TEST_SUITE_END()