123 lines
5.0 KiB
Python
123 lines
5.0 KiB
Python
from ipaddress import IPv4Address
|
|
|
|
from primaite.simulator.network.hardware.node_operating_state import NodeOperatingState
|
|
from primaite.simulator.network.hardware.nodes.server import Server
|
|
from primaite.simulator.system.applications.database_client import DatabaseClient
|
|
from primaite.simulator.system.services.database.database_service import DatabaseService
|
|
from primaite.simulator.system.services.ftp.ftp_server import FTPServer
|
|
from primaite.simulator.system.services.service import ServiceOperatingState
|
|
|
|
|
|
def test_database_client_server_connection(uc2_network):
|
|
web_server: Server = uc2_network.get_node_by_hostname("web_server")
|
|
db_client: DatabaseClient = web_server.software_manager.software.get("DatabaseClient")
|
|
|
|
db_server: Server = uc2_network.get_node_by_hostname("database_server")
|
|
db_service: DatabaseService = db_server.software_manager.software.get("DatabaseService")
|
|
|
|
assert len(db_service.connections) == 1
|
|
|
|
db_client.disconnect()
|
|
assert len(db_service.connections) == 0
|
|
|
|
|
|
def test_database_client_server_correct_password(uc2_network):
|
|
web_server: Server = uc2_network.get_node_by_hostname("web_server")
|
|
db_client: DatabaseClient = web_server.software_manager.software.get("DatabaseClient")
|
|
|
|
db_server: Server = uc2_network.get_node_by_hostname("database_server")
|
|
db_service: DatabaseService = db_server.software_manager.software.get("DatabaseService")
|
|
|
|
db_client.disconnect()
|
|
|
|
db_client.configure(server_ip_address=IPv4Address("192.168.1.14"), server_password="12345")
|
|
db_service.password = "12345"
|
|
|
|
assert db_client.connect()
|
|
|
|
assert len(db_service.connections) == 1
|
|
|
|
|
|
def test_database_client_server_incorrect_password(uc2_network):
|
|
web_server: Server = uc2_network.get_node_by_hostname("web_server")
|
|
db_client: DatabaseClient = web_server.software_manager.software.get("DatabaseClient")
|
|
|
|
db_server: Server = uc2_network.get_node_by_hostname("database_server")
|
|
db_service: DatabaseService = db_server.software_manager.software.get("DatabaseService")
|
|
|
|
db_client.disconnect()
|
|
db_client.configure(server_ip_address=IPv4Address("192.168.1.14"), server_password="54321")
|
|
db_service.password = "12345"
|
|
|
|
assert not db_client.connect()
|
|
assert len(db_service.connections) == 0
|
|
|
|
|
|
def test_database_client_query(uc2_network):
|
|
"""Tests DB query across the network returns HTTP status 200 and date."""
|
|
web_server: Server = uc2_network.get_node_by_hostname("web_server")
|
|
db_client: DatabaseClient = web_server.software_manager.software.get("DatabaseClient")
|
|
|
|
assert db_client.connected
|
|
|
|
assert db_client.query("SELECT")
|
|
|
|
|
|
def test_create_database_backup(uc2_network):
|
|
"""Run the backup_database method and check if the FTP server has the relevant file."""
|
|
db_server: Server = uc2_network.get_node_by_hostname("database_server")
|
|
db_service: DatabaseService = db_server.software_manager.software.get("DatabaseService")
|
|
|
|
# back up should be created
|
|
assert db_service.backup_database() is True
|
|
|
|
backup_server: Server = uc2_network.get_node_by_hostname("backup_server")
|
|
ftp_server: FTPServer = backup_server.software_manager.software.get("FTPServer")
|
|
|
|
# backup file should exist in the backup server
|
|
assert ftp_server.file_system.get_file(folder_name=db_service.uuid, file_name="database.db") is not None
|
|
|
|
|
|
def test_restore_backup(uc2_network):
|
|
"""Run the restore_backup method and check if the backup is properly restored."""
|
|
db_server: Server = uc2_network.get_node_by_hostname("database_server")
|
|
db_service: DatabaseService = db_server.software_manager.software.get("DatabaseService")
|
|
|
|
# create a back up
|
|
assert db_service.backup_database() is True
|
|
|
|
# delete database locally
|
|
db_service.file_system.delete_file(folder_name="database", file_name="database.db")
|
|
|
|
assert db_service.file_system.get_file(folder_name="database", file_name="database.db") is None
|
|
|
|
# back up should be restored
|
|
assert db_service.restore_backup() is True
|
|
|
|
assert db_service.file_system.get_file(folder_name="database", file_name="database.db") is not None
|
|
|
|
|
|
def test_database_client_cannot_query_offline_database_server(uc2_network):
|
|
"""Tests DB query across the network returns HTTP status 404 when db server is offline."""
|
|
db_server: Server = uc2_network.get_node_by_hostname("database_server")
|
|
db_service: DatabaseService = db_server.software_manager.software.get("DatabaseService")
|
|
|
|
assert db_server.operating_state is NodeOperatingState.ON
|
|
assert db_service.operating_state is ServiceOperatingState.RUNNING
|
|
|
|
web_server: Server = uc2_network.get_node_by_hostname("web_server")
|
|
db_client: DatabaseClient = web_server.software_manager.software.get("DatabaseClient")
|
|
assert db_client.connected
|
|
|
|
assert db_client.query("SELECT") is True
|
|
|
|
db_server.power_off()
|
|
|
|
for i in range(db_server.shut_down_duration + 1):
|
|
uc2_network.apply_timestep(timestep=i)
|
|
|
|
assert db_server.operating_state is NodeOperatingState.OFF
|
|
assert db_service.operating_state is ServiceOperatingState.STOPPED
|
|
|
|
assert db_client.query("SELECT") is False
|