def model_service_worker(socket_patches):
model_service_worker = MXNetModelServiceWorker('unix', 'my-socket', None,
None)
model_service_worker.sock = socket_patches.socket
model_service_worker.service = Service('name', 'mpath', 'testmanifest',
None, 0, 1)
return model_service_worker
def test_socket_in_use(self, mocker):
remove = mocker.patch('os.remove')
path_exists = mocker.patch('os.path.exists')
remove.side_effect = OSError()
path_exists.return_value = True
with pytest.raises(Exception, match=r".*socket already in use: sampleSocketName.*"):
MXNetModelServiceWorker('unix', self.socket_name)
def test_success(self, patches):
MXNetModelServiceWorker('unix', self.socket_name)
patches.remove.assert_called_once_with(self.socket_name)
patches.socket.assert_called_once_with(socket.AF_UNIX,
socket.SOCK_STREAM)
def test_missing_socket_name(self):
with pytest.raises(ValueError, match="Invalid socket type provided.*"):
MXNetModelServiceWorker()
def test_missing_socket_name(self):
with pytest.raises(ValueError, match="Incomplete data provided.*"):
MXNetModelServiceWorker()