From 6deec6a735753ba581d2b0974c70c222f43faf51 Mon Sep 17 00:00:00 2001 From: hai007 Date: Thu, 1 Aug 2024 17:32:01 -0700 Subject: [PATCH] Fix bwu_manager_test.cc PiperOrigin-RevId: 658595183 --- connections/implementation/bwu_manager_test.cc | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/connections/implementation/bwu_manager_test.cc b/connections/implementation/bwu_manager_test.cc index 6ef1f963..d1210123 100644 --- a/connections/implementation/bwu_manager_test.cc +++ b/connections/implementation/bwu_manager_test.cc @@ -239,6 +239,13 @@ class BwuManagerTestParam : public BwuManagerTest, }; TEST_P(BwuManagerTestParam, InitiateBwu_Success) { + auto flag = NearbyFlags::GetInstance().GetBoolFlag( + config_package_nearby::nearby_connections_feature:: + kProcessBwuFrameAfterPcpConnected); + NearbyFlags::GetInstance().OverrideBoolFlagValue( + config_package_nearby::nearby_connections_feature:: + kProcessBwuFrameAfterPcpConnected, + false); // Create the initial device-to-device Bluetooth connection. FakeEndpointChannel* initial_channel = CreateInitialEndpoint(kServiceIdA, kEndpointId1, Medium::BLUETOOTH); @@ -299,6 +306,10 @@ TEST_P(BwuManagerTestParam, InitiateBwu_Success) { EXPECT_EQ(location::nearby::proto::connections::DisconnectionReason::UPGRADED, old_channel->disconnection_reason()); UnRegisterChannelForEndpoint(kEndpointId1); + NearbyFlags::GetInstance().OverrideBoolFlagValue( + config_package_nearby::nearby_connections_feature:: + kProcessBwuFrameAfterPcpConnected, + flag); } TEST_P(BwuManagerTestParam,