Update unit tests - no longer using socket.shutdown()

Author: arp102

kmip/tests/unit/services/server/test_server.py

@@ -338,7 +338,6 @@ def test_stop(self, logging_mock, engine_mock):
             s._logger.info.assert_any_call(
                 "Shutting down server socket handler."
             )
-            s._socket.shutdown.assert_called_once_with(socket.SHUT_RDWR)
             s._socket.close.assert_called_once_with()
 
         # Test the expected behavior when stopping multiple server session
@@ -366,7 +365,6 @@ def test_stop(self, logging_mock, engine_mock):
             s._logger.info.assert_any_call(
                 "Shutting down server socket handler."
             )
-            s._socket.shutdown.assert_called_once_with(socket.SHUT_RDWR)
             s._socket.close.assert_called_once_with()
 
         thread_mock.reset_mock()
@@ -392,7 +390,6 @@ def test_stop(self, logging_mock, engine_mock):
             s._logger.info.assert_any_call(
                 "Shutting down server socket handler."
             )
-            s._socket.shutdown.assert_called_once_with(socket.SHUT_RDWR)
             s._socket.close.assert_called_once_with()
 
         # Test that the right errors and log messages are generated when
@@ -403,7 +400,7 @@ def test_stop(self, logging_mock, engine_mock):
         test_exception = Exception()
         s._socket.close = mock.MagicMock(side_effect=test_exception)
 
-        regex = "Server failed to shutdown socket handler."
+        regex = "Server failed to close socket handler."
         self.assertRaisesRegex(
             exceptions.NetworkingError,
             regex,
@@ -415,7 +412,6 @@ def test_stop(self, logging_mock, engine_mock):
         s._logger.info.assert_any_call(
             "Shutting down server socket handler."
         )
-        s._socket.shutdown.assert_called_once_with(socket.SHUT_RDWR)
         s._socket.close.assert_called_once_with()
         s._logger.exception(test_exception)
 
@@ -456,7 +452,6 @@ def test_stop_with_monitor_shutdown_error(self, logging_mock, engine_mock):
         s._logger.info.assert_any_call(
             "Shutting down server socket handler."
         )
-        s._socket.shutdown.assert_called_once_with(socket.SHUT_RDWR)
         s._socket.close.assert_called_once_with()
 
         s.policy_monitor.stop.assert_called_once_with()

kmip/tests/unit/services/server/test_session.py

@@ -21,7 +21,6 @@
 
 import datetime
 import mock
-import socket
 import testtools
 import time
 
@@ -149,9 +148,6 @@ def test_run(self):
 
         kmip_session._logger.info.assert_any_call("Starting session: name")
         self.assertTrue(kmip_session._handle_message_loop.called)
-        kmip_session._connection.shutdown.assert_called_once_with(
-            socket.SHUT_RDWR
-        )
         kmip_session._connection.close.assert_called_once_with()
         kmip_session._logger.info.assert_called_with("Stopping session: name")
 
@@ -180,9 +176,6 @@ def test_run_with_failure(self):
             "Failure handling message loop"
         )
         kmip_session._logger.exception.assert_called_once_with(test_exception)
-        kmip_session._connection.shutdown.assert_called_once_with(
-            socket.SHUT_RDWR
-        )
         kmip_session._connection.close.assert_called_once_with()
         kmip_session._logger.info.assert_called_with("Stopping session: name")
 

kmip/tests/unit/services/test_kmip_client.py

@@ -170,7 +170,6 @@ def test_close(self):
         c.close()
 
         self.assertEqual(None, c.socket)
-        c_socket.shutdown.assert_called_once_with(socket.SHUT_RDWR)
         c_socket.close.assert_called_once()
 
     def test_close_with_shutdown_error(self):
@@ -193,7 +192,6 @@ def test_close_with_shutdown_error(self):
         c.close()
 
         self.assertEqual(None, c.socket)
-        c_socket.shutdown.assert_called_once_with(socket.SHUT_RDWR)
         c_socket.close.assert_not_called()
 
     # TODO (peter-hamilton) Modify for credential type and/or add new test