testing env support

integration_tests/conftest.py

@@ -74,4 +74,5 @@ def dev_session():
     process = spawn_process(command)
     time.sleep(5)
     yield
-    kill_process(process)
+    kill_process(process)
+

integration_tests/test_env_populator.py

@@ -0,0 +1,32 @@
+# from integration_tests.conftest import test_session
+from robyn.env_populator import load_vars, parser
+import pathlib
+import os
+import pytest
+import requests
+
+
+
+path = pathlib.Path(__file__).parent  
+
+#create robyn.env before test and delete it after test
+@pytest.fixture
+def env_file():
+    CONTENT = "ROBYN_PORT=8080" + "\n" + "ROBYN_URL=127.0.1.1"
+    dir = path / "test_dir"
+    env_file = dir / "robyn.env"
+    env_file.write_text(CONTENT)
+    yield
+    env_file.unlink()
+    os.unsetenv("PORT")
+
+# this tests if a connection can be made to the server with the correct port imported from the env file
+def test_env_population(dev_session, env_file):
+    dir = path / "test_dir"
+    env_file = dir / "robyn.env"
+    load_vars(variables = parser(config_path = env_file))
+    PORT = os.environ['ROBYN_PORT'] 
+    HOST = os.environ['ROBYN_URL']
+    BASE_URL = f"http://{HOST}:{PORT}"
+    res = requests.get(f"{BASE_URL}")
+    assert res.status_code == 200

robyn/env_populator.py

@@ -29,10 +29,8 @@ def parser(config_path=CONFIG_PATH):
 def load_vars(variables = None):
     """Main function"""
 
-    variables = parser() 
-
     if variables is None:
-        return
+        variables = parser()
 
     for var in variables:
         if var[0] in os.environ:
@@ -43,6 +41,6 @@ def load_vars(variables = None):
             logger.info(f" Variable {var[0]} set to {var[1]}")
 
 
-    
+
 
 

unit_tests/test_env_pop.py

@@ -0,0 +1,31 @@
+from robyn.env_populator import load_vars, parser
+import pathlib
+import os
+import pytest
+
+
+
+
+path = pathlib.Path(__file__).parent  
+
+#create robyn.env before test and delete it after test
+@pytest.fixture
+def env_file():
+    CONTENT = "PORT=8080"
+    dir = path / "test_dir"
+    env_file = dir / "robyn.env"
+    env_file.write_text(CONTENT)
+    yield
+    env_file.unlink()
+    os.unsetenv("PORT")
+
+def test_parser(env_file):
+    dir = path / "test_dir"
+    env_file = dir / "robyn.env"
+    assert list(parser(config_path = env_file)) == [['PORT', '8080']]
+
+def test_load_vars(env_file):
+    dir = path / "test_dir"
+    env_file = dir / "robyn.env"
+    load_vars(variables = parser(config_path = env_file))
+    assert os.environ['PORT'] == '8080'