From 47944a4d0edc7cb852a882702de809ba77e761bb Mon Sep 17 00:00:00 2001
From: taiyang-li <654010905@qq.com>
Date: Tue, 19 Nov 2024 11:43:28 +0800
Subject: [PATCH] fix failed uts

---
 src/Storages/Hive/HiveCommon.cpp                             | 1 +
 tests/queries/0_stateless/02809_storage_set_analysis_bug.sql | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Storages/Hive/HiveCommon.cpp b/src/Storages/Hive/HiveCommon.cpp
index d58e7b535cc..4ea2220b9ef 100644
--- a/src/Storages/Hive/HiveCommon.cpp
+++ b/src/Storages/Hive/HiveCommon.cpp
@@ -7,6 +7,7 @@
 #include <thrift/transport/TBufferTransports.h>
 #include <thrift/transport/TSocket.h>
 #include <Storages/Hive/HiveFile.h>
+#include <Poco/URI.h>
 
 
 namespace DB
diff --git a/tests/queries/0_stateless/02809_storage_set_analysis_bug.sql b/tests/queries/0_stateless/02809_storage_set_analysis_bug.sql
index 531946acda0..f71494673f2 100644
--- a/tests/queries/0_stateless/02809_storage_set_analysis_bug.sql
+++ b/tests/queries/0_stateless/02809_storage_set_analysis_bug.sql
@@ -4,7 +4,7 @@ DROP TABLE IF EXISTS null_in__fuzz_6;
 set allow_suspicious_low_cardinality_types = 1;
 
 CREATE TABLE null_in__fuzz_6 (`dt` LowCardinality(UInt16), `idx` Int32, `i` Nullable(Int256), `s` Int32) ENGINE = MergeTree PARTITION BY dt ORDER BY idx;
-insert into null_in__fuzz_6 select * from generateRandom() limit 1;
+insert into null_in__fuzz_6 select * from generateRandom() where i is not null limit 1;
 
 SET transform_null_in = 0;