public void testGetValuesWithStateFromCollection() { Collection<DirtyWrapper<Integer>> wrapped = new ArrayList<DirtyWrapper<Integer>>(); wrapped.add(new DirtyWrapper<Integer>(1, DirtyState.ADDED)); wrapped.add(new DirtyWrapper<Integer>(2, DirtyState.ADDED)); wrapped.add(new DirtyWrapper<Integer>(3, DirtyState.REMOVED)); wrapped.add(new DirtyWrapper<Integer>(4, DirtyState.MODIFIED)); wrapped.add(new DirtyWrapper<Integer>(5, DirtyState.MODIFIED)); Collection<Integer> addedOnly = DirtyUtils.getValuesWithState(wrapped, DirtyState.ADDED); Collection<Integer> removedOnly = DirtyUtils.getValuesWithState(wrapped, DirtyState.REMOVED); Collection<Integer> modifiedOnly = DirtyUtils.getValuesWithState(wrapped, DirtyState.MODIFIED); Collection<Integer> unmodifiedOnly = DirtyUtils.getValuesWithState(wrapped, DirtyState.UNMODIFIED); assertNotNull(addedOnly); assertEquals(addedOnly.size(), 2); assertTrue(addedOnly.contains(1)); assertTrue(addedOnly.contains(2)); assertNotNull(removedOnly); assertEquals(removedOnly.size(), 1); assertTrue(removedOnly.contains(3)); assertNotNull(modifiedOnly); assertEquals(modifiedOnly.size(), 2); assertTrue(modifiedOnly.contains(4)); assertTrue(modifiedOnly.contains(5)); assertNotNull(unmodifiedOnly); assertEquals(unmodifiedOnly.size(), 0); }
public void testGetValuesWithStateFromMap() { Map<Integer, DirtyState> wrapped = new HashMap<Integer, DirtyState>(); wrapped.put(1, DirtyState.ADDED); wrapped.put(2, DirtyState.ADDED); wrapped.put(3, DirtyState.REMOVED); wrapped.put(4, DirtyState.MODIFIED); wrapped.put(5, DirtyState.MODIFIED); Collection<Integer> addedOnly = DirtyUtils.getValuesWithState(wrapped, DirtyState.ADDED); Collection<Integer> removedOnly = DirtyUtils.getValuesWithState(wrapped, DirtyState.REMOVED); Collection<Integer> modifiedOnly = DirtyUtils.getValuesWithState(wrapped, DirtyState.MODIFIED); Collection<Integer> unmodifiedOnly = DirtyUtils.getValuesWithState(wrapped, DirtyState.UNMODIFIED); assertNotNull(addedOnly); assertEquals(addedOnly.size(), 2); assertTrue(addedOnly.contains(1)); assertTrue(addedOnly.contains(2)); assertNotNull(removedOnly); assertEquals(removedOnly.size(), 1); assertTrue(removedOnly.contains(3)); assertNotNull(modifiedOnly); assertEquals(modifiedOnly.size(), 2); assertTrue(modifiedOnly.contains(4)); assertTrue(modifiedOnly.contains(5)); assertNotNull(unmodifiedOnly); assertEquals(unmodifiedOnly.size(), 0); }