All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.evosuite.runtime.mock.java.net.MockURL Maven / Gradle / Ivy

There is a newer version: 1.0.6
Show newest version
/**
 * Copyright (C) 2010-2016 Gordon Fraser, Andrea Arcuri and EvoSuite
 * contributors
 *
 * This file is part of EvoSuite.
 *
 * EvoSuite is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3.0 of the License, or
 * (at your option) any later version.
 *
 * EvoSuite is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with EvoSuite. If not, see .
 */
package org.evosuite.runtime.mock.java.net;

import java.io.InputStream;
import java.lang.reflect.InvocationTargetException;
import java.net.MalformedURLException;
import java.net.Proxy;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.URLConnection;
import java.net.URLStreamHandler;
import java.net.URLStreamHandlerFactory;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import org.evosuite.runtime.mock.StaticReplacementMock;
import org.evosuite.runtime.mock.java.io.MockIOException;


public class MockURL implements StaticReplacementMock{

	@Override
	public String getMockedClassName() {
		return URL.class.getName();
	}

	private static URLStreamHandlerFactory factory;
	private static Map handlers = new ConcurrentHashMap<>();
	private static final Object streamHandlerLock = new Object();


	public static void initStaticState(){
		factory = null;
		handlers.clear();
	}

    /**
     * Provide a valid URL example for the http protocol.
     */
    public static URL getHttpExample(){
        try {
            return URL("http://www.someFakeButWellFormedURL.org/fooExample");
        } catch (MalformedURLException e) {
            //should never happen
            throw new RuntimeException(e);
        }
    }

	// -----  constructors ------------

	public static URL URL(String spec) throws MalformedURLException {
		return URL(null, spec);
	}

	public static URL URL(URL context, String spec) throws MalformedURLException {
		return URL(context, spec, null);
	}

	public static URL URL(String protocol, String host, String file)
			throws MalformedURLException {
		return URL(protocol, host, -1, file);
	}

	public static URL URL(String protocol, String host, int port, String file)
			throws MalformedURLException{
		return URL(protocol, host, port, file, null);
	}

	public static URL URL(String protocol, String host, int port, String file,
			URLStreamHandler handler) throws MalformedURLException {


		URL url = new URL(protocol,host,port,file,handler);

		//we just need to deal with "handler" if it wasn't specified
		if(handler == null){
			/*
			 * if no handler is specified, then parent would load one based on 
			 * protocol. As the function there is package level, we cannot modify/override it,
			 * and we still have to call a constructor.
			 * So, just replace the handler via reflection
			 */
			handler = getMockedURLStreamHandler(protocol);
			URLUtil.setHandler(url, handler);
		}

		return url;
	}


	public static URL URL(URL context, String spec, URLStreamHandler handler)
			throws MalformedURLException{

		URL url = new URL(context,spec,handler);

		//we just need to deal with "handler" if it wasn't specified
		if(handler == null){
			/*
			 * if no handler is specified, then parent would load one based on 
			 * protocol. As the function there is package level, we cannot modify/override it,
			 * and we still have to call a constructor.
			 * So, just replace the handler via reflection
			 */
			handler = getMockedURLStreamHandler(url.getProtocol());
			URLUtil.setHandler(url, handler);

			//this is needed, as called on the handler in the constructor we are mocking
			handleParseUrl(url,spec,handler);
		}

		return url;
	}



	private static void handleParseUrl(URL url, String spec, URLStreamHandler handler) throws MalformedURLException {

		//code here is based on URL constructor

		int i, limit, c;
		int start = 0;
		boolean aRef=false;

		limit = spec.length();
		while ((limit > 0) && (spec.charAt(limit - 1) <= ' ')) {
			limit--;        //eliminate trailing whitespace
		}
		while ((start < limit) && (spec.charAt(start) <= ' ')) {
			start++;        // eliminate leading whitespace
		}

		if (spec.regionMatches(true, start, "url:", 0, 4)) {
			start += 4;
		}

		if (start < spec.length() && spec.charAt(start) == '#') {
			aRef=true;
		}

		for (i = start ; !aRef && (i < limit) &&
				((c = spec.charAt(i)) != '/') ; i++) {
			if (c == ':') {

				String s = spec.substring(start, i).toLowerCase();
				if (isValidProtocol(s)) {
					start = i + 1;
				}
				break;
			}
		}

		i = spec.indexOf('#', start);
		if (i >= 0) {
			limit = i;
		}

        try {
            URLStreamHandlerUtil.parseURL(handler, url, spec, start, limit);
        } catch (InvocationTargetException e) {
           throw new MalformedURLException(e.getCause().toString());
        }
    }

