diff --git a/src/mito2/src/engine/scan_test.rs b/src/mito2/src/engine/scan_test.rs index 6adaadbc59..57594a28f1 100644 --- a/src/mito2/src/engine/scan_test.rs +++ b/src/mito2/src/engine/scan_test.rs @@ -75,7 +75,7 @@ async fn test_incremental_query_stale_error() { } _ => panic!("unexpected err: {err}"), }; - assert_eq!(StatusCode::InvalidArguments, err.status_code()); + assert_eq!(StatusCode::RequestOutdated, err.status_code()); let err_msg = err.to_string(); assert!(err_msg.contains("STALE_CURSOR")); assert!(err_msg.contains(®ion_id.to_string())); diff --git a/src/mito2/src/error.rs b/src/mito2/src/error.rs index 7bb63c864a..34cee1b335 100644 --- a/src/mito2/src/error.rs +++ b/src/mito2/src/error.rs @@ -1300,7 +1300,6 @@ impl ErrorExt for Error { | InvalidScanIndex { .. } | InvalidMeta { .. } | InvalidRequest { .. } - | IncrementalQueryStale { .. } | PartitionExprVersionMismatch { .. } | FillDefault { .. } | ConvertColumnDataType { .. } @@ -1319,6 +1318,8 @@ impl ErrorExt for Error { | SerializePartitionExpr { .. } | InvalidSourceAndTargetRegion { .. } => StatusCode::InvalidArguments, + IncrementalQueryStale { .. } => StatusCode::RequestOutdated, + RegionMetadataNotFound { .. } | Join { .. } | WorkerStopped { .. }