SNOW-2255664: Populate type_code for interval types in ResultMetadata (#2467)

Author: sfc-gh-nkumar

DESCRIPTION.md

@@ -23,6 +23,7 @@ Source code is also available at: https://github.com/snowflakedb/snowflake-conne
   - Moved `OAUTH_TYPE` to `CLIENT_ENVIROMENT`.
   - Fix bug where PAT with external session authenticator was used while `external_session_id` was not provided in `SnowflakeRestful.fetch`
   - Added support for parameter `use_vectorized_scanner` in function `write_pandas`.
+  - Populate type_code in ResultMetadata for interval types.
 
 - v3.16.0(July 04,2025)
   - Bumped numpy dependency from <2.1.0 to <=2.2.4.

src/snowflake/connector/constants.py

@@ -183,6 +183,16 @@ def struct_pa_type(metadata: ResultMetadataV2) -> DataType:
     FieldType(
         name="FILE", dbapi_type=[DBAPI_TYPE_STRING], pa_type=lambda _: pa.string()
     ),
+    FieldType(
+        name="INTERVAL_YEAR_MONTH",
+        dbapi_type=[DBAPI_TYPE_NUMBER],
+        pa_type=lambda _: pa.int64(),
+    ),
+    FieldType(
+        name="INTERVAL_DAY_TIME",
+        dbapi_type=[DBAPI_TYPE_NUMBER],
+        pa_type=lambda _: pa.int64(),
+    ),
 )
 
 FIELD_NAME_TO_ID: DefaultDict[Any, int] = defaultdict(int)

test/integ/test_interval_types.py

@@ -6,6 +6,8 @@
 import numpy
 import pytest
 
+from snowflake.connector import constants
+
 pytestmark = pytest.mark.skipolddriver  # old test driver tests won't run this module
 
 
@@ -30,14 +32,17 @@ def test_select_year_month_interval(conn_cnx, use_numpy, result_format):
         cursor.execute("alter session set feature_interval_types=enabled")
         cursor.execute(f"create or replace table {table} (c1 interval year to month)")
         cursor.execute(f"insert into {table} values {values}")
-        result = conn.cursor().execute(f"select * from {table}").fetchall()
+        result = cursor.execute(f"select * from {table}").fetchall()
+        # Validate column metadata.
+        type_code = cursor._description[0].type_code
+        assert (
+            constants.FIELD_ID_TO_NAME[type_code] == "INTERVAL_YEAR_MONTH"
+        ), f"invalid column type: {type_code}"
+        # Validate column values.
         result = [r[0] for r in result]
         assert result == expected
 
 
-@pytest.mark.skip(
-    reason="SNOW-1878635: Add support for day-time interval in ArrowStreamWriter"
-)
 @pytest.mark.parametrize("use_numpy", [True, False])
 @pytest.mark.parametrize("result_format", ["json", "arrow"])
 def test_select_day_time_interval(conn_cnx, use_numpy, result_format):
@@ -71,6 +76,12 @@ def test_select_day_time_interval(conn_cnx, use_numpy, result_format):
             f"create or replace table {table} (c1 interval day(5) to second)"
         )
         cursor.execute(f"insert into {table} values {values}")
-        result = conn.cursor().execute(f"select * from {table}").fetchall()
+        result = cursor.execute(f"select * from {table}").fetchall()
+        # Validate column metadata.
+        type_code = cursor._description[0].type_code
+        assert (
+            constants.FIELD_ID_TO_NAME[type_code] == "INTERVAL_DAY_TIME"
+        ), f"invalid column type: {type_code}"
+        # Validate column values.
         result = [r[0] for r in result]
         assert result == expected