// Changed to work with arraylist public void pickUp(Item i) { if (itemCount < 2) { myItems.add(i); itemCount++; myLoc.removeItem(i); } }
@Test public void testRemoveItem() throws Exception { // setup Room testRemoveItem = new Room("test"); DirtyDiaper diaperInRoom = new DirtyDiaper("somekid"); DirtyDiaper diaperInRoom2 = new DirtyDiaper("somekid2"); testRemoveItem.addItem(diaperInRoom); testRemoveItem.addItem(diaperInRoom2); ArrayList listOfItemsInRoom = testRemoveItem.getItemsInRoom(); // does the diaper exist in the room right now? assertTrue(listOfItemsInRoom.contains(diaperInRoom)); assertTrue(listOfItemsInRoom.contains(diaperInRoom2)); // remove items from the room testRemoveItem.removeItem(diaperInRoom2); // does diaperInRoom2 exist in the room now? assertFalse(listOfItemsInRoom.contains(diaperInRoom2)); }