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)
|
||||
|
||||
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):
|
||||
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