Implement simple mocking framework and test ntcr::StreamSocket shutdown sequence during connection phase using whitebox testing approach

groups/ntc/ntccfg/ntccfg_test.t.cpp

@@ -31,19 +31,238 @@ using namespace BloombergLP;
 // [ 1]
 //-----------------------------------------------------------------------------
 
+namespace mock_test {
+
+class Interface
+{
+    virtual void f()  = 0;
+    virtual int  f1() = 0;
+
+    virtual void f2(int)  = 0;
+    virtual void f3(int*) = 0;
+    virtual void f4(int&) = 0;
+
+    virtual void f5(int, char)    = 0;
+    virtual void f5(int*, double) = 0;
+
+    virtual const bsl::shared_ptr<int>& f6(int*, double&, long) = 0;
+};
+
+NTF_MOCK_CLASS(MyMock, Interface)
+
+NTF_MOCK_METHOD(void, f)
+NTF_MOCK_METHOD(int, f1)
+
+NTF_MOCK_METHOD(void, f2, int)
+NTF_MOCK_METHOD(void, f3, int*)
+NTF_MOCK_METHOD(void, f4, int&)
+
+NTF_MOCK_METHOD(void, f5, int, char)
+NTF_MOCK_METHOD(void, f5, int*, double)
+
+NTF_MOCK_METHOD(const bsl::shared_ptr<int>&, f6, int*, double&, long)
+
+NTF_MOCK_CLASS_END;
+
+}
+
 NTCCFG_TEST_CASE(1)
 {
-    // Concern:
-    // Plan:
+    using namespace mock_test;
+
+    MyMock mock;
+    NTF_EXPECT_0(mock, f).ONCE();
+    mock.f();
+
+    NTF_EXPECT_0(mock, f1).ONCE().RETURN(22);
+    NTF_EXPECT_0(mock, f1).ONCE().RETURN(33);
+
+    NTCCFG_TEST_EQ(mock.f1(), 22);
+    NTCCFG_TEST_EQ(mock.f1(), 33);
+}
+
+NTCCFG_TEST_CASE(2)
+{
+    using namespace mock_test;
+
+    MyMock mock;
+
+    {
+        // it means we do not case what argument is used when f2 is called
+        NTF_EXPECT_1(mock, f2, IGNORE_ARG).ONCE();
+
+        const int val = 22;
+        mock.f2(val);
 
-    ntccfg::TestAllocator ta;
+        // here we expect that the argument used to call f2 equals `expected`
+        const int expected = 22;
+        NTF_EXPECT_1(mock, f2, NTF_EQ(expected)).ONCE();
+        mock.f2(val);
+    }
     {
+        int  value        = 44;
+        int* ptr          = &value;
+        int* expected_ptr = ptr;
+
+        // expect that argument used to call f3 equals `expected_ptr`
+        NTF_EXPECT_1(mock, f3, NTF_EQ(expected_ptr)).ONCE();
+        mock.f3(ptr);
+
+        //expect that when argument used to call f3 is dereferenced it equals
+        //`expected value`
+        int expected_value = value;
+        NTF_EXPECT_1(mock, f3, NTF_EQ_DEREF(expected_value)).ONCE();
+        mock.f3(ptr);
+
+        int& ref = value;
+        NTF_EXPECT_1(mock, f4, NTF_EQ(value)).ONCE();
+        mock.f4(ref);
+    }
+}
+
+NTCCFG_TEST_CASE(3)
+{
+    using namespace mock_test;
+
+    MyMock mock;
+
+    {
+        const int newValue = 55;
+        // when f3 is called, we do not case what arg value is, but we want to
+        // dereference it and set it value to `newValue`
+        NTF_EXPECT_1(mock, f3, IGNORE_ARG)
+            .ONCE()
+            .SET_ARG_1(FROM_DEREF(newValue));
+
+        int val = 0;
+        mock.f3(&val);
+        NTCCFG_TEST_EQ(val, newValue);
+
+        // the same can be done with references
+        NTF_EXPECT_1(mock, f4, IGNORE_ARG).ONCE().SET_ARG_1(FROM(newValue));
+
+        int  data     = 12;
+        int& data_ref = data;
+        mock.f4(data_ref);
+        NTCCFG_TEST_EQ(data, newValue);
+    }
+}
+
+NTCCFG_TEST_CASE(4)
+{
+    using namespace mock_test;
+
+    MyMock mock;
+
+    {
+        // an argument can be saved to external variable to later used
+        int storage = 0;
+        NTF_EXPECT_1(mock, f2, IGNORE_ARG).ONCE().SAVE_ARG_1(TO(&storage));
+
+        int val = 22;
+        mock.f2(val);
+
+        NTCCFG_TEST_EQ(storage, val);
+    }
+    {
+        //the same can be done with raw pointers
+        int* ptr = 0;
+        NTF_EXPECT_1(mock, f3, IGNORE_ARG).ONCE().SAVE_ARG_1(TO(&ptr));
+
+        int val = 6;
+        mock.f3(&val);
+        NTCCFG_TEST_EQ(ptr, &val);
+
+        //pointer argument can be dereferenced before saving
+        int storage = 0;
+        NTF_EXPECT_1(mock, f3, IGNORE_ARG)
+            .ONCE()
+            .SAVE_ARG_1(TO_DEREF(&storage));
+
+        mock.f3(&val);
+        NTCCFG_TEST_EQ(storage, val);
+    }
+    {
+        //the same can be done with references
+        int storage = 0;
+        NTF_EXPECT_1(mock, f4, IGNORE_ARG).ONCE().SAVE_ARG_1(TO(&storage));
+
+        int val = 7;
+        mock.f4(val);
+        NTCCFG_TEST_EQ(storage, val);
+    }
+}
+
+NTCCFG_TEST_CASE(5)
+{
+    using namespace mock_test;
+
+    MyMock mock;
+
+    {
+        //for overloaded methods we need to specify type of an argument using
+        // `_SPEC` addition to NTF_EQ (or IGNORE_ARG_S) macro
+
+        char c = 'a';
+        NTF_EXPECT_2(mock, f5, IGNORE_ARG_S(int), NTF_EQ_SPEC(c, char)).ONCE();
+
+        mock.f5(22, c);
+
+        int    val = 14;
+        double d   = 3.14;
+        NTF_EXPECT_2(mock,
+                     f5,
+                     NTF_EQ_DEREF_SPEC(val, int*),
+                     NTF_EQ_SPEC(d, double))
+            .ONCE();
+
+        mock.f5(&val, d);
+    }
+}
+
+NTCCFG_TEST_CASE(6)
+{
+    using namespace mock_test;
+
+    MyMock mock;
+    {
+        //see how references can be returned and multiple arguments
+        //expectations can be set
+
+        bsl::shared_ptr<int>        sptr(new int(14));
+        const bsl::shared_ptr<int>& sptrRef = sptr;
+
+        int    expectedInt    = 22;
+        double expectedDouble = 7.7;
+        long   expectedLong   = 100;
+        int*   ptr            = 0;
+        double newDouble      = 8.8;
+        NTF_EXPECT_3(mock,
+                     f6,
+                     NTF_EQ_DEREF(expectedInt),
+                     NTF_EQ(expectedDouble),
+                     NTF_EQ(expectedLong))
+            .ONCE()
+            .SAVE_ARG_1(TO(&ptr))
+            .SET_ARG_2(FROM(newDouble))
+            .RETURNREF(sptrRef);
+
+        const bsl::shared_ptr<int>& res =
+            mock.f6(&expectedInt, expectedDouble, expectedLong);
+        NTCCFG_TEST_EQ(ptr, &expectedInt);
+        NTCCFG_TEST_EQ(expectedDouble, newDouble);
+        NTCCFG_TEST_EQ(res, sptrRef);
+        NTCCFG_TEST_EQ(&res, &sptrRef);
     }
-    NTCCFG_TEST_ASSERT(ta.numBlocksInUse() == 0);
 }
 
 NTCCFG_TEST_DRIVER
 {
     NTCCFG_TEST_REGISTER(1);
+    NTCCFG_TEST_REGISTER(2);
+    NTCCFG_TEST_REGISTER(3);
+    NTCCFG_TEST_REGISTER(4);
+    NTCCFG_TEST_REGISTER(5);
+    NTCCFG_TEST_REGISTER(6);
 }
 NTCCFG_TEST_DRIVER_END;

