/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.wicket.markup.html;
import java.io.IOException;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.List;
import org.apache.wicket.MarkupContainer;
import org.apache.wicket.WicketTestCase;
import org.apache.wicket.markup.IMarkupResourceStreamProvider;
import org.apache.wicket.markup.parser.XmlPullParser;
import org.apache.wicket.markup.parser.XmlTag;
import org.apache.wicket.request.resource.PackageResourceReference;
import org.apache.wicket.request.resource.ResourceReference;
import org.apache.wicket.resource.aggregation.AbstractResourceAggregatingHeaderResponse;
import org.apache.wicket.resource.aggregation.ResourceReferenceAndStringData;
import org.apache.wicket.resource.aggregation.ResourceReferenceCollection;
import org.apache.wicket.util.resource.IResourceStream;
import org.apache.wicket.util.resource.ResourceStreamNotFoundException;
import org.apache.wicket.util.resource.StringResourceStream;
/**
* @author Pedro Santos
*/
public class DecoratingHeaderResponseTest extends WicketTestCase
{
/**
* Basic IHeaderResponseDecorator, just prepending the DECORATED string to resource name.
*
* @throws IOException
* @throws ResourceStreamNotFoundException
* @throws ParseException
*/
public void testDecoratedStringPrepend() throws IOException, ResourceStreamNotFoundException,
ParseException
{
tester.getApplication().setHeaderResponseDecorator(new IHeaderResponseDecorator()
{
public IHeaderResponse decorate(IHeaderResponse response)
{
return new DecoratingHeaderResponse(response)
{
@Override
public void renderJavaScriptReference(ResourceReference reference, String id)
{
super.renderJavaScriptReference(new PackageResourceReference("DECORATED-" +
reference.getName()), id);
}
};
}
});
tester.startPage(TestPage.class);
XmlPullParser parser = new XmlPullParser();
parser.parse(tester.getLastResponseAsString());
XmlTag tag = parser.nextTag();
boolean isDecorated = false;
do
{
if (tag.isOpen() && "script".equals(tag.getName()))
{
isDecorated = tag.getAttribute("src").toString().contains("DECORATED");
if (!isDecorated)
{
fail();
}
break;
}
}
while ((tag = parser.nextTag()) != null);
assertTrue(isDecorated);
}
/**
* Test even and odd resources id rendered grouped.
*
* @throws ResourceStreamNotFoundException
* @throws IOException
* @throws ParseException
*
*/
public void testEvenOddResourceIdGroup() throws IOException, ResourceStreamNotFoundException,
ParseException
{
tester.getApplication().setHeaderResponseDecorator(new IHeaderResponseDecorator()
{
public IHeaderResponse decorate(IHeaderResponse response)
{
return new AbstractResourceAggregatingHeaderResponse<ResourceReferenceCollection, Integer>(
response)
{
@Override
protected Integer newGroupingKey(ResourceReferenceAndStringData ref)
{
return Integer.parseInt(ref.getString()) % 2;
}
};
}
});
tester.startPage(TestPage.class);
XmlPullParser parser = new XmlPullParser();
parser.parse(tester.getLastResponseAsString());
XmlTag tag = parser.nextTag();
List<Integer> resourcesId = new ArrayList<Integer>();
do
{
if (tag.isOpen() && "script".equals(tag.getName()))
{
resourcesId.add(Integer.parseInt(tag.getAttribute("id").toString()));
}
}
while ((tag = parser.nextTag()) != null);
int oddEvenCanges = 0;
for (int i = 1; i < resourcesId.size(); i++)
{
if (resourcesId.get(i) % 2 != resourcesId.get(i - 1) % 2)
{
oddEvenCanges++;
}
}
assertEquals(1, oddEvenCanges);
}
/**
*
*/
public static class TestPage extends WebPage implements IMarkupResourceStreamProvider
{
private static final long serialVersionUID = 1L;
@Override
public void renderHead(IHeaderResponse response)
{
for (int i = 0; i < 10; i++)
{
response.renderJavaScriptReference(new PackageResourceReference("res" + i),
Integer.toString(i));
}
}
public IResourceStream getMarkupResourceStream(MarkupContainer container,
Class<?> containerClass)
{
return new StringResourceStream("<html><body></body></html>");
}
}
}