}
        };

        URL url = new URL("http://custom.protocol/path");

        // Configure the mock factory to return our handler for the 'http' protocol.
        when(mockFactory.createURLStreamHandler("http")).thenReturn(mockStreamHandler);

        Handler.setURLStreamHandlerFactory(mockFactory);

        URLConnection connection = handler.openConnection(url);

                // Try to register the handler factory
                URL.setURLStreamHandlerFactory(new URLStreamHandlerFactory() {
                    @Override
                    public URLStreamHandler createURLStreamHandler(String protocol) {
                        if ("smb".equals(protocol)) {
                            return new Handler();
                        }
                        return null;
                    }