def test_ssh_handler_writes_a_file(self): class StubbedSFTPClient(StubbedGetter): def get(self, remote_path, local_path): with open(local_path, "w") as writer: with open(remote_path, "r") as reader: writer.write(reader.read()) expected_path = "/tmp/fake_file" remote_path = "sftp://localhost/%s" % test.do_test_file("config.gem") url = urlparse.urlparse(remote_path) sftp_handler = handlers.SFTPHandler(url, expected_path) guaranteed_file = sftp_handler.handle(getter=StubbedSFTPClient) self.assertTrue(os.path.isfile(guaranteed_file)) os.unlink(guaranteed_file)
def test_ssh_handler_raises_on_bad_netloc(self): url = urlparse.urlparse("sftp://doesntexist/path/to/file") sftp_handler = handlers.SFTPHandler(url, '') self.assertRaises(handlers.HandlerError, sftp_handler.handle)
def test_ssh_handler_raises_on_bad_credentials(self): url = urlparse.urlparse( "sftp://*****:*****@localhost:22/path/to/file") sftp_handler = handlers.SFTPHandler(url, '') self.assertRaises(handlers.HandlerError, sftp_handler.handle)