123 lines
5.2 KiB
Python
123 lines
5.2 KiB
Python
# © Crown-owned copyright 2025, Defence Science and Technology Laboratory UK
|
|
from typing import Tuple
|
|
|
|
import pytest
|
|
|
|
from primaite.simulator.network.hardware.node_operating_state import NodeOperatingState
|
|
from primaite.simulator.network.hardware.nodes.host.computer import Computer
|
|
from primaite.simulator.network.hardware.nodes.host.server import Server
|
|
from primaite.simulator.network.protocols.http import HttpStatusCode
|
|
from primaite.simulator.system.applications.application import ApplicationOperatingState
|
|
from primaite.simulator.system.applications.web_browser import WebBrowser
|
|
from primaite.simulator.system.services.dns.dns_client import DNSClient
|
|
from primaite.simulator.system.services.dns.dns_server import DNSServer
|
|
from primaite.simulator.system.services.web_server.web_server import WebServer
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def web_client_and_web_server(client_server) -> Tuple[WebBrowser, Computer, WebServer, Server]:
|
|
computer, server = client_server
|
|
|
|
# Install Web Browser on computer
|
|
computer.software_manager.install(WebBrowser)
|
|
web_browser: WebBrowser = computer.software_manager.software.get("WebBrowser")
|
|
web_browser.run()
|
|
|
|
# Install DNS Client service on computer
|
|
computer.software_manager.install(DNSClient)
|
|
dns_client: DNSClient = computer.software_manager.software.get("DNSClient")
|
|
# set dns server
|
|
dns_client.dns_server = server.network_interfaces[next(iter(server.network_interfaces))].ip_address
|
|
|
|
# Install Web Server service on server
|
|
server.software_manager.install(WebServer)
|
|
web_server_service: WebServer = server.software_manager.software.get("WebServer")
|
|
web_server_service.start()
|
|
|
|
# Install DNS Server service on server
|
|
server.software_manager.install(DNSServer)
|
|
dns_server: DNSServer = server.software_manager.software.get("DNSServer")
|
|
# register arcd.com to DNS
|
|
dns_server.dns_register(
|
|
domain_name="arcd.com",
|
|
domain_ip_address=server.network_interfaces[next(iter(server.network_interfaces))].ip_address,
|
|
)
|
|
|
|
return web_browser, computer, web_server_service, server
|
|
|
|
|
|
def test_web_page_get_users_page_request_with_domain_name(web_client_and_web_server):
|
|
"""Test to see if the client can handle requests with domain names"""
|
|
web_browser_app, computer, web_server_service, server = web_client_and_web_server
|
|
|
|
web_server_ip = server.network_interfaces.get(next(iter(server.network_interfaces))).ip_address
|
|
web_browser_app.config.target_url = f"http://arcd.com/"
|
|
assert web_browser_app.operating_state == ApplicationOperatingState.RUNNING
|
|
|
|
assert web_browser_app.get_webpage() is True
|
|
|
|
# latest response should have status code 200
|
|
assert web_browser_app.latest_response is not None
|
|
assert web_browser_app.latest_response.status_code == HttpStatusCode.OK
|
|
|
|
|
|
def test_web_page_get_users_page_request_with_ip_address(web_client_and_web_server):
|
|
"""Test to see if the client can handle requests that use ip_address."""
|
|
web_browser_app, computer, web_server_service, server = web_client_and_web_server
|
|
|
|
web_server_ip = server.network_interfaces.get(next(iter(server.network_interfaces))).ip_address
|
|
web_browser_app.config.target_url = f"http://{web_server_ip}/"
|
|
assert web_browser_app.operating_state == ApplicationOperatingState.RUNNING
|
|
|
|
assert web_browser_app.get_webpage() is True
|
|
|
|
# latest response should have status code 200
|
|
assert web_browser_app.latest_response is not None
|
|
assert web_browser_app.latest_response.status_code == HttpStatusCode.OK
|
|
|
|
|
|
def test_web_page_request_from_shut_down_server(web_client_and_web_server):
|
|
"""Test to see that the web server does not respond when the server is off."""
|
|
web_browser_app, computer, web_server_service, server = web_client_and_web_server
|
|
|
|
web_server_ip = server.network_interfaces.get(next(iter(server.network_interfaces))).ip_address
|
|
web_browser_app.config.target_url = f"http://arcd.com/"
|
|
assert web_browser_app.operating_state == ApplicationOperatingState.RUNNING
|
|
|
|
assert web_browser_app.get_webpage() is True
|
|
|
|
# latest response should have status code 200
|
|
assert web_browser_app.latest_response is not None
|
|
assert web_browser_app.latest_response.status_code == HttpStatusCode.OK
|
|
|
|
server.power_off()
|
|
|
|
server.power_off()
|
|
|
|
for i in range(server.shut_down_duration + 1):
|
|
server.apply_timestep(timestep=i)
|
|
|
|
# node should be off
|
|
assert server.operating_state is NodeOperatingState.OFF
|
|
|
|
assert web_browser_app.get_webpage() is False
|
|
assert web_browser_app.latest_response.status_code == HttpStatusCode.NOT_FOUND
|
|
|
|
|
|
def test_web_page_request_from_closed_web_browser(web_client_and_web_server):
|
|
web_browser_app, computer, web_server_service, server = web_client_and_web_server
|
|
|
|
assert web_browser_app.operating_state == ApplicationOperatingState.RUNNING
|
|
web_browser_app.config.target_url = f"http://arcd.com/"
|
|
assert web_browser_app.get_webpage() is True
|
|
|
|
# latest response should have status code 200
|
|
assert web_browser_app.latest_response.status_code == HttpStatusCode.OK
|
|
|
|
web_browser_app.close()
|
|
|
|
# node should be off
|
|
assert web_browser_app.operating_state is ApplicationOperatingState.CLOSED
|
|
|
|
assert web_browser_app.get_webpage() is False
|