Merge branch 'origin/dev' into feature/1812-traverse-actions-dict

tests/unit_tests/_primaite/_simulator/_system/_applications/test_web_browser.py

@@ -0,0 +1,39 @@
+import pytest
+
+from primaite.simulator.network.hardware.nodes.computer import Computer
+from primaite.simulator.network.protocols.http import HttpResponsePacket, HttpStatusCode
+from primaite.simulator.network.transmission.network_layer import IPProtocol
+from primaite.simulator.network.transmission.transport_layer import Port
+from primaite.simulator.system.applications.web_browser import WebBrowser
+
+
+@pytest.fixture(scope="function")
+def web_client() -> Computer:
+    node = Computer(
+        hostname="web_client", ip_address="192.168.1.11", subnet_mask="255.255.255.0", default_gateway="192.168.1.1"
+    )
+    return node
+
+
+def test_create_web_client(web_client):
+    assert web_client is not None
+    web_browser: WebBrowser = web_client.software_manager.software["WebBrowser"]
+    assert web_browser.name is "WebBrowser"
+    assert web_browser.port is Port.HTTP
+    assert web_browser.protocol is IPProtocol.TCP
+
+
+def test_receive_invalid_payload(web_client):
+    web_browser: WebBrowser = web_client.software_manager.software["WebBrowser"]
+
+    assert web_browser.receive(payload={}) is False
+
+
+def test_receive_payload(web_client):
+    payload = HttpResponsePacket(status_code=HttpStatusCode.OK)
+    web_browser: WebBrowser = web_client.software_manager.software["WebBrowser"]
+    assert web_browser.latest_response is None
+
+    web_browser.receive(payload=payload)
+
+    assert web_browser.latest_response is not None

tests/unit_tests/_primaite/_simulator/_system/_services/test_dns.py

@@ -25,7 +25,11 @@ def dns_server() -> Node:
 @pytest.fixture(scope="function")
 def dns_client() -> Node:
     node = Computer(
-        hostname="dns_client", ip_address="192.168.1.11", subnet_mask="255.255.255.0", default_gateway="192.168.1.1"
+        hostname="dns_client",
+        ip_address="192.168.1.11",
+        subnet_mask="255.255.255.0",
+        default_gateway="192.168.1.1",
+        dns_server=IPv4Address("192.168.1.10"),
     )
     return node
 

tests/unit_tests/_primaite/_simulator/_system/_services/test_ftp.py

@@ -5,7 +5,7 @@ import pytest
 from primaite.simulator.network.hardware.base import Node
 from primaite.simulator.network.hardware.nodes.computer import Computer
 from primaite.simulator.network.hardware.nodes.server import Server
-from primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket
+from primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode
 from primaite.simulator.network.transmission.network_layer import IPProtocol
 from primaite.simulator.network.transmission.transport_layer import Port
 from primaite.simulator.system.services.ftp.ftp_client import FTPClient
@@ -78,6 +78,7 @@ def test_ftp_client_store_file(ftp_client):
             "file_size": 24,
         },
         packet_payload_size=24,
+        status_code=FTPStatusCode.OK,
     )
 
     ftp_client_service: FTPClient = ftp_client.software_manager.software["FTPClient"]

tests/unit_tests/_primaite/_simulator/_system/_services/test_web_server.py

@@ -0,0 +1,64 @@
+import pytest
+
+from primaite.simulator.network.hardware.nodes.server import Server
+from primaite.simulator.network.protocols.http import (
+    HttpRequestMethod,
+    HttpRequestPacket,
+    HttpResponsePacket,
+    HttpStatusCode,
+)
+from primaite.simulator.network.transmission.network_layer import IPProtocol
+from primaite.simulator.network.transmission.transport_layer import Port
+from primaite.simulator.system.services.web_server.web_server import WebServer
+
+
+@pytest.fixture(scope="function")
+def web_server() -> Server:
+    node = Server(
+        hostname="web_server", ip_address="192.168.1.10", subnet_mask="255.255.255.0", default_gateway="192.168.1.1"
+    )
+    node.software_manager.install(software_class=WebServer)
+    node.software_manager.software["WebServer"].start()
+    return node
+
+
+def test_create_web_server(web_server):
+    assert web_server is not None
+    web_server_service: WebServer = web_server.software_manager.software["WebServer"]
+    assert web_server_service.name is "WebServer"
+    assert web_server_service.port is Port.HTTP
+    assert web_server_service.protocol is IPProtocol.TCP
+
+
+def test_handling_get_request_not_found_path(web_server):
+    payload = HttpRequestPacket(request_method=HttpRequestMethod.GET, request_url="http://domain.com/fake-path")
+
+    web_server_service: WebServer = web_server.software_manager.software["WebServer"]
+
+    response: HttpResponsePacket = web_server_service._handle_get_request(payload=payload)
+    assert response.status_code == HttpStatusCode.NOT_FOUND
+
+
+def test_handling_get_request_home_page(web_server):
+    payload = HttpRequestPacket(request_method=HttpRequestMethod.GET, request_url="http://domain.com/")
+
+    web_server_service: WebServer = web_server.software_manager.software["WebServer"]
+
+    response: HttpResponsePacket = web_server_service._handle_get_request(payload=payload)
+    assert response.status_code == HttpStatusCode.OK
+
+
+def test_process_http_request_get(web_server):
+    payload = HttpRequestPacket(request_method=HttpRequestMethod.GET, request_url="http://domain.com/")
+
+    web_server_service: WebServer = web_server.software_manager.software["WebServer"]
+
+    assert web_server_service._process_http_request(payload=payload) is True
+
+
+def test_process_http_request_method_not_allowed(web_server):
+    payload = HttpRequestPacket(request_method=HttpRequestMethod.DELETE, request_url="http://domain.com/")
+
+    web_server_service: WebServer = web_server.software_manager.software["WebServer"]
+
+    assert web_server_service._process_http_request(payload=payload) is False