From 95564cb5f760d74802c71de3b596aaec372d1c4a Mon Sep 17 00:00:00 2001
From: discord9 <55937128+discord9@users.noreply.github.com>
Date: Tue, 8 Sep 2026 12:33:54 +0800
Subject: [PATCH] test(promql): decode dictionary labels in stale regression

Signed-off-by: discord9 <55937128+discord9@users.noreply.github.com>
---
 tests-integration/src/tests/promql_test.rs | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/tests-integration/src/tests/promql_test.rs b/tests-integration/src/tests/promql_test.rs
index 852c9cc6b34..67045802968 100644
--- a/tests-integration/src/tests/promql_test.rs
+++ b/tests-integration/src/tests/promql_test.rs
@@ -23,6 +23,8 @@ use common_query::prometheus::PROMETHEUS_STALE_NAN_BITS;
 use common_query::{Output, OutputData};
 use common_recordbatch::util::collect_batches;
 use datatypes::arrow::array::{Float64Array, Int64Array, StringArray, TimestampMillisecondArray};
+use datatypes::arrow::compute::cast;
+use datatypes::arrow::datatypes::DataType;
 use frontend::instance::Instance;
 use query::parser::{PromQuery, QueryLanguageParser, QueryStatement};
 use rstest::rstest;
@@ -1040,12 +1042,12 @@ async fn promql_stale_marker_excludes_series_across_flushes(instance: Arc<dyn Mo
         let mut samples = batches
             .iter()
             .flat_map(|batch| {
-                let series = batch
-                    .column_by_name("series")
-                    .unwrap()
-                    .as_any()
-                    .downcast_ref::<StringArray>()
-                    .unwrap();
+                let series = cast(
+                    batch.column_by_name("series").unwrap().as_ref(),
+                    &DataType::Utf8,
+                )
+                .unwrap();
+                let series = series.as_any().downcast_ref::<StringArray>().unwrap();
                 let values = batch
                     .column_by_name("value")
                     .unwrap()