@@ -149,10 +149,10 @@ TEST_P(DistributorServiceTest, ValidStartRequestNonExistingServiceYieldsResponse
149149 auto status = reader_writer->Finish ();
150150 EXPECT_FALSE (status.ok ());
151151 EXPECT_THAT (status.error_message (), HasSubstr (" One or more execution requests failed" ));
152- ASSERT_EQ (response_.fragment_size (), 1 );
153- EXPECT_TRUE (response_.fragment (0 ).has_error ());
154- EXPECT_EQ (response_.fragment (0 ).error ().code (), grpc::StatusCode::UNAVAILABLE);
155- EXPECT_THAT (response_.fragment (0 ).error ().message (),
152+ ASSERT_EQ (response_.service_response_size (), 1 );
153+ EXPECT_TRUE (response_.service_response (0 ).has_error ());
154+ EXPECT_EQ (response_.service_response (0 ).error ().code (), grpc::StatusCode::UNAVAILABLE);
155+ EXPECT_THAT (response_.service_response (0 ).error ().message (),
156156 HasSubstr (" Distributed Execution Request failed: Failed to write request to the "
157157 " Nighthawk Service gRPC channel" ));
158158}
@@ -173,7 +173,7 @@ TEST_P(DistributorServiceWithMockServiceClientTest, DistributeToTwoServicesYield
173173 ASSERT_TRUE (reader_writer->Read (&response_));
174174 auto status = reader_writer->Finish ();
175175 EXPECT_TRUE (status.ok ());
176- EXPECT_EQ (response_.fragment_size (), 2 );
176+ EXPECT_EQ (response_.service_response_size (), 2 );
177177}
178178
179179TEST_P (DistributorServiceWithMockServiceClientTest,
@@ -192,8 +192,8 @@ TEST_P(DistributorServiceWithMockServiceClientTest,
192192 auto status = reader_writer->Finish ();
193193 EXPECT_FALSE (status.ok ());
194194 EXPECT_THAT (status.error_message (), HasSubstr (" One or more execution requests failed" ));
195- ASSERT_EQ (response_.fragment_size (), 1 );
196- EXPECT_THAT (response_.fragment (0 ).error ().message (),
195+ ASSERT_EQ (response_.service_response_size (), 1 );
196+ EXPECT_THAT (response_.service_response (0 ).error ().message (),
197197 HasSubstr (" artificial nighthawk service error" ));
198198}
199199
0 commit comments