kindwolf.org Git repositories moulti / master tests / test_environ.py
master

Tree @master (Download .tar.gz)

test_environ.py @master — raw · history · blame 

 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
from moulti.environ import env, enva, envd, pint

def test_env(monkeypatch):
	monkeypatch.delenv('NON_EXISTENT_ENV_VAR', False)
	assert env('NON_EXISTENT_ENV_VAR', 'default') == 'default'

	monkeypatch.setenv('EMPTY', '')
	assert env('EMPTY', 'default') == ''

	monkeypatch.setenv('NAME', 'value')
	assert env('NAME', 'default') == 'value'

	monkeypatch.setenv('INTEGER', '123')
	assert env('INTEGER', -1, [int]) == 123

	monkeypatch.setenv('NON_INTEGER', 'foo')
	assert env('NON_INTEGER', -1, [int]) == -1
	assert env('NON_INTEGER', -1, [int, str]) == 'foo'
	assert env('NON_INTEGER', -1, [str, int]) == 'foo'


def test_enva(monkeypatch):
	monkeypatch.delenv('NON_EXISTENT_ENV_VAR', False)
	assert enva('NON_EXISTENT_ENV_VAR', [1, 2, 3]) == [1, 2, 3]

	monkeypatch.setenv('EMPTY', '')
	assert enva('EMPTY', [1, 2, 3]) == []

	monkeypatch.setenv('INTEGERS', '-2,-1,0,1,2')
	assert enva('INTEGERS', [1, 2, 3]) == ['-2', '-1', '0', '1', '2']

	monkeypatch.setenv('INTEGERS', '-2,-1,0,1,2')
	assert enva('INTEGERS', [1, 2, 3], [int]) == [-2, -1, 0, 1, 2]

	monkeypatch.setenv('INTEGERS', '-2,-1,0,1,2')
	assert enva('INTEGERS', [1, 2, 3], [pint]) == [1, 2, 3]

	monkeypatch.setenv('INTEGERS', '-2,-1,0,1,2')
	assert enva('INTEGERS', [1, 2, 3], [pint], strict=False) == [0, 1, 2]

	monkeypatch.setenv('INTEGERS', '-2,-1,0,1,2')
	assert enva('INTEGERS', [1, 2, 3], [pint, str]) == ['-2', '-1', 0, 1, 2]

	monkeypatch.setenv('INTEGERS', '-2;-1;0;1;2')
	assert enva('INTEGERS', [1, 2, 3], [int], sep=';') == [-2, -1, 0, 1, 2]

	# Separator with multiple characters:
	monkeypatch.setenv('INTEGERS', '-2;;-1;;0;;1;;2')
	assert enva('INTEGERS', [1, 2, 3], [int], sep=';;') == [-2, -1, 0, 1, 2]

	monkeypatch.setenv('INTEGERS', '-2;;-1;;0;;1;2')
	assert enva('INTEGERS', [1, 2, 3], [int], sep=';;') == [1,2,3]

	monkeypatch.setenv('INTEGERS', '-2;;-1;;0;;1;2')
	assert enva('INTEGERS', [1, 2, 3], [int], sep=';;', strict=False) == [-2, -1, 0]

	# Trailing separator:
	monkeypatch.setenv('INTEGERS', '-2;;-1;;0;;1;2;;')
	assert enva('INTEGERS', [1, 2, 3], [int], sep=';;', strict=False) == [-2, -1, 0]

	monkeypatch.setenv('INTEGERS', '-2;;-1;;0;;1;2;;')
	assert enva('INTEGERS', [1, 2, 3], [int], sep=';;', strict=True) == [1, 2, 3]


def test_envd(monkeypatch):
	monkeypatch.delenv('NON_EXISTENT_ENV_VAR', False)
	assert envd('NON_EXISTENT_ENV_VAR', {'a': 'b'}) == {'a': 'b'}

	monkeypatch.setenv('EMPTY', '')
	assert envd('EMPTY', {'a': 'b'}) == {}

	monkeypatch.setenv('INTEGERS', 'a=1,b=2,c=3')
	assert envd('INTEGERS') == {'a': '1', 'b': '2', 'c': '3'}

	monkeypatch.setenv('INTEGERS', 'a=1,b=2,c=3')
	assert envd('INTEGERS', types=[int]) == {'a': 1, 'b': 2, 'c': 3}

	monkeypatch.setenv('INTEGERS', 'a=1,b=2,c=3,d=foo')
	assert envd('INTEGERS', types=[int]) is None

	monkeypatch.setenv('INTEGERS', 'a=1,b=2,c=3,d=foo')
	assert envd('INTEGERS', types=[int], strict=False) == {'a': 1, 'b': 2, 'c': 3}

	monkeypatch.setenv('INTEGERS', 'a=1,b=2,c=3,d=foo=')
	assert envd('INTEGERS', types=[int, str]) == {'a': 1, 'b': 2, 'c': 3, 'd': 'foo='}

	monkeypatch.setenv('INTEGERS', 'a==1,b==2,c==3,d==foo==')
	assert envd('INTEGERS', types=[int, str], sepkv='==') == {'a': 1, 'b': 2, 'c': 3, 'd': 'foo=='}

	monkeypatch.setenv('INTEGERS', 'a=1,b,c=3')
	assert envd('INTEGERS', types=[int], sepkv='==') is None