test: fix imports

aws · Nov 25, 2024 · 53c6313 · 53c6313
1 parent 8ccdda2
commit 53c6313
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 10 deletions.
diff --git a/unit_testing/test_utils.cc b/unit_testing/test_utils.cc
@@ -128,33 +128,33 @@ bool TEST_UTILS::try_parse_region_from_secret(std::string secret, std::string& r
 }
 
 bool TEST_UTILS::is_dns_pattern_valid(std::string host) {
-    return FAILOVER_HANDLER::is_dns_pattern_valid(host);
+    return RDS_UTILS::is_dns_pattern_valid(host);
 }
 
 bool TEST_UTILS::is_rds_dns(std::string host) {
-    return FAILOVER_HANDLER::is_rds_dns(host);
+    return RDS_UTILS::is_rds_dns(host);
 }
 
 bool TEST_UTILS::is_rds_cluster_dns(std::string host) {
-    return FAILOVER_HANDLER::is_rds_cluster_dns(host);
+    return RDS_UTILS::is_rds_cluster_dns(host);
 }
 
 bool TEST_UTILS::is_rds_proxy_dns(std::string host) {
-    return FAILOVER_HANDLER::is_rds_proxy_dns(host);
+    return RDS_UTILS::is_rds_proxy_dns(host);
 }
 
 bool TEST_UTILS::is_rds_writer_cluster_dns(std::string host) {
-    return FAILOVER_HANDLER::is_rds_writer_cluster_dns(host);
+    return RDS_UTILS::is_rds_writer_cluster_dns(host);
 }
 
 bool TEST_UTILS::is_rds_custom_cluster_dns(std::string host) {
-    return FAILOVER_HANDLER::is_rds_custom_cluster_dns(host);
+    return RDS_UTILS::is_rds_custom_cluster_dns(host);
 }
 
 std::string TEST_UTILS::get_rds_cluster_host_url(std::string host) {
-    return FAILOVER_HANDLER::get_rds_cluster_host_url(host);
+    return RDS_UTILS::get_rds_cluster_host_url(host);
 }
 
 std::string TEST_UTILS::get_rds_instance_host_pattern(std::string host) {
-    return FAILOVER_HANDLER::get_rds_instance_host_pattern(host);
+    return RDS_UTILS::get_rds_instance_host_pattern(host);
 }
diff --git a/unit_testing/test_utils.h b/unit_testing/test_utils.h
@@ -32,12 +32,11 @@
 
 #include "driver/auth_util.h" 
 #include "driver/driver.h"
-#include "driver/failover.h"
 #include "driver/iam_proxy.h"
-#include "driver/okta_proxy.h"
 #include "driver/monitor.h"
 #include "driver/monitor_thread_container.h"
 #include "driver/secrets_manager_proxy.h"
+#include "driver/rds_utils.h"
 
 void allocate_odbc_handles(SQLHENV& env, DBC*& dbc, DataSource*& ds);
 void cleanup_odbc_handles(SQLHENV env, DBC*& dbc, DataSource*& ds, bool call_myodbc_end = false);