Merge branch 'feature/unittest--transport-tests' of https://github.com/feasel0/connectedhomeip into feature/unittest--transport-tests

Author: feasel0

src/inet/TCPEndPoint.h

@@ -38,10 +38,6 @@
 
 namespace chip {
 
-namespace Transport {
-class TCPTest;
-};
-
 namespace Inet {
 
 class TCPTest;
@@ -529,7 +525,6 @@ class DLL_EXPORT TCPEndPoint : public EndPointBasis<TCPEndPoint>
     constexpr static size_t kMaxReceiveMessageSize = System::PacketBuffer::kMaxSizeWithoutReserve;
 
 protected:
-    friend class ::chip::Transport::TCPTest;
     friend class TCPTest;
 
     TCPEndPoint(EndPointManager<TCPEndPoint> & endPointManager) :

src/lib/dnssd/minimal_mdns/tests/TestResponseSender.cpp

@@ -286,34 +286,34 @@ TEST_F(TestResponseSender, AddManyQueryResponders)
 
 TEST_F(TestResponseSender, PtrSrvTxtMultipleRespondersToInstance)
 {
-    CommonTestElements common1("test1");
-    CommonTestElements common2("test2");
+    auto common1 = std::make_unique<CommonTestElements>("test1");
+    auto common2 = std::make_unique<CommonTestElements>("test2");
 
     // Just use the server from common1.
-    ResponseSender responseSender(&common1.server);
+    ResponseSender responseSender(&common1->server);
 
-    EXPECT_EQ(responseSender.AddQueryResponder(&common1.queryResponder), CHIP_NO_ERROR);
-    common1.queryResponder.AddResponder(&common1.ptrResponder).SetReportInServiceListing(true);
-    common1.queryResponder.AddResponder(&common1.srvResponder);
-    common1.queryResponder.AddResponder(&common1.txtResponder);
+    EXPECT_EQ(responseSender.AddQueryResponder(&common1->queryResponder), CHIP_NO_ERROR);
+    common1->queryResponder.AddResponder(&common1->ptrResponder).SetReportInServiceListing(true);
+    common1->queryResponder.AddResponder(&common1->srvResponder);
+    common1->queryResponder.AddResponder(&common1->txtResponder);
 
-    EXPECT_EQ(responseSender.AddQueryResponder(&common2.queryResponder), CHIP_NO_ERROR);
-    common2.queryResponder.AddResponder(&common2.ptrResponder).SetReportInServiceListing(true);
-    common2.queryResponder.AddResponder(&common2.srvResponder);
-    common2.queryResponder.AddResponder(&common2.txtResponder);
+    EXPECT_EQ(responseSender.AddQueryResponder(&common2->queryResponder), CHIP_NO_ERROR);
+    common2->queryResponder.AddResponder(&common2->ptrResponder).SetReportInServiceListing(true);
+    common2->queryResponder.AddResponder(&common2->srvResponder);
+    common2->queryResponder.AddResponder(&common2->txtResponder);
 
     // Build a query for the second instance.
-    common2.recordWriter.WriteQName(common2.instance);
-    QueryData queryData = QueryData(QType::ANY, QClass::IN, false, common2.requestNameStart, common2.requestBytesRange);
+    common2->recordWriter.WriteQName(common2->instance);
+    QueryData queryData = QueryData(QType::ANY, QClass::IN, false, common2->requestNameStart, common2->requestBytesRange);
 
     // Should get back answers from second instance only.
-    common1.server.AddExpectedRecord(&common2.srvRecord);
-    common1.server.AddExpectedRecord(&common2.txtRecord);
+    common1->server.AddExpectedRecord(&common2->srvRecord);
+    common1->server.AddExpectedRecord(&common2->txtRecord);
 
-    responseSender.Respond(1, queryData, &common1.packetInfo, ResponseConfiguration());
+    responseSender.Respond(1, queryData, &common1->packetInfo, ResponseConfiguration());
 
-    EXPECT_TRUE(common1.server.GetSendCalled());
-    EXPECT_TRUE(common1.server.GetHeaderFound());
+    EXPECT_TRUE(common1->server.GetSendCalled());
+    EXPECT_TRUE(common1->server.GetHeaderFound());
 }
 
 TEST_F(TestResponseSender, PtrSrvTxtMultipleRespondersToServiceListing)

src/transport/raw/TCP.h

@@ -41,6 +41,10 @@
 namespace chip {
 namespace Transport {
 
+// Forward declaration of friend class for test access.
+template <size_t kActiveConnectionsSize, size_t kPendingPacketSize>
+class TCPBaseTestAccess;
+
 /** Defines listening parameters for setting up a TCP transport */
 class TcpListenParameters
 {
@@ -200,7 +204,9 @@ class DLL_EXPORT TCPBase : public Base
     void CloseActiveConnections();
 
 private:
-    friend class TCPTest;
+    // Allow tests to access private members.
+    template <size_t kActiveConnectionsSize, size_t kPendingPacketSize>
+    friend class TCPBaseTestAccess;
 
     /**
      * Allocate an unused connection from the pool
@@ -330,7 +336,6 @@ class TCP : public TCPBase
     ~TCP() override { mPendingPackets.ReleaseAll(); }
 
 private:
-    friend class TCPTest;
     ActiveTCPConnectionState mConnectionsBuffer[kActiveConnectionsSize];
     PoolImpl<PendingPacket, kPendingPacketSize, ObjectPoolMem::kInline, PendingPacketPoolType::Interface> mPendingPackets;
 };

src/transport/raw/tests/BUILD.gn

@@ -49,6 +49,8 @@ chip_test_suite("tests") {
     test_sources += [ "TestTCP.cpp" ]
   }
 
+  sources = [ "TCPBaseTestAccess.h" ]
+
   public_deps = [
     ":helpers",
     "${chip_root}/src/inet/tests:helpers",

src/transport/raw/tests/TCPBaseTestAccess.h

@@ -0,0 +1,49 @@
+/*
+ *
+ *    Copyright (c) 2024 Project CHIP Authors
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+
+#pragma once
+
+#if INET_CONFIG_ENABLE_TCP_ENDPOINT
+#include <transport/raw/TCP.h>
+#endif // INET_CONFIG_ENABLE_TCP_ENDPOINT
+
+namespace chip {
+namespace Transport {
+/**
+ * @brief Class acts as an accessor to private members of the TCPBase class without needing to give
+ *        friend access to each individual test.
+ */
+template <size_t kActiveConnectionsSize, size_t kPendingPacketSize>
+class TCPBaseTestAccess
+{
+public:
+    using TCPImpl = Transport::TCP<kActiveConnectionsSize, kPendingPacketSize>;
+
+    static void * FindActiveConnection(TCPImpl & tcp, Transport::PeerAddress & peerAddress)
+    {
+        return tcp.FindActiveConnection(peerAddress);
+    }
+    static Inet::TCPEndPoint * GetEndpoint(void * state) { return static_cast<ActiveTCPConnectionState *>(state)->mEndPoint; }
+
+    static CHIP_ERROR ProcessReceivedBuffer(TCPImpl & tcp, Inet::TCPEndPoint * endPoint, const PeerAddress & peerAddress,
+                                            System::PacketBufferHandle && buffer)
+    {
+        return tcp.ProcessReceivedBuffer(endPoint, peerAddress, std::move(buffer));
+    }
+};
+} // namespace Transport
+} // namespace chip

src/transport/raw/tests/TestMessageHeader.cpp

@@ -23,14 +23,14 @@
  *
  */
 
+#include <gtest/gtest.h>
+
 #include <lib/core/ErrorStr.h>
 #include <lib/support/CHIPMem.h>
 #include <lib/support/CodeUtils.h>
 #include <protocols/Protocols.h>
 #include <transport/raw/MessageHeader.h>
 
-#include <gtest/gtest.h>
-
 namespace {
 
 using namespace chip;

src/transport/raw/tests/TestPeerAddress.cpp

@@ -21,13 +21,13 @@
 #include <stdint.h>
 #include <string.h>
 
+#include <gtest/gtest.h>
+
 #include <inet/IPAddress.h>
 #include <lib/core/DataModelTypes.h>
 #include <lib/core/PeerId.h>
 #include <transport/raw/PeerAddress.h>
 
-#include <gtest/gtest.h>
-
 namespace {
 
 using namespace chip;