fix: fix failing unit test because it can now accept a StackerDBHands…

…hakeAccept
stacks-network · Jan 27, 2023 · 9fb35ff · 9fb35ff
1 parent c5f0adb
commit 9fb35ff
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/net/chat.rs b/src/net/chat.rs
@@ -3901,7 +3901,8 @@ mod test {
 
             // received a valid HandshakeAccept from peer 2
             match reply_handshake_1.payload {
-                StacksMessageType::HandshakeAccept(ref data) => {
+                StacksMessageType::HandshakeAccept(ref data)
+                | StacksMessageType::StackerDBHandshakeAccept(ref data, ..) => {
                     assert_eq!(data.handshake.addrbytes, local_peer_2.addrbytes);
                     assert_eq!(data.handshake.port, local_peer_2.port);
                     assert_eq!(data.handshake.services, local_peer_2.services);
@@ -4242,7 +4243,8 @@ mod test {
 
             // received a valid HandshakeAccept from peer 2
             match reply_1.payload {
-                StacksMessageType::HandshakeAccept(ref data) => {
+                StacksMessageType::HandshakeAccept(ref data)
+                | StacksMessageType::StackerDBHandshakeAccept(ref data, ..) => {
                     assert_eq!(data.handshake.addrbytes, local_peer_2.addrbytes);
                     assert_eq!(data.handshake.port, local_peer_2.port);
                     assert_eq!(data.handshake.services, local_peer_2.services);