ordertest.java

Java EE 5 许多新功能都包含经过修补的 EJB 架构

/*
 * Copyright 2006 Borys Burnayev
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package test.domain;

import java.math.BigDecimal;
import java.util.List;

import junit.framework.TestCase;

import com.rdacorp.petstore.domain.Item;
import com.rdacorp.petstore.domain.Order;
import com.rdacorp.petstore.domain.OrderItem;
import com.rdacorp.petstore.domain.Product;

/**
 * @author Borys Burnayev
 */
public class OrderTest extends TestCase {
   public void testAddItemDifferentItems() {
      // set up data
      Order order = new Order();
      Product product1 = new Product();
      Product product2 = new Product();
      Item item1 = new Item("D0", "Dog", "", new BigDecimal(100), new Integer(1000), product1);
      Item item2 = new Item("C0", "Cat", "", new BigDecimal(90), new Integer(1000), product2);
      OrderItem orderItem1 = new OrderItem(10, order, item1);
      OrderItem orderItem2 = new OrderItem(1, order, item2);

      // execute target methods
      order.addItem(orderItem1);
      order.addItem(orderItem2);

      // verify the results
      // verify total
      assertEquals(new BigDecimal(1090), order.getAmount());
      // verify quantities
      List orderItems = order.getOrderItems();
      OrderItem updatedOrderItem1 = (OrderItem) orderItems.get(orderItems.indexOf(orderItem1));
      OrderItem updatedOrderItem2 = (OrderItem) orderItems.get(orderItems.indexOf(orderItem2));
      assertEquals(10, updatedOrderItem1.getQuantity());
      assertEquals(1, updatedOrderItem2.getQuantity());
   }

   public void testAddItemSameItem() {
      // set up data
      Order order = new Order();
      Product product1 = new Product();
      Item item1 = new Item("D0", "Dog", "", new BigDecimal(100), new Integer(1000), product1);
      OrderItem orderItem1 = new OrderItem(10, order, item1);
      OrderItem orderItem2 = new OrderItem(1, order, item1);

      // execute target methods
      order.addItem(orderItem1);
      order.addItem(orderItem2);

      // verify the results
      // verify total
      assertEquals(new BigDecimal(1100), order.getAmount());
      // verify quantities
      List orderItems = order.getOrderItems();
      assertTrue(orderItems.contains(orderItem1));
      OrderItem updatedOrderItem1 = (OrderItem) orderItems.get(orderItems.indexOf(orderItem1));
      assertEquals(11, updatedOrderItem1.getQuantity());
   }

   public void testRemoveItemDelete() {
      // set up data
      Order order = new Order();
      Product product1 = new Product();
      Item item1 = new Item("D0", "Dog", "", new BigDecimal(100), new Integer(1000), product1);
      OrderItem orderItemToAdd = new OrderItem(10, order, item1);
      OrderItem orderItemToRemove = new OrderItem(9, order, item1);

      order.addItem(orderItemToAdd);

      // execute target methods
      order.removeItem(orderItemToRemove);

      // verify the results
      // verify total
      assertEquals(new BigDecimal(100), order.getAmount());
      // verify quantities
      List orderItems = order.getOrderItems();
      OrderItem modifiedOrderItem = (OrderItem) orderItems.get(orderItems.indexOf(orderItemToAdd));
      assertEquals(1, modifiedOrderItem.getQuantity());
   }

   public void testRemoveItemUpdate() {
      // set up data
      Order order = new Order();
      Product product1 = new Product();
      Item item1 = new Item("D0", "Dog", "", new BigDecimal(100), new Integer(1000), product1);
      OrderItem orderItemToAdd = new OrderItem(10, order, item1);
      OrderItem orderItemToRemove = new OrderItem(10, order, item1);

      order.addItem(orderItemToAdd);

      // execute target methods
      order.removeItem(orderItemToRemove);

      // verify the results
      // verify total
      assertEquals(new BigDecimal(0), order.getAmount());
      // verify order items
      List orderItems = order.getOrderItems();
      assertTrue(orderItems.isEmpty());
   }
}