def test_parse_address(self): self.assertEqual( ('127.0.0.1', 34805), parse_address('227 Now Entering Passive Mode (127,0,0,1,135,245)')) self.assertEqual( ('127.0.0.1', 254), parse_address('227 Passive Mode! (127, 000, 000, 001, 000, 254)'))
def test_parse_address(self): self.assertEqual( ('127.0.0.1', 34805), parse_address('227 Now Entering Passive Mode (127,0,0,1,135,245)') ) self.assertEqual( ('127.0.0.1', 254), parse_address('227 Passive Mode! (127, 000, 000, 001, 000, 254)') )
def test_control_stream(self): def log_cb(data_type, data): _logger.debug(__('{0}={1}', data_type, data)) connection = Connection(('127.0.0.1', self.server_port())) yield From(connection.connect()) control_stream = ControlStream(connection) control_stream.data_observer.add(log_cb) reply = yield From(control_stream.read_reply()) self.assertEqual(220, reply.code) yield From(control_stream.write_command(Command('USER', 'smaug'))) reply = yield From(control_stream.read_reply()) self.assertEqual(331, reply.code) yield From(control_stream.write_command(Command('PASS', 'gold1'))) reply = yield From(control_stream.read_reply()) self.assertEqual(230, reply.code) yield From(control_stream.write_command(Command('PASV'))) reply = yield From(control_stream.read_reply()) self.assertEqual(227, reply.code) address = parse_address(reply.text) data_connection = Connection(address) yield From(data_connection.connect()) data_stream = DataStream(data_connection) yield From( control_stream.write_command(Command('RETR', 'example (copy).txt'))) reply = yield From(control_stream.read_reply()) self.assertEqual(150, reply.code) my_file = io.BytesIO() yield From(data_stream.read_file(my_file)) reply = yield From(control_stream.read_reply()) self.assertEqual(226, reply.code) self.assertEqual('The real treasure is in Smaug’s heart 💗.\n', my_file.getvalue().decode('utf-8'))
def test_control_stream(self): def log_cb(data_type, data): _logger.debug(__('{0}={1}', data_type, data)) connection = Connection(('127.0.0.1', self.server_port())) yield From(connection.connect()) control_stream = ControlStream(connection) control_stream.data_observer.add(log_cb) reply = yield From(control_stream.read_reply()) self.assertEqual(220, reply.code) yield From(control_stream.write_command(Command('USER', 'smaug'))) reply = yield From(control_stream.read_reply()) self.assertEqual(331, reply.code) yield From(control_stream.write_command(Command('PASS', 'gold1'))) reply = yield From(control_stream.read_reply()) self.assertEqual(230, reply.code) yield From(control_stream.write_command(Command('PASV'))) reply = yield From(control_stream.read_reply()) self.assertEqual(227, reply.code) address = parse_address(reply.text) data_connection = Connection(address) yield From(data_connection.connect()) data_stream = DataStream(data_connection) yield From(control_stream.write_command(Command('RETR', 'example (copy).txt'))) reply = yield From(control_stream.read_reply()) self.assertEqual(150, reply.code) my_file = io.BytesIO() yield From(data_stream.read_file(my_file)) reply = yield From(control_stream.read_reply()) self.assertEqual(226, reply.code) self.assertEqual( 'The real treasure is in Smaug’s heart 💗.\n', my_file.getvalue().decode('utf-8') )