MCPcopy Create free account
hub / github.com/apache/arrow / ASSERT_OK_AND_ASSIGN

Function ASSERT_OK_AND_ASSIGN

cpp/src/arrow/flight/flight_internals_test.cc:190–275  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

188
189TEST(FlightTypes, FlightEndpoint) {
190 ASSERT_OK_AND_ASSIGN(auto location1, Location::ForGrpcTcp("localhost", 1024));
191 ASSERT_OK_AND_ASSIGN(auto location2, Location::ForGrpcTls("localhost", 1024));
192 // 2023-06-19 03:14:06.004339123
193 const auto expiration_time_duration =
194 std::chrono::seconds{1687144446} + std::chrono::nanoseconds{4339123};
195 Timestamp expiration_time(
196 std::chrono::duration_cast<Timestamp::duration>(expiration_time_duration));
197 std::vector<FlightEndpoint> values = {
198 {Ticket{""}, {}, std::nullopt, {}},
199 {Ticket{"foo"}, {}, std::nullopt, {}},
200 {Ticket{"bar"}, {}, std::nullopt, {"\xDE\xAD\xBE\xEF"}},
201 {Ticket{"foo"}, {}, expiration_time, {}},
202 {Ticket{"foo"}, {location1}, std::nullopt, {}},
203 {Ticket{"bar"}, {location1}, std::nullopt, {}},
204 {Ticket{"foo"}, {location2}, std::nullopt, {}},
205 {Ticket{"foo"}, {location1, location2}, std::nullopt, {"\xba\xdd\xca\xfe"}},
206 };
207 std::vector<std::string> reprs = {
208 "<FlightEndpoint ticket=<Ticket ticket=''> locations=[] "
209 "expiration_time=null app_metadata=''>",
210 "<FlightEndpoint ticket=<Ticket ticket='foo'> locations=[] "
211 "expiration_time=null app_metadata=''>",
212 "<FlightEndpoint ticket=<Ticket ticket='bar'> locations=[] "
213 "expiration_time=null app_metadata='DEADBEEF'>",
214 "<FlightEndpoint ticket=<Ticket ticket='foo'> locations=[] "
215 "expiration_time=2023-06-19 03:14:06.004339123 app_metadata=''>",
216 "<FlightEndpoint ticket=<Ticket ticket='foo'> locations="
217 "[grpc+tcp://localhost:1024] expiration_time=null app_metadata=''>",
218 "<FlightEndpoint ticket=<Ticket ticket='bar'> locations="
219 "[grpc+tcp://localhost:1024] expiration_time=null app_metadata=''>",
220 "<FlightEndpoint ticket=<Ticket ticket='foo'> locations="
221 "[grpc+tls://localhost:1024] expiration_time=null app_metadata=''>",
222 "<FlightEndpoint ticket=<Ticket ticket='foo'> locations="
223 "[grpc+tcp://localhost:1024, grpc+tls://localhost:1024] "
224 "expiration_time=null app_metadata='BADDCAFE'>",
225 };
226
227 ASSERT_NO_FATAL_FAILURE(TestRoundtrip<pb::FlightEndpoint>(values, reprs));
228}
229
230TEST(FlightTypes, FlightInfo) {
231 ASSERT_OK_AND_ASSIGN(auto location, Location::ForGrpcTcp("localhost", 1234));
232 Schema schema1({field("ints", int64())});
233 Schema schema2({});
234 auto desc1 = FlightDescriptor::Command("foo");
235 auto desc2 = FlightDescriptor::Command("bar");
236 auto endpoint1 = FlightEndpoint{Ticket{"foo"}, {}, std::nullopt, ""};
237 auto endpoint2 =
238 FlightEndpoint{Ticket{"foo"}, {location}, std::nullopt, "\xCA\xFE\xD0\x0D"};
239 std::vector<FlightInfo> values = {
240 MakeFlightInfo(schema1, desc1, {}, -1, -1, false, ""),
241 MakeFlightInfo(schema1, desc2, {}, -1, -1, true, ""),
242 MakeFlightInfo(schema2, desc1, {}, -1, -1, false, ""),
243 MakeFlightInfo(schema1, desc1, {endpoint1}, -1, 42, true, ""),
244 MakeFlightInfo(schema1, desc2, {endpoint1, endpoint2}, 64, -1, false,
245 "\xDE\xAD\xC0\xDE"),
246 MakeFlightInfo(desc1, {}, -1, -1, false, ""),
247 };

Callers 1

TESTFunction · 0.70

Calls 2

MakeFlightInfoFunction · 0.85
CommandFunction · 0.70

Tested by

no test coverage detected