/* ===========================================================
 
* JFreeChart : a free chart library for the Java(tm) platform
 
* ===========================================================
 
*
 
* (C) Copyright 2000-2016, by Object Refinery Limited and Contributors.
 
*
 
* Project Info:
  
http://www.jfree.org/jfreechart/index.html
 
*
 
* This library is free software; you can redistribute it and/or modify it
 
* under the terms of the GNU Lesser General Public License as published by
 
* the Free Software Foundation; either version 2.1 of the License, or
 
* (at your option) any later version.
 
*
 
* This library is distributed in the hope that it will be useful, but
 
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
 
* License for more details.
 
*
 
* You should have received a copy of the GNU Lesser General Public
 
* License along with this library; if not, write to the Free Software
 
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  
02110-1301,
 
* USA.
 
*
 
* [Oracle and Java are registered trademarks of Oracle and/or its affiliates.
 
* Other names may be trademarks of their respective owners.]
 
*
 
* -----------------------
 
* WaterfallChartTest.java
 
* -----------------------
 
* (C) Copyright 2002-2016, by Object Refinery Limited.
 
*
 
* Original Author:
  
David Gilbert (for Object Refinery Limited);
 
* Contributor(s):
   
-;
 
*
 
* Changes:
 
* --------
 
* 12-Apr-2005 : Version 1 (DG);
 
*
 
*/

package org.jfree.chart;

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

import java.awt.Graphics2D;
import java.awt.geom.Rectangle2D;
import java.awt.image.BufferedImage;

import org.jfree.chart.labels.CategoryToolTipGenerator;
import org.jfree.chart.labels.StandardCategoryToolTipGenerator;
import org.jfree.chart.plot.CategoryPlot;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.renderer.category.CategoryItemRenderer;
import org.jfree.chart.urls.CategoryURLGenerator;
import org.jfree.chart.urls.StandardCategoryURLGenerator;
import org.jfree.data.category.CategoryDataset;
import org.jfree.data.general.DatasetUtils;
import org.junit.Before;
import org.junit.Test;

/**
 
* Some tests for a waterfall chart.
 
*/
public class WaterfallChartTest {

    
/** A chart. */
    
private JFreeChart chart;

    
/**
     
* Common test setup.
     
*/
    
@Before
    
public void setUp() {
        
this.chart = createWaterfallChart();
    
}

    
/**
     
* Draws the chart with a null info object to make sure that no exceptions
     
* are thrown (a problem that was occurring at one point).
     
*/

    
@Test
    
public void testDrawWithNullInfo() {
        
try {
            
BufferedImage image = new BufferedImage(200 , 100,
                    
BufferedImage.TYPE_INT_RGB);
            
Graphics2D g2 = image.createGraphics();
            
this.chart.draw(g2, new Rectangle2D.Double(0, 0, 200, 100), null,
                    
null);
            
g2.dispose();
        
}
        
catch (Exception e) {
            
fail("There should be no exception.");
        
}
    
}

    
/**
     
* Check that setting a tool tip generator for a series does override the
     
* default generator.
     
*/

    
@Test
    
public void testSetSeriesToolTipGenerator() {
        
CategoryPlot plot = (CategoryPlot) this.chart.getPlot();
        
CategoryItemRenderer renderer = plot.getRenderer();
        
StandardCategoryToolTipGenerator tt
                
= new StandardCategoryToolTipGenerator();
        
renderer.setSeriesToolTipGenerator(0, tt);
        
CategoryToolTipGenerator tt2 = renderer.getToolTipGenerator(0, 0);
        
assertTrue(tt2 == tt);
    
}

    
/**
     
* Check that setting a URL generator for a series does override the
     
* default generator.
     
*/

    
@Test
    
public void testSetSeriesURLGenerator() {
        
CategoryPlot plot = (CategoryPlot) this.chart.getPlot();
        
CategoryItemRenderer renderer = plot.getRenderer();
        
StandardCategoryURLGenerator url1
                
= new StandardCategoryURLGenerator();
        
renderer.setSeriesItemURLGenerator(0, url1);
        
CategoryURLGenerator url2 = renderer.getItemURLGenerator(0, 0);
        
assertTrue(url2 == url1);
    
}

    
/**
     
* Create a bar chart with sample data in the range -3 to +3.
     
*
     
* @return The chart.
     
*/

    
private static JFreeChart createWaterfallChart() {
        
Number[][] data = new Integer[][]
            
{{new Integer(-3), new Integer(-2)},
             
{new Integer(-1), new Integer(1)},
             
{new Integer(2), new Integer(3)}};

        
CategoryDataset dataset = DatasetUtils.createCategoryDataset("S",
                
"C", data);
        
return ChartFactory.createWaterfallChart(
            
"Waterfall Chart",
            
"Domain", "Range",
            
dataset,
            
PlotOrientation.HORIZONTAL,
            
true,
     
// include legend
            
true,
            
true
        
);

    
}

}