content: /* ===========================================================
   * JFreeChart : a free chart library for the Java(tm) platform
   * ===========================================================
   *
   * (C) Copyright 2000-2012, 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.]
   *
   * --------------------
   * GanttChartTests.java
   * --------------------
   * (C) Copyright 2005-2008, 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 org.jfree.chart.event.ChartChangeEvent;
  import org.jfree.chart.event.ChartChangeListener;
  import org.jfree.chart.labels.CategoryToolTipGenerator;
  import org.jfree.chart.labels.StandardCategoryToolTipGenerator;
  import org.jfree.chart.plot.CategoryPlot;
  import org.jfree.chart.renderer.category.CategoryItemRenderer;
  import org.jfree.chart.urls.CategoryURLGenerator;
  import org.jfree.chart.urls.StandardCategoryURLGenerator;
  import org.jfree.data.category.IntervalCategoryDataset;
  import org.jfree.data.gantt.Task;
  import org.jfree.data.gantt.TaskSeries;
  import org.jfree.data.gantt.TaskSeriesCollection;
  import org.jfree.data.time.SimpleTimePeriod;
  import org.junit.Before;
  import org.junit.Test;
  
  import java.awt.Graphics2D;
  import java.awt.geom.Rectangle2D;
  import java.awt.image.BufferedImage;
  import java.util.Calendar;
  import java.util.Date;
  
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertSame;
  
  /**
   * Some tests for a Gantt chart.
   */
  public class GanttChartTest  {
  
      /** A chart. */
      private JFreeChart chart;
  
  
  
  
  
      /**
       * Common test setup.
       */
      @Before
      public void setUp() {
          this.chart = createGanttChart();
      }
  
      /**
       * Draws the chart with a <code>null</code> info object to make sure that
       * no exceptions are thrown (a problem that was occurring at one point).
       */
      @Test
      public void testDrawWithNullInfo() {
          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();
          //FIXME we should really assert a value
  
      }
  
      /**
       * Draws the chart with a <code>null</code> info object to make sure that
       * no exceptions are thrown (a problem that was occurring at one point).
       */
      @Test
      public void testDrawWithNullInfo2() {
          JFreeChart chart = createGanttChart();
          CategoryPlot plot = (CategoryPlot) chart.getPlot();
          plot.setDataset(createDataset());
          /* BufferedImage img =*/ chart.createBufferedImage(300, 200, null);
          //FIXME we should really assert a value
      }
  
      /**
       * Replaces the chart's dataset and then checks that the new dataset is OK.
       */
      @Test
      public void testReplaceDataset() {
          LocalListener l = new LocalListener();
          this.chart.addChangeListener(l);
          CategoryPlot plot = (CategoryPlot) this.chart.getPlot();
          plot.setDataset(null);
          assertEquals(true, l.flag);
      }
  
      /**
       * 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);
          assertSame(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);
          assertSame(url2, url1);
      }
  
      /**
       * Create a Gantt chart.
       *
       * @return The chart.
       */
      private static JFreeChart createGanttChart() {
          return ChartFactory.createGanttChart("Gantt Chart", "Domain", "Range",
                  null);
      }
  
      /**
       * Creates a sample dataset for a Gantt chart.
       *
       * @return The dataset.
       */
      public static IntervalCategoryDataset createDataset() {
  
          TaskSeries s1 = new TaskSeries("Scheduled");
          s1.add(new Task("Write Proposal",
                 new SimpleTimePeriod(date(1, Calendar.APRIL, 2001),
                                      date(5, Calendar.APRIL, 2001))));
          s1.add(new Task("Obtain Approval",
                 new SimpleTimePeriod(date(9, Calendar.APRIL, 2001),
                                      date(9, Calendar.APRIL, 2001))));
          s1.add(new Task("Requirements Analysis",
                 new SimpleTimePeriod(date(10, Calendar.APRIL, 2001),
                                      date(5, Calendar.MAY, 2001))));
          s1.add(new Task("Design Phase",
                 new SimpleTimePeriod(date(6, Calendar.MAY, 2001),
                                      date(30, Calendar.MAY, 2001))));
          s1.add(new Task("Design Signoff",
                 new SimpleTimePeriod(date(2, Calendar.JUNE, 2001),
                                      date(2, Calendar.JUNE, 2001))));
          s1.add(new Task("Alpha Implementation",
                 new SimpleTimePeriod(date(3, Calendar.JUNE, 2001),
                                      date(31, Calendar.JULY, 2001))));
          s1.add(new Task("Design Review",
                 new SimpleTimePeriod(date(1, Calendar.AUGUST, 2001),
                                      date(8, Calendar.AUGUST, 2001))));
          s1.add(new Task("Revised Design Signoff",
                 new SimpleTimePeriod(date(10, Calendar.AUGUST, 2001),
                                      date(10, Calendar.AUGUST, 2001))));
          s1.add(new Task("Beta Implementation",
                 new SimpleTimePeriod(date(12, Calendar.AUGUST, 2001),
                                      date(12, Calendar.SEPTEMBER, 2001))));
          s1.add(new Task("Testing",
                 new SimpleTimePeriod(date(13, Calendar.SEPTEMBER, 2001),
                                      date(31, Calendar.OCTOBER, 2001))));
          s1.add(new Task("Final Implementation",
                 new SimpleTimePeriod(date(1, Calendar.NOVEMBER, 2001),
                                      date(15, Calendar.NOVEMBER, 2001))));
          s1.add(new Task("Signoff",
                 new SimpleTimePeriod(date(28, Calendar.NOVEMBER, 2001),
                                      date(30, Calendar.NOVEMBER, 2001))));
  
          TaskSeries s2 = new TaskSeries("Actual");
          s2.add(new Task("Write Proposal",
                 new SimpleTimePeriod(date(1, Calendar.APRIL, 2001),
                                      date(5, Calendar.APRIL, 2001))));
          s2.add(new Task("Obtain Approval",
                 new SimpleTimePeriod(date(9, Calendar.APRIL, 2001),
                                      date(9, Calendar.APRIL, 2001))));
          s2.add(new Task("Requirements Analysis",
                 new SimpleTimePeriod(date(10, Calendar.APRIL, 2001),
                                      date(15, Calendar.MAY, 2001))));
          s2.add(new Task("Design Phase",
                 new SimpleTimePeriod(date(15, Calendar.MAY, 2001),
                                      date(17, Calendar.JUNE, 2001))));
          s2.add(new Task("Design Signoff",
                 new SimpleTimePeriod(date(30, Calendar.JUNE, 2001),
                                      date(30, Calendar.JUNE, 2001))));
          s2.add(new Task("Alpha Implementation",
                 new SimpleTimePeriod(date(1, Calendar.JULY, 2001),
                                      date(12, Calendar.SEPTEMBER, 2001))));
          s2.add(new Task("Design Review",
                 new SimpleTimePeriod(date(12, Calendar.SEPTEMBER, 2001),
                                      date(22, Calendar.SEPTEMBER, 2001))));
          s2.add(new Task("Revised Design Signoff",
                 new SimpleTimePeriod(date(25, Calendar.SEPTEMBER, 2001),
                                      date(27, Calendar.SEPTEMBER, 2001))));
          s2.add(new Task("Beta Implementation",
                 new SimpleTimePeriod(date(27, Calendar.SEPTEMBER, 2001),
                                      date(30, Calendar.OCTOBER, 2001))));
          s2.add(new Task("Testing",
                 new SimpleTimePeriod(date(31, Calendar.OCTOBER, 2001),
                                      date(17, Calendar.NOVEMBER, 2001))));
          s2.add(new Task("Final Implementation",
                 new SimpleTimePeriod(date(18, Calendar.NOVEMBER, 2001),
                                      date(5, Calendar.DECEMBER, 2001))));
          s2.add(new Task("Signoff",
                 new SimpleTimePeriod(date(10, Calendar.DECEMBER, 2001),
                                      date(11, Calendar.DECEMBER, 2001))));
  
          TaskSeriesCollection collection = new TaskSeriesCollection();
          collection.add(s1);
          collection.add(s2);
  
          return collection;
      }
  
      /**
       * Utility method for creating <code>Date</code> objects.
       *
       * @param day  the date.
       * @param month  the month.
       * @param year  the year.
       *
       * @return a date.
       */
      private static Date date(int day, int month, int year) {
  
          Calendar calendar = Calendar.getInstance();
          calendar.set(year, month, day);
          Date result = calendar.getTime();
          return result;
  
      }    /**
       * A chart change listener.
       *
       */
      static class LocalListener implements ChartChangeListener {
  
          /** A flag. */
          private boolean flag;
  
          /**
           * Event handler.
           *
           * @param event  the event.
           */
          @Override
          public void chartChanged(ChartChangeEvent event) {
              this.flag = true;
          }
  
      }
  
  }
  