/*
* $Id: ChartResultTest.java 651946 2008-04-27 13:41:38Z apetrelli $
*
* 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.struts2.dispatcher;
import com.mockobjects.dynamic.Mock;
import org.apache.struts2.ServletActionContext;
import com.opensymphony.xwork2.ActionContext;
import com.opensymphony.xwork2.ActionInvocation;
import junit.framework.TestCase;
import ognl.Ognl;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.JFreeChart;
import org.jfree.data.general.DefaultPieDataset;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
/**
*/
public class ChartResultTest extends TestCase {
private ActionInvocation actionInvocation;
private JFreeChart mockChart;
private Mock responseMock;
private MockServletOutputStream os;
public void testChart() throws Exception {
responseMock.expectAndReturn("getOutputStream", os);
ChartResult result = new ChartResult();
result.setChart(mockChart);
result.setHeight(10);
result.setWidth(10);
result.execute(actionInvocation);
responseMock.verify();
assertTrue(os.isWritten());
}
public void testChartNotSet() {
ChartResult result = new ChartResult();
// expect exception if chart not set.
result.setChart(null);
try {
result.execute(actionInvocation);
fail();
} catch (Exception e) {
}
responseMock.verify();
assertFalse(os.isWritten());
}
protected void setUp() throws Exception {
DefaultPieDataset data = new DefaultPieDataset();
data.setValue("Java", new Double(43.2));
data.setValue("Visual Basic", new Double(0.0));
data.setValue("C/C++", new Double(17.5));
mockChart = ChartFactory.createPieChart("Pie Chart", data, true, true, false);
Mock mockActionInvocation = new Mock(ActionInvocation.class);
actionInvocation = (ActionInvocation) mockActionInvocation.proxy();
os = new MockServletOutputStream();
responseMock = new Mock(HttpServletResponse.class);
ActionContext.setContext(new ActionContext(Ognl.createDefaultContext(null)));
ServletActionContext.setResponse((HttpServletResponse) responseMock.proxy());
}
protected void tearDown() throws Exception {
actionInvocation = null;
os = null;
responseMock = null;
}
private class MockServletOutputStream extends ServletOutputStream {
// very simple check that outputStream was written to.
private boolean written = false;
/**
* @return Returns the written.
*/
public boolean isWritten() {
return written;
}
public void write(int arg0) throws IOException {
written = true;
}
}
}