add tests

arthurpassos · arthurpassos · commit 819816100ecd · 2025-08-11T13:34:28.000-03:00
diff --git a/tests/integration/test_storage_azure_blob_storage/test.py b/tests/integration/test_storage_azure_blob_storage/test.py
@@ -647,6 +647,51 @@ def test_simple_write_named_collection_1_table_function(cluster):
     )
 
 
+def test_named_collection_hive_partitioning_partition_columns_in_data_file(cluster):
+    node = cluster.instances["node"]
+    port = cluster.env_variables["AZURITE_PORT"]
+    azure_query(
+        node,
+        f"""INSERT INTO TABLE FUNCTION azureBlobStorage(azure_conf2,
+        storage_account_url = '{cluster.env_variables['AZURITE_STORAGE_ACCOUNT_URL']}',
+        container='cont',
+        blob_path='test_hive_partitioning_partition_columns_in_data_file',
+        format=Parquet,
+        partition_strategy = 'hive',
+        partition_columns_in_data_file = 1)
+        PARTITION BY (year, country)
+        SELECT 1 as key, 2020 as year, 'USA' as country"""
+    )
+
+    azure_query(
+        node,
+        f"""INSERT INTO TABLE FUNCTION azureBlobStorage(azure_conf2,
+        storage_account_url = '{cluster.env_variables['AZURITE_STORAGE_ACCOUNT_URL']}',
+        container='cont',
+        blob_path='test_hive_partitioning_partition_columns_not_in_data_file',
+        format=Parquet,
+        partition_strategy = 'hive',
+        partition_columns_in_data_file = 0) PARTITION BY (year, country)
+        SELECT 1 as key, 2020 as year, 'USA' as country""",
+    )
+
+    assert azure_query(node,
+        f"""SELECT num_columns FROM azureBlobStorage(
+        azure_conf2,
+        storage_account_url = '{cluster.env_variables['AZURITE_STORAGE_ACCOUNT_URL']}',
+        container='cont',
+        blob_path='test_hive_partitioning_partition_columns_in_data_file/**.parquet',
+        format=ParquetMetadata)""") == "3\n"
+
+    assert azure_query(node,
+        f"""SELECT num_columns FROM azureBlobStorage(
+        azure_conf2,
+        storage_account_url = '{cluster.env_variables['AZURITE_STORAGE_ACCOUNT_URL']}',
+        container='cont',
+        blob_path='test_hive_partitioning_partition_columns_not_in_data_file/**.parquet',
+        format=ParquetMetadata)""") == "1\n"
+
+
 def test_simple_write_named_collection_2_table_function(cluster):
     node = cluster.instances["node"]
     port = cluster.env_variables["AZURITE_PORT"]
