Explicitly set gekodriver in environment

Signed-off-by: John Rofrano <johnnyroy@johnrofrano.com>

Author: rofrano

features/environment.py

@@ -2,12 +2,12 @@
 Environment for Behave Testing
 """
 
-from os import getenv
+import os
 from selenium import webdriver
 
-WAIT_SECONDS = int(getenv("WAIT_SECONDS", "30"))
-BASE_URL = getenv("BASE_URL", "http://localhost:8080")
-DRIVER = getenv("DRIVER", "firefox").lower()
+WAIT_SECONDS = int(os.getenv("WAIT_SECONDS", "30"))
+BASE_URL = os.getenv("BASE_URL", "http://localhost:8080")
+DRIVER = os.getenv("DRIVER", "firefox").lower()
 
 
 def before_all(context):
@@ -47,8 +47,8 @@ def get_chrome():
 def get_firefox():
     """Creates a headless Firefox driver"""
     print("Running Behave using the Firefox driver...\n")
+    # Set the environment variable for the Selenium WebDriver
+    os.environ["SELENIUM_DRIVER"] = "geckodriver"
     options = webdriver.FirefoxOptions()
-    options.add_argument("--no-sandbox")
-    options.add_argument("--disable-dev-shm-usage")
     options.add_argument("--headless")
     return webdriver.Firefox(options=options)