#1859 - Added the call to file system reset

src/primaite/simulator/system/services/ftp/ftp_server.py

@@ -29,6 +29,19 @@ class FTPServer(FTPServiceABC):
         super().__init__(**kwargs)
         self.start()
 
+    def set_original_state(self):
+        """Sets the original state."""
+        print(f"Setting FTPServer original state on node {self.software_manager.node.hostname}")
+        super().set_original_state()
+        vals_to_include = {"server_password"}
+        self._original_state.update(self.model_dump(include=vals_to_include))
+
+    def reset_component_for_episode(self, episode: int):
+        """Reset the original state of the SimComponent."""
+        print(f"Resetting FTPServer state on node {self.software_manager.node.hostname}")
+        self.connections.clear()
+        super().reset_component_for_episode(episode)
+
     def _process_ftp_command(self, payload: FTPPacket, session_id: Optional[str] = None, **kwargs) -> FTPPacket:
         """
         Process the command in the FTP Packet.