Coverage for test_main.py: 94%

1import unittest 

2import main 

3import os 

4 

5class TestMain(unittest.TestCase): 

6 def test_read_environment_var1(self): 

7 filename = 'config.json' 

8 env_Var = main.load_config(filename) 

9 self.assertEqual(env_Var.get('consumer_led_id', 17),17 ) 

10 

11 def test_read_environment_var2(self): 

12 filename = 'config.json' 

13 env_Var = main.load_config(filename) 

14 self.assertEqual(env_Var.get('consumer_led_id', 17),17 ) 

15 

16 def test_get_environment_filename(self): 

17 value = 'prod_env.json' 

18 os.environ['WORKING_ENV'] = value # Setzen der Umgebungsvariable 

19 self.assertEqual(main.get_enviroment_filename(), 'prod_env.json') 

20 

21if __name__ == "__main__": 

22 unittest.main()