nearby/cpp/platform_v2/public/wifi_lan_test.cc

#include "platform_v2/public/wifi_lan.h"

#include <memory>

#include "platform_v2/base/medium_environment.h"
#include "platform_v2/public/count_down_latch.h"
#include "platform_v2/public/logging.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"

namespace location {
namespace nearby {
namespace {

constexpr absl::string_view kServiceID{"com.google.location.nearby.apps.test"};

class WifiLanMediumTest : public ::testing::Test {
 protected:
  using DiscoveredServiceCallback = WifiLanMedium::DiscoveredServiceCallback;

  WifiLanMediumTest() { env_.Stop(); }

  MediumEnvironment& env_{MediumEnvironment::Instance()};
};

TEST_F(WifiLanMediumTest, ConstructorDestructorWorks) {
  env_.Start();
  WifiLanMedium medium_a;
  WifiLanMedium medium_b;

  // Make sure we can create functional mediums.
  ASSERT_TRUE(medium_a.IsValid());
  ASSERT_TRUE(medium_b.IsValid());

  // Make sure we can create 2 distinct mediums.
  EXPECT_NE(&medium_a.GetImpl(), &medium_b.GetImpl());
  env_.Stop();
}

TEST_F(WifiLanMediumTest, CanStartDiscoveryAndServiceIndeedDiscovered) {
  env_.Start();
  WifiLanMedium medium;
  CountDownLatch found_latch(1);
  CountDownLatch lost_latch(1);

  medium.StartDiscovery(std::string(kServiceID),
                        DiscoveredServiceCallback{
                            .service_discovered_cb =
                                [&found_latch](WifiLanService& service,
                                               const std::string& service_id) {
                                  NEARBY_LOG(INFO, "Service discovered: %s",
                                             service.GetName().c_str());
                                  EXPECT_EQ(kServiceID, service_id);
                                  found_latch.CountDown();
                                },
                            .service_lost_cb =
                                [&lost_latch](WifiLanService& service,
                                              const std::string& service_id) {
                                  NEARBY_LOG(INFO, "Service lost: %s",
                                             service.GetName().c_str());
                                  EXPECT_EQ(kServiceID, service_id);
                                  lost_latch.CountDown();
                                },
                        });
  EXPECT_TRUE(found_latch.Await(absl::Milliseconds(1000)).result());
  env_.Stop();
}

TEST_F(WifiLanMediumTest, CanStopDiscovery) {
  env_.Start();
  WifiLanMedium medium;
  CountDownLatch found_latch(1);
  CountDownLatch lost_latch(1);

  medium.StartDiscovery(std::string(kServiceID),
                        DiscoveredServiceCallback{
                            .service_discovered_cb =
                                [&found_latch](WifiLanService& service,
                                               const std::string& service_id) {
                                  NEARBY_LOG(INFO, "Service discovered: %s",
                                             service.GetName().c_str());
                                  EXPECT_EQ(kServiceID, service_id);
                                  found_latch.CountDown();
                                },
                            .service_lost_cb =
                                [&lost_latch](WifiLanService& service,
                                              const std::string& service_id) {
                                  NEARBY_LOG(INFO, "Service lost: %s",
                                             service.GetName().c_str());
                                  EXPECT_EQ(kServiceID, service_id);
                                  lost_latch.CountDown();
                                },
                        });
  EXPECT_TRUE(found_latch.Await(absl::Milliseconds(1000)).result());
  bool stop = medium.StopDiscovery(std::string(kServiceID));
  EXPECT_TRUE(stop);
  env_.Stop();
}

}  // namespace
}  // namespace nearby
}  // namespace location