Update battery_test.py

Author: enese94

battery_test.py

@@ -18,20 +18,48 @@ def get_browser_path(browser_name):
             path = r"C:\Program Files\Mozilla Firefox\firefox.exe"
         elif browser_name == "vivaldi":
             path = r"C:\Users\YourUserName\AppData\Local\Vivaldi\Application\vivaldi.exe"
+        elif browser_name == "edge":
+            path = r"C:\Program Files (x86)\Microsoft\Edge\Application\msedge.exe"
+        elif browser_name == "opera":
+            path = r"C:\Users\YourUserName\AppData\Local\Programs\Opera\opera.exe"
+        elif browser_name == "brave":
+            path = r"C:\Program Files\BraveSoftware\Brave-Browser\Application\brave.exe"
+        elif browser_name == "zen": # example if zen browser is installed on windows
+            path = r"C:\Users\YourUserName\AppData\Local\ZenBrowser\Application\zen.exe"
+
     elif os_name == "Darwin":
         if browser_name == "chrome":
             path = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
         elif browser_name == "firefox":
             path = "/Applications/Firefox.app/Contents/MacOS/firefox"
         elif browser_name == "vivaldi":
             path = "/Applications/Vivaldi.app/Contents/MacOS/Vivaldi"
+        elif browser_name == "edge":
+            path = "/Applications/Microsoft Edge.app/Contents/MacOS/Microsoft Edge"
+        elif browser_name == "opera":
+            path = "/Applications/Opera.app/Contents/MacOS/Opera"
+        elif browser_name == "brave":
+            path = "/Applications/Brave Browser.app/Contents/MacOS/Brave Browser"
+        elif browser_name == "safari":
+            path = "/Applications/Safari.app/Contents/MacOS/Safari"
+        elif browser_name == "zen": #example if zen browser is installed on MacOS
+             path = "/Applications/ZenBrowser.app/Contents/MacOS/ZenBrowser"
+
     elif os_name == "Linux":
         if browser_name == "chrome":
             path = "/usr/bin/google-chrome"
         elif browser_name == "firefox":
             path = "/usr/bin/firefox"
         elif browser_name == "vivaldi":
             path = "/usr/bin/vivaldi"
+        elif browser_name == "edge":
+            path = "/usr/bin/microsoft-edge"
+        elif browser_name == "opera":
+            path = "/usr/bin/opera"
+        elif browser_name == "brave":
+            path = "/usr/bin/brave-browser"
+        elif browser_name == "zen": #example if zen browser is installed on linux.
+            path = "/usr/bin/zenbrowser"
 
     if path and os.path.exists(path):
         return path
@@ -54,7 +82,7 @@ def open_browser(browser_path, start_url):
 def navigate_to_url(browser_path, url):
     """Navigates the browser to the given URL using subprocess."""
     try:
-        subprocess.Popen([browser_path, "--remote-debugging-port=9222", "--remote-debugging-address=127.0.0.1", url]) #this will open a new instance of the browser if one isn't already open.
+        subprocess.Popen([browser_path, "--remote-debugging-port=9222", "--remote-debugging-address=127.0.0.1", url])
         return True
     except Exception as e:
         print(f"Error navigating to URL: {e}")
@@ -64,7 +92,7 @@ def scroll_and_wait(duration=30):
     """Scrolls up and down for a specified duration."""
     start_time = time.time()
     while time.time() - start_time < duration:
-        scroll_amount = random.randint(-2500, 2500)
+        scroll_amount = random.randint(-500, 500)
         pyautogui.scroll(scroll_amount)
         time.sleep(0.1)
 
@@ -79,7 +107,7 @@ def battery_test(website_list, browser_name, start_url):
     if not open_browser(browser_path, start_url):
         return
 
-    time.sleep(5) #Wait for browser to open.
+    time.sleep(5)
 
     try:
         while True:
@@ -128,6 +156,6 @@ def battery_test(website_list, browser_name, start_url):
     "https://www.shopify.com",
 ]
 
-browser_name = input("Enter the name of the browser you want to test (chrome, firefox, vivaldi or another): ").lower()
-start_url = "about:blank" # Open a blank page first.
+browser_name = input("Enter the name of the browser you want to test (chrome, firefox, vivaldi, edge, opera, brave, safari, zen or another): ").lower()
+start_url = "about:blank"
 battery_test(website_list, browser_name, start_url)