def test_load_env(self):
os.environ['OPT1'] = "value1"
os.environ['OPT2'] = "value2"
globals().update(flexible_config.load_local_config(['OPT1','OPT2'], config_file = "nofile"))
self.assertIn('OPT1', globals())
self.assertEqual("value1", globals()['OPT1'])
flexible_config.check_required_options(['OPT1', 'OPT2', 'OPT3'], globals())
def test_load_file(self):
with open("/tmp/opts", "w") as f:
f.write("[dev]\n")
f.write("FOPT1=fval1\n")
f.write("FOPT2=fval2\n")
f.write("[prod]\n")
f.write("FOPT1=fprod1\n")
f.write("FOPT2=fprod2\n")
self.assertNotIn('FOPT1', globals())
if 'CCENV' in os.environ:
del os.environ['CCENV']
globals().update(flexible_config.load_local_config(['FOPT1','FOPT2'], config_file = "/tmp/opts"))
self.assertIn('FOPT1', globals())
self.assertEqual(FOPT1, "fval1")
os.environ['CCENV'] = 'prod'
globals().update(flexible_config.load_local_config(['FOPT1','FOPT2'], config_file = "/tmp/opts"))
self.assertEqual(FOPT1, "fprod1")
self.assertEqual(FOPT2, "fprod2")