	//From URL
	private static boolean isValidProtocol(String protocol) {
		int len = protocol.length();
		if (len < 1)
			return false;
		char c = protocol.charAt(0);
		if (!Character.isLetter(c))
			return false;
		for (int i = 1; i < len; i++) {
			c = protocol.charAt(i);
			if (!Character.isLetterOrDigit(c) && c != '.' && c != '+' &&
					c != '-') {
				return false;
			}
		}
		return true;
	}

	// ---------------------



	public static String getQuery(URL url) {
		return url.getQuery();
	}

	public static String getPath(URL url) {
		return url.getPath();
	}

	public static String getUserInfo(URL url) {
		return url.getUserInfo();
	}

	public static String getAuthority(URL url) {
		return url.getAuthority();
	}

	public static int getPort(URL url) {
		return url.getPort();
	}

	public static int getDefaultPort(URL url) {
		return url.getDefaultPort();
	}

	public static String getProtocol(URL url) {
		return url.getProtocol();
	}

	public static String getHost(URL url) {
		return url.getHost();
	}

	public static String getFile(URL url) {
		return url.getFile();
	}

	public static String getRef(URL url) {
		return url.getRef();
	}


	public static boolean equals(URL url, Object obj) {
		return url.equals(obj);
	}


	public static synchronized int hashCode(URL url) {
		return url.hashCode();
	}


	public static boolean sameFile(URL url, URL other) {
		return url.sameFile(other);
	}


	public static String toString(URL url) {
		return url.toString();
	}


	public static String toExternalForm(URL url) {
		return url.toExternalForm();
	}


	public static URI toURI(URL url) throws URISyntaxException {
		return new URI (url.toString());
	}


	public static URLConnection openConnection(URL url) throws java.io.IOException {
		return url.openConnection();
	}


	public static URLConnection openConnection(URL url, Proxy proxy)
			throws java.io.IOException {
		if (proxy == null) {
			throw new IllegalArgumentException("proxy can not be null");
		}

		// Create a copy of Proxy as a security measure
		//Proxy p = proxy == Proxy.NO_PROXY ? Proxy.NO_PROXY : sun.net.ApplicationProxy.create(proxy);
        try {
            return URLStreamHandlerUtil.openConnection(URLUtil.getHandler(url), url, proxy);
        } catch (InvocationTargetException e) {
            throw new MockIOException(e.getCause());
        }
    }


	public static InputStream openStream(URL url) throws java.io.IOException {
		return url.openStream();
	}

	public static Object getContent(URL url) throws java.io.IOException {
		return url.getContent();
	}

	public static Object getContent(URL url, Class[] classes)
			throws java.io.IOException {
		return url.getContent(classes);
	}

	public static void setURLStreamHandlerFactory(URLStreamHandlerFactory fac) {
		synchronized (streamHandlerLock) {
			if (factory != null) {
				throw new Error("factory already defined");
			}
			handlers.clear();
			factory = fac;
		}
	}

	protected static URLStreamHandler getMockedURLStreamHandler(String protocol) throws MalformedURLException {

		URLStreamHandler handler = (URLStreamHandler)handlers.get(protocol);
		if (handler == null) {

			// Use the factory (if any)
			if (factory != null) {
				handler = factory.createURLStreamHandler(protocol);
			}

			// create new instance
			if (handler == null){
				if(EvoURLStreamHandler.isValidProtocol(protocol)) {
					handler = new EvoURLStreamHandler(protocol);
				} else {
					throw new MalformedURLException("unknown protocol: "+protocol);
				}
			}

			handlers.put(protocol, handler);
		}

		return handler;
	}
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy