Address comments

bloomberg · Apr 8, 2022 · 71f0992 · 71f0992
1 parent 2ecc6f5
commit 71f0992
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/amqpprox_session.t.cpp b/tests/amqpprox_session.t.cpp
@@ -368,13 +368,15 @@ void SessionTest::testSetupProxyOutOfOrderOpen(int idx)
 
 void SessionTest::testSetupProxyReceiveClose(int idx)
 {
-    // Client  <--------Close------  Proxy  <--------Close------  Broker
-    d_clientState.pushItem(idx, Data(encode(close())));
+    // Client  <--------Close-----  Proxy  <--------Close------  Broker
+    methods::Close receivedClose;
+    receivedClose.setReply(123, "Broker is closing the connection");
+    d_clientState.pushItem(idx, Data(encode(receivedClose)));
 
-    d_serverState.expect(idx, [this](const auto &items) {
+    d_serverState.expect(idx, [this, receivedClose](const auto &items) {
         auto data = filterVariant<Data>(items);
         ASSERT_EQ(data.size(), 1);
-        EXPECT_EQ(data[0], Data(encode(close())));
+        EXPECT_EQ(data[0], Data(encode(receivedClose)));
         EXPECT_THAT(items, Contains(VariantWith<Call>(Call("shutdown"))));
         EXPECT_THAT(items, Contains(VariantWith<Call>(Call("close"))));
     });