Fixed unit-tests

denisenkom · Jan 27, 2024 · b04d08e · b04d08e
1 parent 14b0284
commit b04d08e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/pytds/tds_session.py b/src/pytds/tds_session.py
@@ -1270,7 +1270,7 @@ def process_prelogin(self, login: _TdsLogin) -> None:
         size = len(p)
         if size <= 0 or resp_header.type != tds_base.PacketType.REPLY:
             self.bad_stream(
-                "Invalid packet type: {0}, expected PRELOGIN(4)".format(
+                "Invalid packet type: {0}, expected REPLY(4)".format(
                     self._reader.packet_type
                 )
             )

diff --git a/tests/simple_server.py b/tests/simple_server.py
@@ -181,7 +181,7 @@ def handle(self):
                     except OpenSSL.SSL.WantReadError:
                         pass
                     else:
-                        w.begin_packet(pytds.tds_base.PacketType.REPLY)
+                        w.begin_packet(pytds.tds_base.PacketType.PRELOGIN)
                         w.write(buf)
                         w.flush()
 
@@ -195,7 +195,7 @@ def handle(self):
                     except OpenSSL.SSL.WantReadError:
                         pass
                     else:
-                        w.begin_packet(pytds.tds_base.PacketType.REPLY)
+                        w.begin_packet(pytds.tds_base.PacketType.PRELOGIN)
                         w.write(buf)
                         w.flush()