From e9059c214c1b055fb0850a4ef858c43f9bb3912f Mon Sep 17 00:00:00 2001
From: Dominika Maziec <int_domz@pl004007.MobicaPL.local>
Date: Thu, 1 Aug 2019 16:40:04 +0200
Subject: [PATCH] TLSSocketWrapper::recvfrom sets SocketAddress output variable

UNITTEST added. TLSSocketWrapper unittest does not use stoip4_stub anymore.
---
 .../netsocket/TLSSocketWrapper/test_TLSSocketWrapper.cpp       | 1 +
 UNITTESTS/features/netsocket/TLSSocketWrapper/unittest.cmake   | 2 --
 features/netsocket/TLSSocketWrapper.cpp                        | 3 +++
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/UNITTESTS/features/netsocket/TLSSocketWrapper/test_TLSSocketWrapper.cpp b/UNITTESTS/features/netsocket/TLSSocketWrapper/test_TLSSocketWrapper.cpp
index 201924e794..1d62f948d1 100644
--- a/UNITTESTS/features/netsocket/TLSSocketWrapper/test_TLSSocketWrapper.cpp
+++ b/UNITTESTS/features/netsocket/TLSSocketWrapper/test_TLSSocketWrapper.cpp
@@ -307,6 +307,7 @@ TEST_F(TestTLSSocketWrapper, recv_from)
     EXPECT_EQ(wrapper->connect(a), NSAPI_ERROR_OK);
     SocketAddress b;
     EXPECT_EQ(wrapper->recvfrom(&b, dataBuf, dataSize), NSAPI_ERROR_OK);
+    EXPECT_EQ(a, b);
 }
 
 TEST_F(TestTLSSocketWrapper, recv_from_null)
diff --git a/UNITTESTS/features/netsocket/TLSSocketWrapper/unittest.cmake b/UNITTESTS/features/netsocket/TLSSocketWrapper/unittest.cmake
index 2ff13c2dc6..d126328baf 100644
--- a/UNITTESTS/features/netsocket/TLSSocketWrapper/unittest.cmake
+++ b/UNITTESTS/features/netsocket/TLSSocketWrapper/unittest.cmake
@@ -28,8 +28,6 @@ set(unittest-test-sources
   stubs/mbed_shared_queues_stub.cpp
   stubs/nsapi_dns_stub.cpp
   stubs/EventFlags_stub.cpp
-  stubs/stoip4_stub.c
-  stubs/ip4tos_stub.c
   stubs/SocketStats_Stub.cpp
 )
 
diff --git a/features/netsocket/TLSSocketWrapper.cpp b/features/netsocket/TLSSocketWrapper.cpp
index 605926477e..68d24127b2 100644
--- a/features/netsocket/TLSSocketWrapper.cpp
+++ b/features/netsocket/TLSSocketWrapper.cpp
@@ -398,6 +398,9 @@ nsapi_size_or_error_t TLSSocketWrapper::recv(void *data, nsapi_size_t size)
 
 nsapi_size_or_error_t TLSSocketWrapper::recvfrom(SocketAddress *address, void *data, nsapi_size_t size)
 {
+    if (address) {
+        getpeername(address);
+    }
     return recv(data, size);
 }