|
|
|
@ -244,17 +244,15 @@ class End2endTest : public ::testing::TestWithParam<TestScenario> { |
|
|
|
|
BuildAndStartServer(processor); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void RestartServer(const std::shared_ptr<AuthMetadataProcessor>& processor, |
|
|
|
|
bool surface_user_agent = false) { |
|
|
|
|
void RestartServer(const std::shared_ptr<AuthMetadataProcessor>& processor) { |
|
|
|
|
if (is_server_started_) { |
|
|
|
|
server_->Shutdown(); |
|
|
|
|
BuildAndStartServer(processor, surface_user_agent); |
|
|
|
|
BuildAndStartServer(processor); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void BuildAndStartServer( |
|
|
|
|
const std::shared_ptr<AuthMetadataProcessor>& processor, |
|
|
|
|
bool surface_user_agent = false) { |
|
|
|
|
const std::shared_ptr<AuthMetadataProcessor>& processor) { |
|
|
|
|
ServerBuilder builder; |
|
|
|
|
ConfigureServerBuilder(&builder); |
|
|
|
|
auto server_creds = GetCredentialsProvider()->GetServerCredentials( |
|
|
|
@ -270,9 +268,6 @@ class End2endTest : public ::testing::TestWithParam<TestScenario> { |
|
|
|
|
builder.SetSyncServerOption(ServerBuilder::SyncServerOption::NUM_CQS, 4); |
|
|
|
|
builder.SetSyncServerOption( |
|
|
|
|
ServerBuilder::SyncServerOption::CQ_TIMEOUT_MSEC, 10); |
|
|
|
|
if (surface_user_agent) { |
|
|
|
|
builder.AddChannelArgument(GRPC_ARG_SURFACE_USER_AGENT, 1); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
server_ = builder.BuildAndStart(); |
|
|
|
|
is_server_started_ = true; |
|
|
|
@ -669,8 +664,6 @@ TEST_P(End2endTest, SimpleRpcWithCustomUserAgentPrefix) { |
|
|
|
|
} |
|
|
|
|
user_agent_prefix_ = "custom_prefix"; |
|
|
|
|
ResetStub(); |
|
|
|
|
RestartServer(std::shared_ptr<AuthMetadataProcessor>(), true); |
|
|
|
|
ResetChannel(); |
|
|
|
|
EchoRequest request; |
|
|
|
|
EchoResponse response; |
|
|
|
|
request.set_message("Hello hello hello hello"); |
|
|
|
|