MultiplePiePlotTest.java 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206
  1. /* ===========================================================
  2. * JFreeChart : a free chart library for the Java(tm) platform
  3. * ===========================================================
  4. *
  5. * (C) Copyright 2000-2013, by Object Refinery Limited and Contributors.
  6. *
  7. * Project Info: http://www.jfree.org/jfreechart/index.html
  8. *
  9. * This library is free software; you can redistribute it and/or modify it
  10. * under the terms of the GNU Lesser General Public License as published by
  11. * the Free Software Foundation; either version 2.1 of the License, or
  12. * (at your option) any later version.
  13. *
  14. * This library is distributed in the hope that it will be useful, but
  15. * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  16. * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
  17. * License for more details.
  18. *
  19. * You should have received a copy of the GNU Lesser General Public
  20. * License along with this library; if not, write to the Free Software
  21. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
  22. * USA.
  23. *
  24. * [Oracle and Java are registered trademarks of Oracle and/or its affiliates.
  25. * Other names may be trademarks of their respective owners.]
  26. *
  27. * ------------------------
  28. * MultiplePiePlotTest.java
  29. * ------------------------
  30. * (C) Copyright 2005-2013, by Object Refinery Limited and Contributors.
  31. *
  32. * Original Author: David Gilbert (for Object Refinery Limited);
  33. * Contributor(s): -;
  34. *
  35. * Changes
  36. * -------
  37. * 16-Jun-2005 : Version 1 (DG);
  38. * 06-Apr-2006 : Added tests for new fields (DG);
  39. * 18-Apr-2008 : Added testConstructor() (DG);
  40. * 30-Dec-2008 : Updated for new legendItemShape field (DG);
  41. * 01-Jun-2009 : Added test for getLegendItems() bug, series key is not
  42. * set (DG);
  43. *
  44. */
  45. package org.jfree.chart.plot;
  46. import static org.junit.Assert.assertEquals;
  47. import static org.junit.Assert.assertFalse;
  48. import static org.junit.Assert.assertTrue;
  49. import static org.junit.Assert.assertNull;
  50. import java.awt.Color;
  51. import java.awt.GradientPaint;
  52. import java.awt.geom.Rectangle2D;
  53. import org.jfree.chart.ChartFactory;
  54. import org.jfree.chart.JFreeChart;
  55. import org.jfree.chart.LegendItem;
  56. import org.jfree.chart.LegendItemCollection;
  57. import org.jfree.chart.TestUtilities;
  58. import org.jfree.chart.event.PlotChangeEvent;
  59. import org.jfree.chart.event.PlotChangeListener;
  60. import org.jfree.data.category.DefaultCategoryDataset;
  61. import org.jfree.util.TableOrder;
  62. import org.junit.Test;
  63. /**
  64. * Some tests for the {@link MultiplePiePlot} class.
  65. */
  66. public class MultiplePiePlotTest implements PlotChangeListener {
  67. /** The last event received. */
  68. PlotChangeEvent lastEvent;
  69. /**
  70. * Receives a plot change event and records it. Some tests will use this
  71. * to check that events have been generated (or not) when required.
  72. *
  73. * @param event the event.
  74. */
  75. @Override
  76. public void plotChanged(PlotChangeEvent event) {
  77. this.lastEvent = event;
  78. }
  79. /**
  80. * Some checks for the constructors.
  81. */
  82. @Test
  83. public void testConstructor() {
  84. MultiplePiePlot plot = new MultiplePiePlot();
  85. assertNull(plot.getDataset());
  86. // the following checks that the plot registers itself as a listener
  87. // with the dataset passed to the constructor - see patch 1943021
  88. DefaultCategoryDataset dataset = new DefaultCategoryDataset();
  89. plot = new MultiplePiePlot(dataset);
  90. assertTrue(dataset.hasListener(plot));
  91. }
  92. /**
  93. * Check that the equals() method distinguishes the required fields.
  94. */
  95. @Test
  96. public void testEquals() {
  97. MultiplePiePlot p1 = new MultiplePiePlot();
  98. MultiplePiePlot p2 = new MultiplePiePlot();
  99. assertTrue(p1.equals(p2));
  100. assertTrue(p2.equals(p1));
  101. p1.setDataExtractOrder(TableOrder.BY_ROW);
  102. assertFalse(p1.equals(p2));
  103. p2.setDataExtractOrder(TableOrder.BY_ROW);
  104. assertTrue(p1.equals(p2));
  105. p1.setLimit(1.23);
  106. assertFalse(p1.equals(p2));
  107. p2.setLimit(1.23);
  108. assertTrue(p1.equals(p2));
  109. p1.setAggregatedItemsKey("Aggregated Items");
  110. assertFalse(p1.equals(p2));
  111. p2.setAggregatedItemsKey("Aggregated Items");
  112. assertTrue(p1.equals(p2));
  113. p1.setAggregatedItemsPaint(new GradientPaint(1.0f, 2.0f, Color.red,
  114. 3.0f, 4.0f, Color.yellow));
  115. assertFalse(p1.equals(p2));
  116. p2.setAggregatedItemsPaint(new GradientPaint(1.0f, 2.0f, Color.red,
  117. 3.0f, 4.0f, Color.yellow));
  118. assertTrue(p1.equals(p2));
  119. p1.setPieChart(ChartFactory.createPieChart("Title", null, true, true,
  120. true));
  121. assertFalse(p1.equals(p2));
  122. p2.setPieChart(ChartFactory.createPieChart("Title", null, true, true,
  123. true));
  124. assertTrue(p1.equals(p2));
  125. p1.setLegendItemShape(new Rectangle2D.Double(1.0, 2.0, 3.0, 4.0));
  126. assertFalse(p1.equals(p2));
  127. p2.setLegendItemShape(new Rectangle2D.Double(1.0, 2.0, 3.0, 4.0));
  128. assertTrue(p1.equals(p2));
  129. }
  130. /**
  131. * Some basic checks for the clone() method.
  132. */
  133. @Test
  134. public void testCloning() throws CloneNotSupportedException {
  135. MultiplePiePlot p1 = new MultiplePiePlot();
  136. Rectangle2D rect = new Rectangle2D.Double(1.0, 2.0, 3.0, 4.0);
  137. p1.setLegendItemShape(rect);
  138. MultiplePiePlot p2 = (MultiplePiePlot) p1.clone();
  139. assertTrue(p1 != p2);
  140. assertTrue(p1.getClass() == p2.getClass());
  141. assertTrue(p1.equals(p2));
  142. // check independence
  143. rect.setRect(2.0, 3.0, 4.0, 5.0);
  144. assertFalse(p1.equals(p2));
  145. }
  146. /**
  147. * Serialize an instance, restore it, and check for equality.
  148. */
  149. @Test
  150. public void testSerialization() {
  151. MultiplePiePlot p1 = new MultiplePiePlot(null);
  152. p1.setAggregatedItemsPaint(new GradientPaint(1.0f, 2.0f, Color.yellow,
  153. 3.0f, 4.0f, Color.red));
  154. MultiplePiePlot p2 = (MultiplePiePlot) TestUtilities.serialised(p1);
  155. assertEquals(p1, p2);
  156. }
  157. /**
  158. * Fetches the legend items and checks the values.
  159. */
  160. @Test
  161. public void testGetLegendItems() {
  162. DefaultCategoryDataset dataset = new DefaultCategoryDataset();
  163. dataset.addValue(35.0, "S1", "C1");
  164. dataset.addValue(45.0, "S1", "C2");
  165. dataset.addValue(55.0, "S2", "C1");
  166. dataset.addValue(15.0, "S2", "C2");
  167. MultiplePiePlot plot = new MultiplePiePlot(dataset);
  168. JFreeChart chart = new JFreeChart(plot);
  169. LegendItemCollection legendItems = plot.getLegendItems();
  170. assertEquals(2, legendItems.getItemCount());
  171. LegendItem item1 = legendItems.get(0);
  172. assertEquals("S1", item1.getLabel());
  173. assertEquals("S1", item1.getSeriesKey());
  174. assertEquals(0, item1.getSeriesIndex());
  175. assertEquals(dataset, item1.getDataset());
  176. assertEquals(0, item1.getDatasetIndex());
  177. LegendItem item2 = legendItems.get(1);
  178. assertEquals("S2", item2.getLabel());
  179. assertEquals("S2", item2.getSeriesKey());
  180. assertEquals(1, item2.getSeriesIndex());
  181. assertEquals(dataset, item2.getDataset());
  182. assertEquals(0, item2.getDatasetIndex());
  183. }
  184. }