Add a test case for echo message operation

Author: yourslab

eventstream_rpc/tests/CMakeLists.txt

@@ -44,6 +44,7 @@ aws_use_package(aws-crt-cpp)
 aws_use_package(EventstreamRpc-cpp)
 
 add_test_case(EventStreamConnect)
+add_test_case(OperateWhileDisconnected)
 generate_cpp_test_driver(${TEST_BINARY_NAME})
 target_include_directories(${TEST_BINARY_NAME} PUBLIC
     $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>

eventstream_rpc/tests/EventStreamClientTest.cpp

@@ -12,8 +12,10 @@
 #include <sstream>
 
 using namespace Aws::Eventstreamrpc;
+using namespace Awstest;
 
 static int s_TestEventStreamConnect(struct aws_allocator *allocator, void *ctx);
+static int s_TestOperationWhileDisconnected(struct aws_allocator *allocator, void *ctx);
 
 class TestLifecycleHandler : public ConnectionLifecycleHandler
 {
@@ -54,6 +56,11 @@ class TestLifecycleHandler : public ConnectionLifecycleHandler
     int lastErrorCode;
 };
 
+static void s_onMessageFlush(int errorCode)
+{
+    (void) errorCode;
+}
+
 static int s_TestEventStreamConnect(struct aws_allocator *allocator, void *ctx)
 {
     (void)ctx;
@@ -75,7 +82,7 @@ static int s_TestEventStreamConnect(struct aws_allocator *allocator, void *ctx)
         ASSERT_TRUE(clientBootstrap);
         clientBootstrap.EnableBlockingShutdown();
         MessageAmendment connectionAmendment;
-        auto messageAmender = [&](void) -> MessageAmendment & { return connectionAmendment; };
+        auto messageAmender = [&](void) -> const MessageAmendment & { return connectionAmendment; };
 
         ConnectionConfig config;
         config.SetHostName(Aws::Crt::String("127.0.0.1"));
@@ -113,4 +120,58 @@ static int s_TestEventStreamConnect(struct aws_allocator *allocator, void *ctx)
     return AWS_OP_SUCCESS;
 }
 
+static int s_TestOperationWhileDisconnected(struct aws_allocator *allocator, void *ctx)
+{
+    (void)ctx;
+    {
+        Aws::Crt::ApiHandle apiHandle(allocator);
+        Aws::Crt::Io::TlsContextOptions tlsCtxOptions = Aws::Crt::Io::TlsContextOptions::InitDefaultClient();
+        Aws::Crt::Io::TlsContext tlsContext(tlsCtxOptions, Aws::Crt::Io::TlsMode::CLIENT, allocator);
+        ASSERT_TRUE(tlsContext);
+
+        Aws::Crt::Io::TlsConnectionOptions tlsConnectionOptions = tlsContext.NewConnectionOptions();
+
+        Aws::Crt::Io::EventLoopGroup eventLoopGroup(0, allocator);
+        ASSERT_TRUE(eventLoopGroup);
+
+        Aws::Crt::Io::DefaultHostResolver defaultHostResolver(eventLoopGroup, 8, 30, allocator);
+        ASSERT_TRUE(defaultHostResolver);
+
+        Aws::Crt::Io::ClientBootstrap clientBootstrap(eventLoopGroup, defaultHostResolver, allocator);
+        ASSERT_TRUE(clientBootstrap);
+        clientBootstrap.EnableBlockingShutdown();
+        MessageAmendment connectionAmendment;
+        auto messageAmender = [&](void) -> MessageAmendment & { return connectionAmendment; };
+
+        {
+            ConnectionLifecycleHandler lifecycleHandler;
+            Awstest::EchoTestRpcClient client(clientBootstrap, allocator);
+            auto connectedStatus = client.Connect(lifecycleHandler);
+            ASSERT_TRUE(connectedStatus.get().baseStatus == EVENT_STREAM_RPC_SUCCESS);
+            auto echoMessage = client.NewEchoMessage();
+            EchoMessageRequest echoMessageRequest;
+            MessageData messageData;
+            messageData.SetStringMessage("wtf");
+            echoMessageRequest.SetMessage(messageData);
+            auto requestFuture = echoMessage.Activate(echoMessageRequest, s_onMessageFlush);
+            ASSERT_TRUE(requestFuture.get().baseStatus == EVENT_STREAM_RPC_CONNECTION_CLOSED);
+            auto result = echoMessage.GetResult().get();
+            if (result) {
+                std::cout << result.GetOperationResponse() << std::endl;
+            } else {
+                auto errorType = result.GetResultType();
+                if (errorType == RPC_ERROR) {
+                    std::cout << result.GetRpcError() << std::endl;
+                } else {
+                    auto *error = result.GetOperationError();
+                    (void)error;
+                }
+            }
+        }
+    }
+
+    return AWS_OP_SUCCESS;
+}
+
 AWS_TEST_CASE(EventStreamConnect, s_TestEventStreamConnect)
+AWS_TEST_CASE(OperateWhileDisconnected, s_TestOperationWhileDisconnected)