groups/ntc/ntcd/ntcd_blobbufferfactory.cpp

@@ -0,0 +1,19 @@
+// Copyright 2024 Bloomberg Finance L.P.
+// SPDX-License-Identifier: Apache-2.0
+//
+// 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.
+
+#include <ntcd_blobbufferfactory.h>
+
+#include <bsls_ident.h>
+BSLS_IDENT_RCSID(ntcd_blobbufferfactory_cpp, "$Id$ $CSID$")

groups/ntc/ntcd/ntcd_blobbufferfactory.h

@@ -0,0 +1,34 @@
+// Copyright 2024 Bloomberg Finance L.P.
+// SPDX-License-Identifier: Apache-2.0
+//
+// 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.
+#ifndef INCLUDED_NTCD_BLOBBUFFERFACTORY
+#define INCLUDED_NTCD_BLOBBUFFERFACTORY
+
+#include <bsls_ident.h>
+BSLS_IDENT("$Id: $")
+
+#include <bdlbb_blob.h>
+#include <ntccfg_test.h>
+
+namespace BloombergLP {
+namespace ntcd {
+
+NTF_MOCK_CLASS(BufferFactoryMock, bdlbb::BlobBufferFactory)
+NTF_MOCK_METHOD(void, allocate, bdlbb::BlobBuffer*)
+NTF_MOCK_CLASS_END;
+
+}  // close package namespace
+}  // close enterprise namespace
+
+#endif

