risingwave_regress_test/
env.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::env::{remove_var, set_var, var};
16
17pub(crate) const VAR_PG_OPTS: &str = "PGOPTIONS";
18
19/// Setup some environments for psql.
20///
21/// This is useful since it may affect psql's behavior.
22pub(crate) fn init_env() {
23    // Set default application name.
24    unsafe { set_var("PGAPPNAME", "risingwave_regress") };
25
26    // Set translation-related settings to English; otherwise psql will
27    // produce translated messages and produce diffs.  (XXX If we ever support
28    // translation of pg_regress, this needs to be moved elsewhere, where psql
29    // is actually called.)
30    unsafe { remove_var("LANGUAGE") };
31    unsafe { remove_var("LC_ALL") };
32    unsafe { set_var("LC_MESSAGES", "C") };
33
34    // Set timezone and datestyle for datetime-related tests
35    unsafe { set_var("PGTZ", "PST8PDT") };
36    unsafe { set_var("PGDATESTYLE", "Postgres, MDY") };
37
38    // Likewise set intervalstyle to ensure consistent results.  This is a bit
39    // more painful because we must use PGOPTIONS, and we want to preserve the
40    // user's ability to set other variables through that.
41    {
42        let mut pg_opts = var(VAR_PG_OPTS).unwrap_or_else(|_| "".to_owned());
43        pg_opts.push_str(" -c intervalstyle=postgres_verbose");
44        unsafe { set_var(VAR_PG_OPTS, pg_opts) };
45    }
46}
risingwave_regress_test/env.rs

risingwave_regress_test/
env.rs
