expose mock_socket as a decorator