diff --git a/test/core/client_channel/lb_policies_test.c b/test/core/client_channel/lb_policies_test.c index bf657f89eee..51df52c3e33 100644 --- a/test/core/client_channel/lb_policies_test.c +++ b/test/core/client_channel/lb_policies_test.c @@ -40,7 +40,6 @@ #include #include #include -#include #include "src/core/ext/client_channel/client_channel.h" #include "src/core/ext/client_channel/lb_policy_registry.h" @@ -788,12 +787,8 @@ static void verify_total_carnage_round_robin(const servers_fixture *f, * subchannel already transitioned to retrying. */ for (size_t i = 0; i < sequences->n; i++) { const grpc_connectivity_state actual = sequences->connectivity_states[i]; - uint32_t expected_bitset; - GPR_BITSET(&expected_bitset, GRPC_CHANNEL_TRANSIENT_FAILURE); - GPR_BITSET(&expected_bitset, GRPC_CHANNEL_CONNECTING); - uint32_t actual_bitset; - GPR_BITSET(&actual_bitset, actual); - if ((actual_bitset & expected_bitset) == 0) { + if (actual != GRPC_CHANNEL_TRANSIENT_FAILURE && + actual != GRPC_CHANNEL_CONNECTING) { gpr_log(GPR_ERROR, "CONNECTIVITY STATUS SEQUENCE FAILURE: expected " "GRPC_CHANNEL_TRANSIENT_FAILURE or GRPC_CHANNEL_CONNECTING, got " @@ -850,13 +845,9 @@ static void verify_partial_carnage_round_robin( * are gone. May also be CONNECTING if, under load, this check took too long * to run and the subchannel already transitioned to retrying. */ actual = sequences->connectivity_states[num_iters - 1]; - uint32_t expected_bitset; - GPR_BITSET(&expected_bitset, GRPC_CHANNEL_TRANSIENT_FAILURE); - GPR_BITSET(&expected_bitset, GRPC_CHANNEL_CONNECTING); - uint32_t actual_bitset; - GPR_BITSET(&actual_bitset, actual); for (i = 0; i < sequences->n; i++) { - if ((actual_bitset & expected_bitset) == 0) { + if (actual != GRPC_CHANNEL_TRANSIENT_FAILURE && + actual != GRPC_CHANNEL_CONNECTING) { gpr_log(GPR_ERROR, "CONNECTIVITY STATUS SEQUENCE FAILURE: expected " "GRPC_CHANNEL_TRANSIENT_FAILURE or GRPC_CHANNEL_CONNECTING, got "