-
Notifications
You must be signed in to change notification settings - Fork 3
/
tests.py
173 lines (131 loc) · 4.37 KB
/
tests.py
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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
import io
import os
import stat
import textwrap
import pytest
import reify
def test_parse_envfile():
envfile = io.StringIO(textwrap.dedent("""
X=x
Y=y$X # substitution within file
Z=z$Z # base env substitution
quotes="some value" # test quoted values
quotes_single='some value' # test quoted values
# line level comment, then a blank line
"""))
env = {'Z': 'z'}
reify.parse_envfile(env, envfile)
assert env == {
'X': 'x',
'Y': 'yx',
'Z': 'zz',
'quotes': 'some value',
'quotes_single': 'some value',
}
def test_parse_envfile_error():
envfile = io.StringIO("X=x\nY=y foo")
env = {}
with pytest.raises(Exception) as e:
reify.parse_envfile(env, envfile)
assert 'X=x y' in str(e)
assert 'line 2' in str(e)
def test_parse_yamlfile():
assert reify.parse_yamlfile(io.StringIO("")) == {}
assert reify.parse_yamlfile(io.StringIO("{}")) == {}
assert reify.parse_yamlfile(io.StringIO("[]")) == {}
non_dict = io.StringIO("[1]")
non_dict.name = 'test'
with pytest.raises(Exception):
reify.parse_yamlfile(non_dict)
d = reify.parse_yamlfile(io.StringIO(textwrap.dedent("""
foo:
bar:
- 1
- 2
""")))
assert d == {'foo': {'bar': [1, 2]}}
def test_parse_charm_defaults():
assert reify.parse_charm_defaults(io.StringIO("")) == {}
assert reify.parse_charm_defaults(io.StringIO("{}")) == {}
defaults = reify.parse_charm_defaults(io.StringIO(textwrap.dedent("""
options:
string:
default: string
type: string
int:
default: 10
type: int
float:
default: 1.23
type: float
boolean:
default: false
type: boolean
no-type:
default: 1.0
""")))
assert defaults == {
'string': 'string',
'int': 10,
'float': 1.23,
'boolean': False,
'no-type': '1.0', # defaults to string
}
def test_atomic_write(tmpdir):
path = str(tmpdir.join('file'))
reify.atomic_write(path, 'hi')
assert open(path).read() == 'hi'
assert not os.path.exists(path + '.reify.tmp')
def test_atomic_write_mode(tmpdir):
path = str(tmpdir.join('file'))
reify.atomic_write(path, 'hi', mode=0o400)
assert open(path).read() == 'hi'
assert not os.path.exists(path + '.reify.tmp')
assert stat.S_IMODE(os.stat(path).st_mode) == 0o400
def test_atomic_write_rename_fails(tmpdir, monkeypatch):
class TestException(Exception):
pass
def rename(x, y):
raise TestException()
monkeypatch.setattr(os, 'rename', rename)
path = str(tmpdir.join('file'))
with open(path, 'w') as f:
f.write('other')
with pytest.raises(TestException):
reify.atomic_write(path, 'hi')
assert not os.path.exists(path + '.reify.tmp')
assert open(path).read() == 'other'
TEMPLATE = "'{{ test }}' '{{ env['TEST'] }}'"
def test_render_none():
output = reify.render(TEMPLATE, {}, None, {})
assert output == "'' ''\n"
def test_render_simple():
output = reify.render(TEMPLATE, {'test': 'ctx'}, None, {})
assert output == "'ctx' ''\n"
def test_render_envvar():
output = reify.render(TEMPLATE, {}, None, {'TEST': 'env'})
assert output == "'' 'env'\n"
def test_render_envfile():
output = reify.render(TEMPLATE, {}, io.StringIO('TEST=envfile'), {})
assert output == "'' 'envfile'\n"
def test_render_envfile_overrides_env():
output = reify.render(
TEMPLATE, {}, io.StringIO('TEST=envfile'), {'TEST': 'env'})
assert output == "'' 'envfile'\n"
def test_render_ctx_overrides_envfile():
output = reify.render(
TEMPLATE, {'env': {'TEST': 'ctx'}}, io.StringIO('TEST=envfile'), {})
assert output == "'' 'ctx'\n"
def test_render_charm_config():
charm_config = io.StringIO(textwrap.dedent("""
options:
test:
default: charm default
type: string
"""))
output = reify.render(TEMPLATE, {}, None, {}, charm_config)
assert output == "'charm default' ''\n"
def test_reify_function(tmpdir):
path = str(tmpdir.join('file'))
reify.reify(path, TEMPLATE, {'test': 'test'})
assert open(path).read() == "'test' ''\n"