}
        };

        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);

            if (handler != null) {
                return handler;
            }
            if (factory != null) {
                handler = factory.createURLStreamHandler(protocol);
            }
            if (handler == null) {
                final String path = System.getProperty(HANDLER_PKGS_PROPERTY);

            if (handler != null) {
                return handler;
            }
            if (factory != null) {
                handler = factory.createURLStreamHandler(protocol);
            }
            if (handler == null) {
                final String path = System.getProperty(HANDLER_PKGS_PROPERTY);

                // 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;
                    }