update test/ogawayama/stub/endpoint.h to handle UpdateExpirationTime

project-tsurugi · Jul 18, 2024 · 9da2776 · 9da2776
1 parent 860607f
commit 9da2776
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/test/ogawayama/stub/endpoint.h b/test/ogawayama/stub/endpoint.h
@@ -157,6 +157,22 @@ class endpoint {
                         auto reply_message = ss.str();
                         wire_->get_response_wire().write(reply_message.data(), tateyama::common::wire::response_header(index, reply_message.length(), RESPONSE_HANDSHAKE));
                         continue;
+                    } else if (service_id == tateyama::framework::service_id_routing) {
+                        std::stringstream ss{};
+                        ::tateyama::proto::framework::response::Header header{};
+                        if(auto res = tateyama::utils::SerializeDelimitedToOstream(header, std::addressof(ss)); ! res) {
+                            throw std::runtime_error("error formatting response message");
+                        }
+                        tateyama::proto::core::response::UpdateExpirationTime rp{};
+                        (void) rp.mutable_success();
+                        auto body = rp.SerializeAsString();
+                        if(auto res = tateyama::utils::PutDelimitedBodyToOstream(body, std::addressof(ss)); ! res) {
+                            throw std::runtime_error("error formatting response message");
+                        }
+                        rp.clear_success();
+                        auto reply_message = ss.str();
+                        wire_->get_response_wire().write(reply_message.data(), tateyama::common::wire::response_header(index, reply_message.length(), RESPONSE_HANDSHAKE));
+                        continue;
                     }
                 }