LoadFromREST.java

package org.jbehave.core.io.rest;

import org.jbehave.core.io.InvalidStoryResource;
import org.jbehave.core.io.ResourceLoader;
import org.jbehave.core.io.rest.RESTClient.Type;

/**
 * Loads resource from REST
 */
public class LoadFromREST implements ResourceLoader {

    private RESTClient client; 
    
    public LoadFromREST(Type type) {
        this(type, null, null);
    }

    public LoadFromREST(Type type, String username, String password) {
        this.client = new RESTClient(type, username, password);
    }
    
    public LoadFromREST(RESTClient client) {
        this.client = client;
    }
    
    public String loadResourceAsText(String resourcePath) {
		try {
			Type type = client.getType();
            return text(get(uri(resourcePath, type)), type);
		} catch (Exception cause) {
			throw new InvalidStoryResource(resourcePath, cause);
		}
	}

	protected String uri(String resourcePath, Type type) {
		return resourcePath;
	}

	protected String text(String entity, Type type) {
		return entity;
	}

	private String get(String uri) {
		return client.get(uri);
	}

}