From 774b97a26b6f5558cc741c5f87944f1a523d6580 Mon Sep 17 00:00:00 2001
From: Manuel Rego Casasnovas <rego@igalia.com>
Date: Tue, 6 Nov 2012 12:42:13 +0100
Subject: [PATCH] Add test for new delete order elements service

FEA: ItEr77S13AllowDeleteOrderElements
---
 .../ws/orders/OrderElementServiceTest.java    | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/libreplan-webapp/src/test/java/org/libreplan/web/test/ws/orders/OrderElementServiceTest.java b/libreplan-webapp/src/test/java/org/libreplan/web/test/ws/orders/OrderElementServiceTest.java
index c50fc8bfe..bc4ad34c6 100644
--- a/libreplan-webapp/src/test/java/org/libreplan/web/test/ws/orders/OrderElementServiceTest.java
+++ b/libreplan-webapp/src/test/java/org/libreplan/web/test/ws/orders/OrderElementServiceTest.java
@@ -45,6 +45,8 @@ import java.util.SortedSet;
 import java.util.UUID;
 
 import javax.annotation.Resource;
+import javax.ws.rs.core.Response;
+import javax.ws.rs.core.Response.Status;
 
 import org.hibernate.SessionFactory;
 import org.joda.time.LocalDate;
@@ -536,6 +538,18 @@ public class OrderElementServiceTest {
     public void validOrderWithOrderLineGroup() {
         String code = UUID.randomUUID().toString();
 
+        OrderDTO orderDTO = createOrderDTOWithChildren(code);
+
+        OrderListDTO orderListDTO = createOrderListDTO(orderDTO);
+        List<InstanceConstraintViolationsDTO> instanceConstraintViolationsList = orderElementService
+                .addOrders(orderListDTO).instanceConstraintViolationsList;
+        assertTrue(instanceConstraintViolationsList.toString(),
+                instanceConstraintViolationsList.size() == 0);
+
+        checkIfExistsByCodeInAnotherTransaction(code);
+    }
+
+    private OrderDTO createOrderDTOWithChildren(String code) {
         OrderDTO orderDTO = new OrderDTO();
         orderDTO.name = "Order name " + UUID.randomUUID().toString();
         orderDTO.code = code;
@@ -557,6 +571,14 @@ public class OrderElementServiceTest {
         orderLineGroupDTO.children.add(orderLineDTO);
 
         orderDTO.children.add(orderLineGroupDTO);
+        return orderDTO;
+    }
+
+    @Test
+    public void removeOrderElement() {
+        String code = UUID.randomUUID().toString();
+
+        OrderDTO orderDTO = createOrderDTOWithChildren(code);
 
         OrderListDTO orderListDTO = createOrderListDTO(orderDTO);
         List<InstanceConstraintViolationsDTO> instanceConstraintViolationsList = orderElementService
@@ -565,6 +587,24 @@ public class OrderElementServiceTest {
                 instanceConstraintViolationsList.size() == 0);
 
         checkIfExistsByCodeInAnotherTransaction(code);
+
+        String codeToRemove = orderDTO.children.get(0).code;
+        Response response = orderElementService
+                .removeOrderElement(codeToRemove);
+        assertThat(response.getStatus(), equalTo(Status.OK.getStatusCode()));
+
+        try {
+            orderElementDAO.findByCode(codeToRemove);
+        } catch (InstanceNotFoundException e) {
+            assertTrue(true);
+        }
+
+        try {
+            OrderElement order = orderElementDAO.findByCode(code);
+            assertTrue(order.getChildren().isEmpty());
+        } catch (InstanceNotFoundException e) {
+            fail();
+        }
     }
 
     @Test