Beta

Author: I'm Matheus

.gitignore

@@ -3,4 +3,7 @@ src/api/__pycache__/
 src/core/__pycache__/
 src/middleware/__pycache__/
 src/models/__pycache__/
-temp/
+temp/
+dist/
+build/
+optifyx.spec

requirements.txt

@@ -14,6 +14,7 @@ setuptools
 pyperclip
 plyer
 qrcode
+websockets
 
 # For tests
 httpx

src/api/endpoint_misc.py

@@ -1,4 +1,5 @@
 from fastapi import APIRouter, HTTPException
+import socket
 from src.core.internet import get_public_ip
 
 router = APIRouter()
@@ -13,6 +14,10 @@ async def public_ip():
 @router.get("/check_online_connections")
 async def check_online_connections():
     try:
-        return {"status": "Online"}
+        device_name = socket.gethostname()
+        return {
+            "status": "Online",
+            "device_name": device_name
+        }
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))

tests/endpoints.py

@@ -37,5 +37,25 @@ def test_endpoints(self):
                 except Exception as e:
                     self.fail(f"{name} raised an exception: {e}")
 
+    def test_invalid_endpoints(self):
+        invalid_endpoints = {
+            "POST /invalid_start_device": lambda: requests.post(f"{BASE_URL}/invalid_start_device"),
+            "POST /invalid_connection_code": lambda: requests.post(f"{BASE_URL}/invalid_connection_code"),
+            "GET /invalid_system_realtime": lambda: requests.get(f"{BASE_URL}/invalid_system_realtime"),
+            "GET /invalid_disk_realtime": lambda: requests.get(f"{BASE_URL}/invalid_disk_realtime"),
+            "GET /invalid_home_realtime": lambda: requests.get(f"{BASE_URL}/invalid_home_realtime"),
+            "GET /invalid_security_realtime": lambda: requests.get(f"{BASE_URL}/invalid_security_realtime"),
+            "GET /invalid_webcam_check": lambda: requests.get(f"{BASE_URL}/invalid_webcam_check"),
+            "GET /invalid_public_ip": lambda: requests.get(f"{BASE_URL}/invalid_public_ip"),
+            "GET /invalid_check_online_connections": lambda: requests.get(f"{BASE_URL}/invalid_check_online_connections"),
+        }
+        for name, request_func in invalid_endpoints.items():
+            with self.subTest(endpoint=name):
+                try:
+                    response = request_func()
+                    self.assertTrue(response.status_code in [404, 405], f"{name} failed with status code {response.status_code}")
+                except Exception as e:
+                    self.fail(f"{name} raised an exception: {e}")
+
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()