/sigmah/src/test/java/org/sigmah/server/report/generator/map/PieMapMarkerTest.java

http://sigma-h.googlecode.com/ · Java · 70 lines · 47 code · 16 blank · 7 comment · 0 complexity · 0ca64a3b946e1adec9231b30e63ac995 MD5 · raw file

  1. /*
  2. * All Sigmah code is released under the GNU General Public License v3
  3. * See COPYRIGHT.txt and LICENSE.txt.
  4. */
  5. package org.sigmah.server.report.generator.map;
  6. import junit.framework.Assert;
  7. import org.junit.Test;
  8. import org.sigmah.shared.dao.SiteTableColumn;
  9. import org.sigmah.server.domain.SiteData;
  10. import org.sigmah.shared.report.content.LatLng;
  11. import org.sigmah.shared.report.content.MapContent;
  12. import org.sigmah.shared.report.content.PieMapMarker;
  13. import org.sigmah.shared.report.model.BubbleMapLayer;
  14. import org.sigmah.shared.report.model.Dimension;
  15. import org.sigmah.shared.report.model.DimensionType;
  16. import org.sigmah.shared.report.model.MapElement;
  17. import java.util.ArrayList;
  18. import java.util.List;
  19. /**
  20. * @author Alex Bertram
  21. */
  22. public class PieMapMarkerTest {
  23. @Test
  24. public void testPies() {
  25. Dimension dim = new Dimension(DimensionType.Indicator);
  26. dim.setCategoryColor(101, 255);
  27. dim.setCategoryColor(102, 0x00FF00);
  28. dim.setCategoryColor(103, 0x0000FF);
  29. SiteData site1 = new SiteData();
  30. site1.setValue(SiteTableColumn.id, 1);
  31. site1.setValue(SiteTableColumn.x, 0d);
  32. site1.setValue(SiteTableColumn.y, 0d);
  33. site1.setIndicatorValue(101, 50d);
  34. site1.setIndicatorValue(102, 40d);
  35. site1.setIndicatorValue(103, 10d);
  36. List<SiteData> sites = new ArrayList<SiteData>();
  37. sites.add(site1);
  38. BubbleMapLayer layer = new BubbleMapLayer();
  39. layer.setPie(true);
  40. layer.addIndicator(101);
  41. layer.addIndicator(102);
  42. layer.addIndicator(103);
  43. layer.getColorDimensions().add(dim);
  44. MapElement mapElement = new MapElement();
  45. mapElement.addLayer(layer);
  46. MapContent content = new MapContent();
  47. TiledMap map = new TiledMap(640, 480, new LatLng(0, 0), 6);
  48. BubbleLayerGenerator gtor = new BubbleLayerGenerator(mapElement, layer);
  49. gtor.generate(sites, map, content);
  50. Assert.assertEquals(1, content.getMarkers().size());
  51. PieMapMarker marker = (PieMapMarker) content.getMarkers().get(0);
  52. Assert.assertEquals(3, marker.getSlices().size());
  53. }
  54. }