@@ -216,62 +216,6 @@ TEST_F(VersionedLayerClientPrefetchPartitionsTest, PrefetchPartitionsFails) {
216216 }
217217}
218218
219- TEST_F (VersionedLayerClientPrefetchPartitionsTest, PrefetchPartitionCancelled) {
220- auto partitions_count = 1 ;
221- auto client =
222- read::VersionedLayerClient (kCatalogHrn , kLayer , boost::none, settings_);
223- auto partitions = GeneratePartitionIds (partitions_count);
224- auto partitions_response =
225- ReadDefaultResponses::GeneratePartitionsResponse (partitions_count);
226- const auto request =
227- read::PrefetchPartitionsRequest ().WithPartitionIds (partitions);
228- {
229- SCOPED_TRACE (" Get version fails" );
230-
231- ExpectVersionRequest (
232- http::NetworkResponse ().WithStatus (HttpStatusCode::BAD_REQUEST));
233-
234- auto future = client.PrefetchPartitions (request).GetFuture ();
235- ASSERT_NE (future.wait_for (std::chrono::seconds (kTimeout )),
236- std::future_status::timeout);
237-
238- auto response = future.get ();
239- ASSERT_FALSE (response.IsSuccessful ());
240- }
241- {
242- SCOPED_TRACE (" Query partition fails" );
243- ExpectVersionRequest ();
244- ExpectQueryPartitionsRequest (
245- partitions, partitions_response,
246- http::NetworkResponse ().WithStatus (HttpStatusCode::BAD_REQUEST));
247-
248- auto future = client.PrefetchPartitions (request).GetFuture ();
249- ASSERT_NE (future.wait_for (std::chrono::seconds (kTimeout )),
250- std::future_status::timeout);
251-
252- auto response = future.get ();
253- ASSERT_FALSE (response.IsSuccessful ());
254- }
255- {
256- SCOPED_TRACE (" Get data cancelled" );
257- ExpectQueryPartitionsRequest (partitions, partitions_response);
258-
259- ExpectBlobRequest (
260- partitions_response.GetPartitions ().begin ()->GetDataHandle (), " data" ,
261- http::NetworkResponse ().WithStatus (
262- static_cast <int >(http::ErrorCode::CANCELLED_ERROR)));
263-
264- auto test_request = client.PrefetchPartitions (request);
265- auto future = test_request.GetFuture ();
266- ASSERT_NE (future.wait_for (std::chrono::seconds (kTimeout )),
267- std::future_status::timeout);
268-
269- auto response = future.get ();
270- ASSERT_FALSE (response.IsSuccessful ());
271- ASSERT_EQ (response.GetError ().GetErrorCode (), client::ErrorCode::Cancelled);
272- }
273- }
274-
275219TEST_F (VersionedLayerClientPrefetchPartitionsTest, PrefetchBatchFails) {
276220 // Should result in two metadata query
277221 constexpr auto partitions_count = 200u ;
0 commit comments