| 158 | } |
| 159 | |
| 160 | void TestMaxMessageSize(std::string subscriber_id, bool expect_throw, |
| 161 | bool is_external_facing = true) { |
| 162 | int port = GetServerPort(); |
| 163 | ThriftServer* server; |
| 164 | ThriftServerBuilder server_builder("DummyStatestore", MakeProcessor(), port); |
| 165 | server_builder |
| 166 | .is_external_facing(is_external_facing) |
| 167 | .cipher_list(ciphers); |
| 168 | EXPECT_OK(server_builder.Build(&server)); |
| 169 | ASSERT_OK(server->Start()); |
| 170 | |
| 171 | ThriftClient<StatestoreServiceClientWrapper> client( |
| 172 | "localhost", port, "", nullptr, false); |
| 173 | ASSERT_OK(client.Open()); |
| 174 | TRegisterSubscriberRequest req; |
| 175 | TRegisterSubscriberResponse resp; |
| 176 | bool send_done = false; |
| 177 | |
| 178 | req.subscriber_id = move(subscriber_id); |
| 179 | if (expect_throw) { |
| 180 | EXPECT_THROW( |
| 181 | client.iface()->RegisterSubscriber(resp, req, &send_done), TTransportException); |
| 182 | } else { |
| 183 | EXPECT_NO_THROW(client.iface()->RegisterSubscriber(resp, req, &send_done)); |
| 184 | } |
| 185 | } |
| 186 | |
| 187 | TEST(ThriftTestBase, MaxMessageSizeFit) { |
| 188 | TestMaxMessageSize("dummy_id", false); |
no test coverage detected