Fix state handling APIs

tony-josi-aws · Jun 2, 2024 · b33985c · b33985c
1 parent 43e7c21
commit b33985c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/unit-test/FreeRTOS_TCP_State_Handling/FreeRTOS_TCP_State_Handling_utest.c b/test/unit-test/FreeRTOS_TCP_State_Handling/FreeRTOS_TCP_State_Handling_utest.c
@@ -1800,6 +1800,7 @@ void test_prvTCPSocketCopy_BindError( void )
     pxSocket->u.xTCP.uxTxWinSize = 0x123456;
     pxSocket->pxSocketSet = ( struct xSOCKET_SET * ) 0x1111111;
     pxSocket->xSelectBits = eSELECT_READ;
+    pxSocket->u.xTCP.pxPeerSocket = &MockReturnSocket;
 
     FreeRTOS_GetLocalAddress_ExpectAndReturn( pxSocket, NULL, pdTRUE );
     FreeRTOS_GetLocalAddress_IgnoreArg_pxAddress();
@@ -1811,6 +1812,7 @@ void test_prvTCPSocketCopy_BindError( void )
     TEST_ASSERT_NOT_EQUAL( pxSocket->usLocalPort, MockReturnSocket.usLocalPort );
     TEST_ASSERT_EQUAL( pxSocket->u.xTCP.uxTxWinSize, MockReturnSocket.u.xTCP.uxTxWinSize );
     TEST_ASSERT_EQUAL( ( pxSocket->xSelectBits | eSELECT_READ | eSELECT_EXCEPT ), MockReturnSocket.xSelectBits );
+    TEST_ASSERT_EQUAL( pxSocket->u.xTCP.pxPeerSocket, &MockReturnSocket );
 }
 
 /**