From 53c63138681b03f050e0d9ee689a5f8e4d0f103a Mon Sep 17 00:00:00 2001
From: Karen Chen <karenc@bitquilltech.com>
Date: Mon, 25 Nov 2024 15:20:18 -0800
Subject: [PATCH] test: fix imports

---
 unit_testing/test_utils.cc | 16 ++++++++--------
 unit_testing/test_utils.h  |  3 +--
 2 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/unit_testing/test_utils.cc b/unit_testing/test_utils.cc
index 6d0709d8b..3b96eb9bf 100644
--- 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
index a2766633b..0202f87e5 100644
--- 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);