Added unit test for CargoRoutingDTOAssembler.

dddsample/src/test/java/se/citerus/dddsample/service/dto/assembler/CargoRoutingDTOAssemblerTest.java

@@ -0,0 +1,55 @@
+package se.citerus.dddsample.service.dto.assembler;
+
+import junit.framework.TestCase;
+import se.citerus.dddsample.domain.*;
+import static se.citerus.dddsample.domain.SampleLocations.*;
+import se.citerus.dddsample.service.dto.CargoRoutingDTO;
+import se.citerus.dddsample.service.dto.LegDTO;
+import se.citerus.dddsample.service.dto.assembler.CargoRoutingDTOAssembler;
+
+public class CargoRoutingDTOAssemblerTest extends TestCase {
+
+  public void testToDTO() throws Exception {
+    final CargoRoutingDTOAssembler assembler = new CargoRoutingDTOAssembler();
+
+    final Location origin = STOCKHOLM;
+    final Location destination = MELBOURNE;
+    final Cargo cargo = new Cargo(new TrackingId("XYZ"), origin, destination);
+
+    final CarrierMovement cm = new CarrierMovement(
+      new CarrierMovementId("ABC"), origin, destination);
+
+    final Itinerary itinerary = new Itinerary(
+      new Leg(cm, origin, SHANGHAI),
+      new Leg(cm, ROTTERDAM, destination)
+    );
+
+    cargo.setItinerary(itinerary);
+
+    final CargoRoutingDTO dto = assembler.toDTO(cargo);
+
+    assertEquals(2, dto.getLegs().size());
+
+    LegDTO legDTO = dto.getLegs().get(0);
+    assertEquals("ABC", legDTO.getCarrierMovementId());
+    assertEquals("SESTO (Stockholm)", legDTO.getFrom());
+    assertEquals("CNSHA (Shanghai)", legDTO.getTo());
+
+    legDTO = dto.getLegs().get(1);
+    assertEquals("ABC", legDTO.getCarrierMovementId());
+    assertEquals("NLRTM (Rotterdam)", legDTO.getFrom());
+    assertEquals("AUMEL (Melbourne)", legDTO.getTo());
+  }
+
+  public void testToDTO_NoItinerary() throws Exception {
+    final CargoRoutingDTOAssembler assembler = new CargoRoutingDTOAssembler();
+
+    final Cargo cargo = new Cargo(new TrackingId("XYZ"), STOCKHOLM, MELBOURNE);
+    final CargoRoutingDTO dto = assembler.toDTO(cargo);
+
+    assertEquals("XYZ", dto.getTrackingId());
+    assertEquals("SESTO (Stockholm)", dto.getOrigin());
+    assertEquals("AUMEL (Melbourne)", dto.getFinalDestination());
+    assertTrue(dto.getLegs().isEmpty());
+  }
+}