From fe1e14a02eccfe13f3530a62b13cc1b972b0265e Mon Sep 17 00:00:00 2001
From: Hai Shang <hais@google.com>
Date: Mon, 5 Jun 2023 16:16:42 -0700
Subject: [PATCH] internal clean up.

PiperOrigin-RevId: 538009138
---
 presence/implementation/scan_manager_test.cc | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/presence/implementation/scan_manager_test.cc b/presence/implementation/scan_manager_test.cc
index 73172b54..604015b1 100644
--- a/presence/implementation/scan_manager_test.cc
+++ b/presence/implementation/scan_manager_test.cc
@@ -299,15 +299,22 @@ TEST_F(ScanManagerTest, StopOneSessionFromAnotherDeadlock) {
 TEST_F(ScanManagerTest, NoDeviceFoundAfterStopScan) {
   Mediums mediums;
   ScanManager manager(mediums, credential_manager_, executor_);
+  CountDownLatch start_scan_latch{1};
   nearby::BluetoothAdapter server_adapter;
   Ble ble2(server_adapter);
   std::atomic_bool stopped = false;
   ScanSessionId scan_session = manager.StartScan(
       MakeDefaultScanRequest(),
-      ScanCallback{.start_scan_cb = [](absl::Status status) {},
+      ScanCallback{.start_scan_cb =
+                       [&start_scan_latch](absl::Status status) {
+                         if (status.ok()) {
+                           start_scan_latch.CountDown();
+                         }
+                       },
                    .on_discovered_cb =
                        [&](PresenceDevice pd) { EXPECT_FALSE(stopped); }});
 
+  start_scan_latch.Await();
   manager.StopScan(scan_session);
   stopped = true;
   std::unique_ptr<AdvertisingSession> advertising_session =