mirror of https://github.com/grpc/grpc.git
Mark D. Roth
7 years ago
parent
31bdbbeacf
commit
9db86fcc63
11 changed files with 465 additions and 170 deletions
@ -0,0 +1,142 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2018 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include "src/core/ext/filters/client_channel/retry_throttle.h" |
||||
|
||||
#include <gtest/gtest.h> |
||||
|
||||
#include "test/core/util/test_config.h" |
||||
|
||||
namespace grpc_core { |
||||
namespace internal { |
||||
namespace { |
||||
|
||||
TEST(ServerRetryThrottleData, Basic) { |
||||
// Max token count is 4, so threshold for retrying is 2.
|
||||
// Token count starts at 4.
|
||||
// Each failure decrements by 1. Each success increments by 1.6.
|
||||
auto throttle_data = |
||||
MakeRefCounted<ServerRetryThrottleData>(4000, 1600, nullptr); |
||||
// Failure: token_count=3. Above threshold.
|
||||
EXPECT_TRUE(throttle_data->RecordFailure()); |
||||
// Success: token_count=4. Not incremented beyond max.
|
||||
throttle_data->RecordSuccess(); |
||||
// Failure: token_count=3. Above threshold.
|
||||
EXPECT_TRUE(throttle_data->RecordFailure()); |
||||
// Failure: token_count=2. At threshold, so no retries.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Failure: token_count=1. Below threshold, so no retries.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Failure: token_count=0. Below threshold, so no retries.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Failure: token_count=0. Below threshold, so no retries. Not
|
||||
// decremented below min.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Success: token_count=1.6.
|
||||
throttle_data->RecordSuccess(); |
||||
// Success: token_count=3.2.
|
||||
throttle_data->RecordSuccess(); |
||||
// Failure: token_count=2.2. Above threshold.
|
||||
EXPECT_TRUE(throttle_data->RecordFailure()); |
||||
// Failure: token_count=1.2. Below threshold, so no retries.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Success: token_count=2.8.
|
||||
throttle_data->RecordSuccess(); |
||||
// Failure: token_count=1.8. Below threshold, so no retries.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Success: token_count=3.4.
|
||||
throttle_data->RecordSuccess(); |
||||
// Failure: token_count=2.4. Above threshold.
|
||||
EXPECT_TRUE(throttle_data->RecordFailure()); |
||||
} |
||||
|
||||
TEST(ServerRetryThrottleData, Replacement) { |
||||
// Create old throttle data.
|
||||
// Max token count is 4, so threshold for retrying is 2.
|
||||
// Token count starts at 4.
|
||||
// Each failure decrements by 1. Each success increments by 1.
|
||||
auto old_throttle_data = |
||||
MakeRefCounted<ServerRetryThrottleData>(4000, 1000, nullptr); |
||||
// Failure: token_count=3. Above threshold.
|
||||
EXPECT_TRUE(old_throttle_data->RecordFailure()); |
||||
// Create new throttle data.
|
||||
// Max token count is 10, so threshold for retrying is 5.
|
||||
// Token count starts at 7.5 (ratio inherited from old_throttle_data).
|
||||
// Each failure decrements by 1. Each success increments by 3.
|
||||
auto throttle_data = MakeRefCounted<ServerRetryThrottleData>( |
||||
10000, 3000, old_throttle_data.get()); |
||||
// Failure via old_throttle_data: token_count=6.5.
|
||||
EXPECT_TRUE(old_throttle_data->RecordFailure()); |
||||
// Failure: token_count=5.5.
|
||||
EXPECT_TRUE(old_throttle_data->RecordFailure()); |
||||
// Failure via old_throttle_data: token_count=4.5. Below threshold.
|
||||
EXPECT_FALSE(old_throttle_data->RecordFailure()); |
||||
// Failure: token_count=3.5. Below threshold.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Success: token_count=6.5.
|
||||
throttle_data->RecordSuccess(); |
||||
// Failure via old_throttle_data: token_count=5.5. Above threshold.
|
||||
EXPECT_TRUE(old_throttle_data->RecordFailure()); |
||||
// Failure: token_count=4.5. Below threshold.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
} |
||||
|
||||
TEST(ServerRetryThrottleMap, Replacement) { |
||||
ServerRetryThrottleMap::Init(); |
||||
const char kServerName[] = "server_name"; |
||||
// Create old throttle data.
|
||||
// Max token count is 4, so threshold for retrying is 2.
|
||||
// Token count starts at 4.
|
||||
// Each failure decrements by 1. Each success increments by 1.
|
||||
auto old_throttle_data = |
||||
ServerRetryThrottleMap::GetDataForServer(kServerName, 4000, 1000); |
||||
// Failure: token_count=3. Above threshold.
|
||||
EXPECT_TRUE(old_throttle_data->RecordFailure()); |
||||
// Create new throttle data.
|
||||
// Max token count is 10, so threshold for retrying is 5.
|
||||
// Token count starts at 7.5 (ratio inherited from old_throttle_data).
|
||||
// Each failure decrements by 1. Each success increments by 3.
|
||||
auto throttle_data = |
||||
ServerRetryThrottleMap::GetDataForServer(kServerName, 10000, 3000); |
||||
// Failure via old_throttle_data: token_count=6.5.
|
||||
EXPECT_TRUE(old_throttle_data->RecordFailure()); |
||||
// Failure: token_count=5.5.
|
||||
EXPECT_TRUE(old_throttle_data->RecordFailure()); |
||||
// Failure via old_throttle_data: token_count=4.5. Below threshold.
|
||||
EXPECT_FALSE(old_throttle_data->RecordFailure()); |
||||
// Failure: token_count=3.5. Below threshold.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Success: token_count=6.5.
|
||||
throttle_data->RecordSuccess(); |
||||
// Failure via old_throttle_data: token_count=5.5. Above threshold.
|
||||
EXPECT_TRUE(old_throttle_data->RecordFailure()); |
||||
// Failure: token_count=4.5. Below threshold.
|
||||
EXPECT_FALSE(throttle_data->RecordFailure()); |
||||
// Clean up.
|
||||
ServerRetryThrottleMap::Shutdown(); |
||||
} |
||||
|
||||
} // namespace
|
||||
} // namespace internal
|
||||
} // namespace grpc_core
|
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc_test_init(argc, argv); |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
return RUN_ALL_TESTS(); |
||||
} |
||||
Loading…
Reference in new issue