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 +#include #include #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