From 4157b1f70b1e9d8523a23be6967dc20d9bb2ddfa Mon Sep 17 00:00:00 2001
From: Hai Shang <hais@google.com>
Date: Mon, 17 Oct 2022 16:52:08 -0700
Subject: [PATCH] Add unit test for PayloadListner

PiperOrigin-RevId: 481773299
---
 connections/listeners_test.cc | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/connections/listeners_test.cc b/connections/listeners_test.cc
index 2595899a..82c34941 100644
--- a/connections/listeners_test.cc
+++ b/connections/listeners_test.cc
@@ -15,11 +15,13 @@
 #include "connections/listeners.h"
 
 #include <memory>
+#include <string>
 #include <utility>
 
 #include "gmock/gmock.h"
 #include "protobuf-matchers/protocol-buffer-matchers.h"
 #include "gtest/gtest.h"
+#include "internal/platform/byte_array.h"
 
 namespace location {
 namespace nearby {
@@ -54,6 +56,24 @@ TEST(ListenersTest, EnsurePartiallyInitializedIsCallable) {
   EXPECT_TRUE(initiated_cb_called);
 }
 
+TEST(ListenersTest, PayloadListener_PayloadCb_Works) {
+  bool payload_content_match = false;
+  const std::string input_bytes = "input";
+  Payload payload = Payload(ByteArray(input_bytes));
+
+  PayloadListener listener{
+      .payload_cb =
+          [&](const std::string& endpoint_id, Payload payload) {
+            if (payload.AsBytes().data() == input_bytes) {
+              payload_content_match = true;
+            }
+          },
+  };
+  listener.payload_cb("endpoint_id", std::move(payload));
+
+  EXPECT_TRUE(payload_content_match);
+}
+
 }  // namespace
 }  // namespace connections
 }  // namespace nearby