/*
* JBoss, Home of Professional Open Source
* Copyright 2009, Red Hat Middleware LLC, and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.acme.web;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.net.URI;
import java.net.URL;
import org.jboss.arquillian.api.ArquillianResource;
import org.jboss.arquillian.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
/**
* LocalRunServletTestCase
*
* @author <a href="mailto:aslak@redhat.com">Aslak Knutsen</a>
* @version $Revision: $
*/
@RunWith(Arquillian.class)
public class LocalRunServletTestCase
{
@Deployment(testable = false)
public static WebArchive createDeployment()
{
return ShrinkWrap.create(WebArchive.class, "test.war")
.addClass(TestServlet.class);
}
@ArquillianResource
private URI uri;
@Test
public void shouldResolveURI() throws Exception
{
Assert.assertNotNull(uri);
}
@Test
public void shouldBeAbleToInjectBaseHTTPContext(@ArquillianResource URL httpContext) throws Exception
{
String body = readAllAndClose(new URL(httpContext, "/test/TestServlet").openStream());
Assert.assertEquals(
"Verify that the servlet was deployed and returns expected result",
"hello",
body);
}
@Test
public void shouldBeAbleToInjectBaseServletContext(@ArquillianResource(TestServlet.class) URL testServlet) throws Exception
{
String body = readAllAndClose(testServlet.openStream());
Assert.assertEquals(
"Verify that the servlet was deployed and returns expected result",
"hello",
body);
}
private String readAllAndClose(InputStream is) throws Exception
{
ByteArrayOutputStream out = new ByteArrayOutputStream();
try
{
int read;
while( (read = is.read()) != -1)
{
out.write(read);
}
}
finally
{
try { is.close(); } catch (Exception e) { }
}
return out.toString();
}
}