Strip boolean environment variable strings
Update tests
This commit is contained in:
parent
47d54b1103
commit
05d6351686
@ -30,7 +30,8 @@ class EnvBase(lib_util.LoggedClass):
|
|||||||
return environ.get(envvar, default)
|
return environ.get(envvar, default)
|
||||||
|
|
||||||
def boolean(self, envvar, default):
|
def boolean(self, envvar, default):
|
||||||
return bool(self.default(envvar, default))
|
default = 'Yes' if default else ''
|
||||||
|
return bool(self.default(envvar, default).strip())
|
||||||
|
|
||||||
def required(self, envvar):
|
def required(self, envvar):
|
||||||
value = environ.get(envvar)
|
value = environ.get(envvar)
|
||||||
|
|||||||
49
tests/lib/test_env_base.py
Normal file
49
tests/lib/test_env_base.py
Normal file
@ -0,0 +1,49 @@
|
|||||||
|
# Tests of server/env.py
|
||||||
|
|
||||||
|
import os
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from lib.env_base import EnvBase
|
||||||
|
|
||||||
|
|
||||||
|
os.environ.update({
|
||||||
|
'int': '32',
|
||||||
|
'intspace': ' 32 ',
|
||||||
|
'true': 'x',
|
||||||
|
'false': '',
|
||||||
|
'space': ' ',
|
||||||
|
})
|
||||||
|
|
||||||
|
def test_default():
|
||||||
|
e = EnvBase()
|
||||||
|
assert e.default('int', '33') == '32'
|
||||||
|
assert e.default('baz', 'z') == 'z'
|
||||||
|
|
||||||
|
def test_boolean():
|
||||||
|
e = EnvBase()
|
||||||
|
assert e.boolean('true', False)
|
||||||
|
assert not e.boolean('false', True)
|
||||||
|
assert not e.boolean('space', True)
|
||||||
|
assert e.boolean('missing', True)
|
||||||
|
assert not e.boolean('missing', False)
|
||||||
|
|
||||||
|
def test_required():
|
||||||
|
e = EnvBase()
|
||||||
|
assert e.required('true') == 'x'
|
||||||
|
with pytest.raises(e.Error):
|
||||||
|
e.required('missing')
|
||||||
|
|
||||||
|
def test_integer():
|
||||||
|
e = EnvBase()
|
||||||
|
assert e.integer('int', 33) == 32
|
||||||
|
assert e.integer('missing', 33) == 33
|
||||||
|
assert e.integer('intspace', 33) == 32
|
||||||
|
assert e.integer('missing', None) is None
|
||||||
|
with pytest.raises(e.Error):
|
||||||
|
e.integer('true', 1)
|
||||||
|
|
||||||
|
def test_obsolete():
|
||||||
|
e = EnvBase()
|
||||||
|
with pytest.raises(e.Error):
|
||||||
|
e.obsolete(['z', 'space'])
|
||||||
Loading…
Reference in New Issue
Block a user