socket.send does not guarantee that all data will be send. (#765)

sendall should be used instead.

Co-authored-by: Geeta Chauhan <[email protected]>
Co-authored-by: dhanainme <[email protected]>

Author: 3 people

ts/model_service_worker.py

@@ -111,12 +111,12 @@ def handle_connection(self, cl_socket):
                 pr.enable()
             if cmd == b'I':
                 resp = service.predict(msg)
-                cl_socket.send(resp)
+                cl_socket.sendall(resp)
             elif cmd == b'L':
                 service, result, code = self.load_model(msg)
                 resp = bytearray()
                 resp += create_load_model_response(code, result)
-                cl_socket.send(resp)
+                cl_socket.sendall(resp)
                 if code != 200:
                     raise RuntimeError("{} - {}".format(code, result))
             else:

ts/tests/unit_tests/test_model_service_worker.py

@@ -170,4 +170,4 @@ def test_handle_connection(self, patches, model_service_worker):
         with pytest.raises(ValueError, match=r"Received unknown command.*"):
             model_service_worker.handle_connection(cl_socket)
 
-        cl_socket.send.assert_called()
+        cl_socket.sendall.assert_called()