100 lines
4.2 KiB
Python
100 lines
4.2 KiB
Python
from ipaddress import IPv4Address
|
|
|
|
from primaite.simulator.network.hardware.nodes.computer import Computer
|
|
from primaite.simulator.network.hardware.nodes.server import Server
|
|
from primaite.simulator.system.services.ftp.ftp_client import FTPClient
|
|
from primaite.simulator.system.services.ftp.ftp_server import FTPServer
|
|
from primaite.simulator.system.services.service import ServiceOperatingState
|
|
|
|
|
|
def test_ftp_client_store_file_in_server(uc2_network):
|
|
"""
|
|
Test checks to see if the client is able to store files in the backup server.
|
|
"""
|
|
client_1: Computer = uc2_network.get_node_by_hostname("client_1")
|
|
backup_server: Server = uc2_network.get_node_by_hostname("backup_server")
|
|
|
|
ftp_client: FTPClient = client_1.software_manager.software["FTPClient"]
|
|
ftp_server_service: FTPServer = backup_server.software_manager.software["FTPServer"]
|
|
|
|
assert ftp_client.operating_state == ServiceOperatingState.RUNNING
|
|
assert ftp_server_service.operating_state == ServiceOperatingState.RUNNING
|
|
|
|
# create file on ftp client
|
|
ftp_client.file_system.create_file(file_name="test_file.txt")
|
|
|
|
assert ftp_client.send_file(
|
|
src_folder_name="root",
|
|
src_file_name="test_file.txt",
|
|
dest_folder_name="client_1_backup",
|
|
dest_file_name="test_file.txt",
|
|
dest_ip_address=backup_server.nics.get(next(iter(backup_server.nics))).ip_address,
|
|
)
|
|
|
|
assert ftp_server_service.file_system.get_file(folder_name="client_1_backup", file_name="test_file.txt")
|
|
|
|
|
|
def test_ftp_client_retrieve_file_from_server(uc2_network):
|
|
"""
|
|
Test checks to see if the client is able to retrieve files from the backup server.
|
|
"""
|
|
client_1: Computer = uc2_network.get_node_by_hostname("client_1")
|
|
backup_server: Server = uc2_network.get_node_by_hostname("backup_server")
|
|
|
|
ftp_client: FTPClient = client_1.software_manager.software["FTPClient"]
|
|
ftp_server_service: FTPServer = backup_server.software_manager.software["FTPServer"]
|
|
|
|
assert ftp_client.operating_state == ServiceOperatingState.RUNNING
|
|
assert ftp_server_service.operating_state == ServiceOperatingState.RUNNING
|
|
|
|
# create file on ftp server
|
|
ftp_server_service.file_system.create_file(file_name="test_file.txt", folder_name="file_share")
|
|
|
|
assert ftp_client.request_file(
|
|
src_folder_name="file_share",
|
|
src_file_name="test_file.txt",
|
|
dest_folder_name="downloads",
|
|
dest_file_name="test_file.txt",
|
|
dest_ip_address=backup_server.nics.get(next(iter(backup_server.nics))).ip_address,
|
|
)
|
|
|
|
# client should have retrieved the file
|
|
assert ftp_client.file_system.get_file(folder_name="downloads", file_name="test_file.txt")
|
|
|
|
|
|
def test_ftp_client_tries_to_connect_to_offline_server(uc2_network):
|
|
"""Test checks to make sure that the client can't do anything when the server is offline."""
|
|
client_1: Computer = uc2_network.get_node_by_hostname("client_1")
|
|
backup_server: Server = uc2_network.get_node_by_hostname("backup_server")
|
|
|
|
ftp_client: FTPClient = client_1.software_manager.software["FTPClient"]
|
|
ftp_server_service: FTPServer = backup_server.software_manager.software["FTPServer"]
|
|
|
|
assert ftp_client.operating_state == ServiceOperatingState.RUNNING
|
|
assert ftp_server_service.operating_state == ServiceOperatingState.RUNNING
|
|
|
|
# create file on ftp server
|
|
ftp_server_service.file_system.create_file(file_name="test_file.txt", folder_name="file_share")
|
|
|
|
backup_server.power_off()
|
|
|
|
for i in range(backup_server.shut_down_duration + 1):
|
|
uc2_network.apply_timestep(timestep=i)
|
|
|
|
assert ftp_client.operating_state == ServiceOperatingState.RUNNING
|
|
assert ftp_server_service.operating_state == ServiceOperatingState.STOPPED
|
|
|
|
assert (
|
|
ftp_client.request_file(
|
|
src_folder_name="file_share",
|
|
src_file_name="test_file.txt",
|
|
dest_folder_name="downloads",
|
|
dest_file_name="test_file.txt",
|
|
dest_ip_address=backup_server.nics.get(next(iter(backup_server.nics))).ip_address,
|
|
)
|
|
is False
|
|
)
|
|
|
|
# client should have retrieved the file
|
|
assert ftp_client.file_system.get_file(folder_name="downloads", file_name="test_file.txt") is None
|