#1916: FTP client STOR request to FTP server

tests/integration_tests/system/test_ftp_client_server.py

@@ -0,0 +1,59 @@
+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: FTPServer = backup_server.software_manager.software["FTPServer"]
+
+    assert ftp_client.operating_state == ServiceOperatingState.RUNNING
+    assert ftp_server.operating_state == ServiceOperatingState.RUNNING
+
+    # create file on ftp client
+    ftp_client.file_system.create_file(file_name="test_file.txt")
+
+    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.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: FTPServer = backup_server.software_manager.software["FTPServer"]
+
+    assert ftp_client.operating_state == ServiceOperatingState.RUNNING
+    assert ftp_server.operating_state == ServiceOperatingState.RUNNING
+
+    # create file on ftp server
+    ftp_server.file_system.create_file(file_name="test_file.txt", folder_name="file_share")
+
+    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,
+    )

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

@@ -0,0 +1,41 @@
+from ipaddress import IPv4Address
+
+import pytest
+
+from primaite.simulator.network.hardware.base import Node
+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
+from primaite.simulator.system.services.ftp.ftp_server import FTPServer
+
+
+@pytest.fixture(scope="function")
+def ftp_server() -> Node:
+    node = Node(hostname="ftp_server")
+    node.software_manager.install(software_class=FTPServer)
+    node.software_manager.software["FTPServer"].start()
+    return node
+
+
+@pytest.fixture(scope="function")
+def ftp_client() -> Node:
+    node = Node(hostname="ftp_client")
+    node.software_manager.install(software_class=FTPClient)
+    node.software_manager.software["FTPClient"].start()
+    return node
+
+
+def test_create_ftp_server(ftp_server):
+    assert ftp_server is not None
+    ftp_server_service: FTPServer = ftp_server.software_manager.software["FTPServer"]
+    assert ftp_server_service.name is "FTPServer"
+    assert ftp_server_service.port is Port.FTP
+    assert ftp_server_service.protocol is IPProtocol.TCP
+
+
+def test_create_ftp_client(ftp_client):
+    assert ftp_client is not None
+    ftp_client_service: FTPClient = ftp_client.software_manager.software["FTPClient"]
+    assert ftp_client_service.name is "FTPClient"
+    assert ftp_client_service.port is Port.FTP
+    assert ftp_client_service.protocol is IPProtocol.TCP