From fb37eba2d076965251706223d1b742ff73e8f478 Mon Sep 17 00:00:00 2001
From: Christian Sahlmann <christian.sahlmann@telekom.de>
Date: Wed, 7 Feb 2024 22:03:36 +0100
Subject: [PATCH] telnet: fix prompt handling for config commands

---
 mb_netmgmt/__init__.py | 2 +-
 mb_netmgmt/telnet.py   | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/mb_netmgmt/__init__.py b/mb_netmgmt/__init__.py
index 327d14a..4083d23 100644
--- a/mb_netmgmt/__init__.py
+++ b/mb_netmgmt/__init__.py
@@ -18,7 +18,7 @@
 # along with mb-netmgmt. If not, see <https://www.gnu.org/licenses/
 
 """Network Management Protocols for Mountebank"""
-__version__ = "0.0.80"
+__version__ = "0.0.81"
 
 import os
 import subprocess
diff --git a/mb_netmgmt/telnet.py b/mb_netmgmt/telnet.py
index 564637e..fe5f414 100644
--- a/mb_netmgmt/telnet.py
+++ b/mb_netmgmt/telnet.py
@@ -54,7 +54,7 @@ def handle_command(self, command, request_id):
         return self.handle_request({"command": command.decode()}, request_id)
 
     def send_upstream(self, request, request_id):
-        self.telnet.execute(request["command"])
+        self.telnet.send(request["command"])
 
     def read_request(self):
         return self.rfile.readline(), None
@@ -65,7 +65,8 @@ def respond(self, response, request_id):
             self.stopped = True
 
     def read_proxy_response(self):
-        return {"response": self.telnet.response}
+        _, match = self.telnet.expect_prompt()
+        return {"response": match.string}
 
     def handle_username_prompt(self):
         username_prompt = b"Username: "