def test_can_close_the_wormhole_and_transit(self): ftp = FileTransferProtocol(self.reactor, self.signals) ftp.open(None) ftp.shutdown() self.wormhole.close.assert_called() self.transit.close.assert_called()
def test_emits_signal_once_wormhole_is_closed(self): ftp = FileTransferProtocol(self.reactor, self.signals) ftp.open(None) ftp.shutdown() ftp._wormhole_delegate.wormhole_closed(result="ok") self.signals.wormhole_shutdown.emit.assert_called()
def test_doesnt_send_message_if_peer_connect_mode_not_supported(self): ftp = FileTransferProtocol(self.reactor, self.signals) wormhole_open = self.connect(self.signals.wormhole_open) ftp.open(None) wormhole_open() ftp.shutdown() self.wormhole.send_message.assert_not_called()
def test_doesnt_send_message_if_not_connected(self): ftp = FileTransferProtocol(self.reactor, self.signals) versions_received = self.connect(self.signals.versions_received) ftp.open(None) versions_received({"v0": {"mode": "connect"}}) ftp.shutdown() self.wormhole.send_message.assert_not_called()
def test_sends_message_if_connected_and_connect_mode_supported(self): ftp = FileTransferProtocol(self.reactor, self.signals) wormhole_open = self.connect(self.signals.wormhole_open) versions_received = self.connect(self.signals.versions_received) ftp.open(None) wormhole_open() versions_received({"v0": {"mode": "connect"}}) ftp.shutdown() self.wormhole.send_message.assert_called_with( b'{"command": "shutdown"}')
def test_still_emits_signal_if_wormhole_is_not_open(self): ftp = FileTransferProtocol(self.reactor, self.signals) ftp.shutdown() self.signals.wormhole_shutdown.emit.assert_called()