groups/ntc/ntcd/ntcd_blobbufferfactory.t.cpp

@@ -0,0 +1,28 @@
+// Copyright 2024 Bloomberg Finance L.P.
+// SPDX-License-Identifier: Apache-2.0
+//
+// 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.
+
+#include <ntcd_blobbufferfactory.h>
+
+#include <ntccfg_test.h>
+
+NTCCFG_TEST_CASE(1)
+{
+}
+
+NTCCFG_TEST_DRIVER
+{
+    NTCCFG_TEST_REGISTER(1);
+}
+NTCCFG_TEST_DRIVER_END;

groups/ntc/ntcd/ntcd_datapool.cpp

@@ -0,0 +1,19 @@
+// Copyright 2024 Bloomberg Finance L.P.
+// SPDX-License-Identifier: Apache-2.0
+//
+// 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.
+
+#include <ntcd_datapool.h>
+
+#include <bsls_ident.h>
+BSLS_IDENT_RCSID(ntcd_datapool_cpp, "$Id$ $CSID$")

groups/ntc/ntcd/ntcd_datapool.h

@@ -0,0 +1,43 @@
+// Copyright 2024 Bloomberg Finance L.P.
+// SPDX-License-Identifier: Apache-2.0
+//
+// 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.
+#ifndef INCLUDED_NTCD_DATAPOOL
+#define INCLUDED_NTCD_DATAPOOL
+
+#include <bsls_ident.h>
+BSLS_IDENT("$Id: $")
+
+#include <ntci_datapool.h>
+#include <ntccfg_test.h>
+
+namespace BloombergLP {
+namespace ntcd {
+
+NTF_MOCK_CLASS(DataPoolMock, ntci::DataPool)
+NTF_MOCK_METHOD(bsl::shared_ptr<ntsa::Data>, createIncomingData)
+NTF_MOCK_METHOD(bsl::shared_ptr<ntsa::Data>, createOutgoingData)
+NTF_MOCK_METHOD(bsl::shared_ptr<bdlbb::Blob>, createIncomingBlob)
+NTF_MOCK_METHOD(bsl::shared_ptr<bdlbb::Blob>, createOutgoingBlob)
+NTF_MOCK_METHOD(void, createIncomingBlobBuffer, bdlbb::BlobBuffer*)
+NTF_MOCK_METHOD(void, createOutgoingBlobBuffer, bdlbb::BlobBuffer*)
+NTF_MOCK_METHOD_CONST(const bsl::shared_ptr<bdlbb::BlobBufferFactory>&,
+                      incomingBlobBufferFactory)
+NTF_MOCK_METHOD_CONST(const bsl::shared_ptr<bdlbb::BlobBufferFactory>&,
+                      outgoingBlobBufferFactory)
+NTF_MOCK_CLASS_END;
+
+}  // close package namespace
+}  // close enterprise namespace
+
+#endif