io.yawp.testing.EndpointTestCase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yawp-testing-core Show documentation
YAWP! testing core
There is a newer version: 2.08alpha
package io.yawp.testing;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import io.yawp.commons.http.HttpException;
import io.yawp.commons.utils.JsonUtils;
import io.yawp.driver.api.testing.TestHelper;
import io.yawp.driver.api.testing.TestHelperFactory;
import io.yawp.repository.EndpointScanner;
import io.yawp.repository.Feature;
import io.yawp.repository.ObjectHolder;
import io.yawp.repository.Repository;
import io.yawp.repository.RepositoryFeatures;
import io.yawp.servlet.EndpointServlet;

import org.junit.After;
import org.junit.Before;

public class EndpointTestCase extends Feature {

	private static RepositoryFeatures features;

	private static EndpointServlet servlet;

	protected TestHelper helper;

	@Before
	public void setUp() {
		yawp = Repository.r().setFeatures(getFeatures());
		helper = testHelperDriver(yawp);
		helper.setUp();
	}

	protected String getAppPackage() {
		return "io.yawp";
	}

	private RepositoryFeatures getFeatures() {
		if (features != null) {
			return features;
		}
		features = new EndpointScanner(getAppPackage()).scan();
		return features;
	}

	private TestHelper testHelperDriver(Repository r) {
		return TestHelperFactory.getTestHelper(r);
	}

	@After
	public void tearDownHelper() {
		helper.tearDown();
	}

	private EndpointServlet servlet() {
		if (servlet != null) {
			return servlet;
		}

		servlet = new EndpointServlet(getAppPackage()) {

			private static final long serialVersionUID = 3374113392343671861L;

			@Override
			protected Repository getRepository(Map params) {
				return yawp;
			}

		};

		return servlet;
	}

	protected String get(String uri) {
		return get(uri, new HashMap());
	}

	protected String get(String uri, Map params) {
		return servlet().execute("GET", uri, null, params).getText();
	}

	protected void assertGetWithStatus(String uri, int status) {
		try {
			get(uri);
		} catch (HttpException e) {
			assertEquals(status, e.getHttpStatus());
			return;
		}
		assertTrue(status == 200);
	}

	protected String post(String uri, String json) {
		return post(uri, json, new HashMap());
	}

	protected String post(String uri, String json, Map params) {
		return servlet().execute("POST", uri, json, params).getText();
	}

	protected void assertPostWithStatus(String uri, String json, int status) {
		try {
			post(uri, json);
		} catch (HttpException e) {
			assertEquals(status, e.getHttpStatus());
			return;
		}
		assertTrue(status == 200);
	}

	protected String put(String uri) {
		return put(uri, null, new HashMap());
	}

	protected String put(String uri, String json) {
		return put(uri, json, new HashMap());
	}

	protected String patch(String uri) {
		return patch(uri, null, new HashMap());
	}

	protected String patch(String uri, String json) {
		return patch(uri, json, new HashMap());
	}

	protected void assertPutWithStatus(String uri, String json, Map params, int status) {
		try {
			put(uri, json, params);
		} catch (HttpException e) {
			assertEquals(status, e.getHttpStatus());
			return;
		}
		assertTrue(status == 200);
	}

	protected void assertPutWithStatus(String uri, int status) {
		assertPutWithStatus(uri, null, new HashMap(), status);
	}

	protected void assertPutWithStatus(String uri, String json, int status) {
		assertPutWithStatus(uri, json, new HashMap(), status);
	}

	protected void assertPutWithStatus(String uri, Map params, int status) {
		assertPutWithStatus(uri, null, params, status);
	}

	protected String put(String uri, Map params) {
		return servlet().execute("PUT", uri, null, params).getText();
	}

	protected String put(String uri, String json, Map params) {
		return servlet().execute("PUT", uri, json, params).getText();
	}

	protected String patch(String uri, Map params) {
		return servlet().execute("PATCH", uri, null, params).getText();
	}

	protected String patch(String uri, String json, Map params) {
		return servlet().execute("PATCH", uri, json, params).getText();
	}

	protected String delete(String uri) {
		return servlet().execute("DELETE", uri, null, new HashMap()).getText();
	}

	protected void assertDeleteWithStatus(String uri, int status) {
		try {
			delete(uri);
		} catch (HttpException e) {
			assertEquals(status, e.getHttpStatus());
			return;
		}
		assertTrue(status == 200);
	}

	protected  T from(String json, Class clazz) {
		return JsonUtils.from(yawp, json, clazz);
	}

	protected  List fromList(String json, Class clazz) {
		return JsonUtils.fromList(yawp, json, clazz);
	}

	protected String parseIds(String format, Object... objects) {
		List longIds = new ArrayList();

		for (Object object : objects) {
			ObjectHolder objectHolder = new ObjectHolder(object);
			longIds.add(String.valueOf(objectHolder.getId().getSimpleValue()));
		}

		return String.format(format, longIds.toArray());
	}

	protected String uri(String uriFormat, Object... objects) {
		return parseIds(uriFormat, objects);
	}

	protected String json(String uriFormat, Object... objects) {
		return parseIds(uriFormat, objects);
	}

	protected Map params(String key, String value) {
		Map map = new HashMap();
		map.put(key, value);
		return map;
	}
}