1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
use std::collections::BTreeMap;
use xshell::cmd;
use crate::setup;
#[test]
fn test_env() {
let sh = setup();
let v1 = "xshell_test_123";
let v2 = "xshell_test_456";
assert_env(cmd!(sh, "xecho -$ {v1}").env(v1, "123"), &[(v1, Some("123"))]);
assert_env(
cmd!(sh, "xecho -$ {v1} {v2}").envs([(v1, "123"), (v2, "456")].iter().copied()),
&[(v1, Some("123")), (v2, Some("456"))],
);
assert_env(
cmd!(sh, "xecho -$ {v1} {v2}")
.envs([(v1, "123"), (v2, "456")].iter().copied())
.env_remove(v2),
&[(v1, Some("123")), (v2, None)],
);
assert_env(
cmd!(sh, "xecho -$ {v1} {v2}")
.envs([(v1, "123"), (v2, "456")].iter().copied())
.env_remove("nothing"),
&[(v1, Some("123")), (v2, Some("456"))],
);
let _g1 = sh.push_env(v1, "foobar");
let _g2 = sh.push_env(v2, "quark");
assert_env(cmd!(sh, "xecho -$ {v1} {v2}"), &[(v1, Some("foobar")), (v2, Some("quark"))]);
assert_env(
cmd!(sh, "xecho -$ {v1} {v2}").env(v1, "wombo"),
&[(v1, Some("wombo")), (v2, Some("quark"))],
);
assert_env(cmd!(sh, "xecho -$ {v1} {v2}").env_remove(v1), &[(v1, None), (v2, Some("quark"))]);
assert_env(
cmd!(sh, "xecho -$ {v1} {v2}").env_remove(v1).env(v1, "baz"),
&[(v1, Some("baz")), (v2, Some("quark"))],
);
assert_env(
cmd!(sh, "xecho -$ {v1} {v2}").env(v1, "baz").env_remove(v1),
&[(v1, None), (v2, Some("quark"))],
);
}
#[test]
fn test_env_clear() {
let sh = setup();
let v1 = "xshell_test_123";
let v2 = "xshell_test_456";
let xecho = format!("./target/xecho{}", std::env::consts::EXE_SUFFIX);
assert_env(
cmd!(sh, "{xecho} -$ {v1} {v2}")
.envs([(v1, "123"), (v2, "456")].iter().copied())
.env_clear(),
&[(v1, None), (v2, None)],
);
assert_env(
cmd!(sh, "{xecho} -$ {v1} {v2}")
.envs([(v1, "123"), (v2, "456")].iter().copied())
.env_clear()
.env(v1, "789"),
&[(v1, Some("789")), (v2, None)],
);
let _g1 = sh.push_env(v1, "foobar");
let _g2 = sh.push_env(v2, "quark");
assert_env(cmd!(sh, "{xecho} -$ {v1} {v2}").env_clear(), &[(v1, None), (v2, None)]);
assert_env(
cmd!(sh, "{xecho} -$ {v1} {v2}").env_clear().env(v1, "baz"),
&[(v1, Some("baz")), (v2, None)],
);
assert_env(
cmd!(sh, "{xecho} -$ {v1} {v2}").env(v1, "baz").env_clear(),
&[(v1, None), (v2, None)],
);
}
#[track_caller]
fn assert_env(xecho_env_cmd: xshell::Cmd, want_env: &[(&str, Option<&str>)]) {
let output = xecho_env_cmd.output().unwrap();
let env = String::from_utf8_lossy(&output.stdout)
.lines()
.filter(|line| !line.is_empty())
.map(|line| {
let (key, val) = line.split_once('=').unwrap_or_else(|| {
panic!("failed to parse line from `xecho -$` output: {:?}", line)
});
(key.to_owned(), val.to_owned())
})
.collect::<BTreeMap<_, _>>();
check_env(&env, want_env);
}
#[track_caller]
fn check_env(env: &BTreeMap<String, String>, wanted_env: &[(&str, Option<&str>)]) {
let mut failed = false;
let mut seen = env.clone();
for &(k, val) in wanted_env {
match (seen.remove(k), val) {
(Some(env_v), Some(want_v)) if env_v == want_v => {}
(None, None) => {}
(have, want) => {
eprintln!("mismatch on env var {:?}: have `{:?}`, want `{:?}` ", k, have, want);
failed = true;
}
}
}
for (k, v) in seen {
eprintln!("Unexpected env key {:?} (value: {:?})", k, v);
failed = true;
}
assert!(
!failed,
"env didn't match (see stderr for cleaner output):\nsaw: {:?}\n\nwanted: {:?}",
env, wanted_env,
);
}
|
