Skip to content

Commit 41f6fa8

Browse files
committed
Fix flaky reverse_interface test
1 parent 97f4f0a commit 41f6fa8

1 file changed

Lines changed: 15 additions & 4 deletions

File tree

tests/test_reverse_interface.cpp

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -536,8 +536,11 @@ TEST_F(ReverseInterfaceTest, disconnected_callbacks_are_called)
536536
// Close the client connection
537537
client_->close();
538538
EXPECT_TRUE(waitForProgramState(1000, false));
539-
std::unique_lock<std::mutex> lk(g_connection_mutex);
540-
g_connection_condition.wait_for(lk, std::chrono::seconds(1), [&]() { return !reverse_interface_->connected.load(); });
539+
{
540+
std::unique_lock<std::mutex> lk(g_connection_mutex);
541+
g_connection_condition.wait_for(lk, std::chrono::seconds(1),
542+
[&]() { return !reverse_interface_->connected.load(); });
543+
}
541544
EXPECT_TRUE(disconnect_called_1);
542545
EXPECT_TRUE(disconnect_called_2);
543546

@@ -548,7 +551,11 @@ TEST_F(ReverseInterfaceTest, disconnected_callbacks_are_called)
548551
EXPECT_TRUE(waitForProgramState(1000, true));
549552
reverse_interface_->unregisterDisconnectionCallback(disconnection_callback_id_1);
550553
client_->close();
551-
g_connection_condition.wait_for(lk, std::chrono::seconds(1), [&]() { return !reverse_interface_->connected.load(); });
554+
{
555+
std::unique_lock<std::mutex> lk(g_connection_mutex);
556+
g_connection_condition.wait_for(lk, std::chrono::seconds(1),
557+
[&]() { return !reverse_interface_->connected.load(); });
558+
}
552559
EXPECT_TRUE(waitForProgramState(1000, false));
553560
EXPECT_FALSE(disconnect_called_1);
554561
EXPECT_TRUE(disconnect_called_2);
@@ -560,7 +567,11 @@ TEST_F(ReverseInterfaceTest, disconnected_callbacks_are_called)
560567
EXPECT_TRUE(waitForProgramState(1000, true));
561568
reverse_interface_->unregisterDisconnectionCallback(disconnection_callback_id_2);
562569
client_->close();
563-
g_connection_condition.wait_for(lk, std::chrono::seconds(1), [&]() { return !reverse_interface_->connected.load(); });
570+
{
571+
std::unique_lock<std::mutex> lk(g_connection_mutex);
572+
g_connection_condition.wait_for(lk, std::chrono::seconds(1),
573+
[&]() { return !reverse_interface_->connected.load(); });
574+
}
564575
EXPECT_TRUE(waitForProgramState(1000, false));
565576
EXPECT_FALSE(disconnect_called_1);
566577
EXPECT_FALSE(disconnect_called_2);

0 commit comments

Comments
 (0)