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
|
from __future__ import annotations
import contextlib
from plumbum import local
from plumbum._testtools import skip_on_windows
with contextlib.suppress(ModuleNotFoundError):
from plumbum.cmd import printenv
@skip_on_windows
class TestEnv:
def test_change_env(self):
with local.env(silly=12):
assert local.env["silly"] == 12
actual = {x.split("=")[0] for x in printenv().splitlines() if "=" in x}
localenv = {x[0] for x in local.env}
print(actual, localenv)
assert localenv == actual
assert len(local.env) == len(actual)
def test_dictlike(self):
keys = {x.split("=")[0] for x in printenv().splitlines() if "=" in x}
values = {
x.split("=", 1)[1].strip() for x in printenv().splitlines() if "=" in x
}
assert keys == set(local.env.keys())
assert len(values) == len(set(local.env.values()))
def test_custom_env(self):
with local.env():
items = {"one": "OnE", "tww": "TWOO"}
local.env.update(items)
assert "tww" in local.env
local.env.clear()
assert "tww" not in local.env
def test_item(self):
with local.env():
local.env["simple_plum"] = "thing"
assert "simple_plum" in local.env
del local.env["simple_plum"]
assert "simple_plum" not in local.env
local.env["simple_plum"] = "thing"
assert "simple_plum" in local.env
assert local.env.pop("simple_plum") == "thing"
assert "simple_plum" not in local.env
local.env["simple_plum"] = "thing"
assert "simple_plum" not in local.env
@skip_on_windows
def test_home(self):
assert local.env.home == local.env["HOME"]
old_home = local.env.home
with local.env():
local.env.home = "Nobody"
assert local.env.home == local.env["HOME"]
assert local.env.home == "Nobody"
assert local.env.home == old_home
@skip_on_windows
def test_user(self):
assert local.env.user
|
