Index: webrtc/p2p/client/basicportallocator_unittest.cc |
diff --git a/webrtc/p2p/client/basicportallocator_unittest.cc b/webrtc/p2p/client/basicportallocator_unittest.cc |
index 93951ff80bbbc31183cb9ccf4f38a8a377f28f06..c9831f525b9b14bd1964b42713256da778b71433 100644 |
--- a/webrtc/p2p/client/basicportallocator_unittest.cc |
+++ b/webrtc/p2p/client/basicportallocator_unittest.cc |
@@ -1608,6 +1608,7 @@ TEST_F(BasicPortAllocatorTest, TestStopGettingPorts) { |
// After stopping getting ports, adding a new interface will not start |
// getting ports again. |
+ allocator_->set_step_delay(kMinimumStepDelay); |
candidates_.clear(); |
ports_.clear(); |
candidate_allocation_done_ = false; |
@@ -1625,17 +1626,18 @@ TEST_F(BasicPortAllocatorTest, TestClearGettingPorts) { |
ASSERT_EQ_WAIT(2U, candidates_.size(), 1000); |
EXPECT_EQ(2U, ports_.size()); |
session_->ClearGettingPorts(); |
- WAIT(candidate_allocation_done_, 1000); |
- EXPECT_FALSE(candidate_allocation_done_); |
+ EXPECT_TRUE_WAIT(candidate_allocation_done_, 1000); |
// After clearing getting ports, adding a new interface will start getting |
// ports again. |
+ allocator_->set_step_delay(kMinimumStepDelay); |
candidates_.clear(); |
ports_.clear(); |
candidate_allocation_done_ = false; |
network_manager_.AddInterface(kClientAddr2); |
ASSERT_EQ_WAIT(2U, candidates_.size(), 1000); |
EXPECT_EQ(2U, ports_.size()); |
+ EXPECT_TRUE_WAIT(candidate_allocation_done_, kDefaultAllocationTimeout); |
} |
// Test that the ports and candidates are updated with new ufrag/pwd/etc. when |