25 lines
793 B
Rust
25 lines
793 B
Rust
|
use std::ffi::OsStr;
|
||
|
|
|
||
|
|
use xai_grok_pager_pty_harness::{EnvOp, oauth_credential_ops};
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn set_and_remove_operations_have_one_typed_surface() {
|
||
|
|
let key = OsStr::new("FEATURE_FLAG");
|
||
|
|
let value = OsStr::new("enabled");
|
||
|
|
let operations: [EnvOp<'_>; 4] = [
|
||
|
|
EnvOp::set("FEATURE_FLAG", "enabled"),
|
||
|
|
EnvOp::remove("XAI_API_KEY"),
|
||
|
|
EnvOp::set_os(key, value),
|
||
|
|
EnvOp::remove_os(key),
|
||
|
|
];
|
||
|
|
|
||
|
|
assert!(matches!(operations[0], EnvOp::Set(_, _)));
|
||
|
|
assert!(matches!(operations[1], EnvOp::Remove(_)));
|
||
|
|
assert!(matches!(operations[2], EnvOp::Set(_, _)));
|
||
|
|
assert!(matches!(operations[3], EnvOp::Remove(_)));
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn oauth_credential_operations_remove_the_api_key() {
|
||
|
|
assert_eq!(oauth_credential_ops(), [EnvOp::remove("XAI_API_KEY")],);
|
||
|
|
}
|