@SuppressWarnings({"resource", "unchecked", "rawtypes"}) @Test public void updateStreamWithListeners() { Repository decoratedRepository = mock(Repository.class); EntityListenerRepositoryDecorator entityListenerRepositoryDecorator = new EntityListenerRepositoryDecorator(decoratedRepository); EntityListener entityListener0 = when(mock(EntityListener.class).getEntityId()).thenReturn(Integer.valueOf(1)).getMock(); EntityListener entityListener1 = when(mock(EntityListener.class).getEntityId()).thenReturn(Integer.valueOf(2)).getMock(); entityListenerRepositoryDecorator.addEntityListener(entityListener0); entityListenerRepositoryDecorator.addEntityListener(entityListener1); Entity entity0 = when(mock(Entity.class).getIdValue()).thenReturn(Integer.valueOf(1)).getMock(); Entity entity1 = when(mock(Entity.class).getIdValue()).thenReturn(Integer.valueOf(2)).getMock(); Stream<Entity> entities = Stream.of(entity0, entity1); entityListenerRepositoryDecorator.update(entities); ArgumentCaptor<Stream<Entity>> captor = ArgumentCaptor.forClass((Class) Stream.class); verify(decoratedRepository).update(captor.capture()); assertEquals(captor.getValue().collect(Collectors.toList()), Arrays.asList(entity0, entity1)); verify(entityListener0, times(1)).postUpdate(entity0); verify(entityListener1, times(1)).postUpdate(entity1); }
@SuppressWarnings("resource") @Test public void updateEntityWithListeners() { @SuppressWarnings("unchecked") Repository<Entity> decoratedRepository = Mockito.mock(Repository.class); Mockito.when(decoratedRepository.getName()).thenReturn("entityFullName"); EntityListenerRepositoryDecorator entityListenerRepositoryDecorator = new EntityListenerRepositoryDecorator(decoratedRepository, entityListenersService); EntityListener entityListener0 = Mockito.when(Mockito.mock(EntityListener.class).getEntityId()) .thenReturn(Integer.valueOf(1)) .getMock(); EntityListener entityListener1 = Mockito.when(Mockito.mock(EntityListener.class).getEntityId()) .thenReturn(Integer.valueOf(1)) .getMock(); entityListenersService.addEntityListener("entityFullName", entityListener0); entityListenersService.addEntityListener("entityFullName", entityListener1); Entity entity = Mockito.when(Mockito.mock(Entity.class).getIdValue()) .thenReturn(Integer.valueOf(1)) .getMock(); entityListenerRepositoryDecorator.update(entity); Mockito.verify(decoratedRepository).update(entity); Mockito.verify(entityListener0, Mockito.times(1)).postUpdate(entity); Mockito.verify(entityListener1, Mockito.times(1)).postUpdate(entity); }
@SuppressWarnings({"resource", "unchecked", "rawtypes"}) @Test public void updateStreamWithSomeListeners() { Repository<Entity> decoratedRepository = Mockito.mock(Repository.class); Mockito.when(decoratedRepository.getName()).thenReturn("entityFullName"); EntityListenerRepositoryDecorator entityListenerRepositoryDecorator = new EntityListenerRepositoryDecorator(decoratedRepository, entityListenersService); EntityListener entityListener1 = Mockito.when(Mockito.mock(EntityListener.class).getEntityId()) .thenReturn(Integer.valueOf(2)) .getMock(); entityListenersService.addEntityListener("entityFullName", entityListener1); Entity entity0 = Mockito.when(Mockito.mock(Entity.class).getIdValue()) .thenReturn(Integer.valueOf(1)) .getMock(); Entity entity1 = Mockito.when(Mockito.mock(Entity.class).getIdValue()) .thenReturn(Integer.valueOf(2)) .getMock(); Stream<Entity> entities = Stream.of(entity0, entity1); entityListenerRepositoryDecorator.update(entities); ArgumentCaptor<Stream<Entity>> captor = ArgumentCaptor.forClass((Class) Stream.class); Mockito.verify(decoratedRepository).update(captor.capture()); Assert.assertEquals( captor.getValue().collect(Collectors.toList()), Arrays.asList(entity0, entity1)); Mockito.verify(entityListener1, Mockito.times(1)).postUpdate(entity1); }
@SuppressWarnings("resource") @Test public void updateEntityNoListeners() { Repository decoratedRepository = mock(Repository.class); EntityListenerRepositoryDecorator entityListenerRepositoryDecorator = new EntityListenerRepositoryDecorator(decoratedRepository); Entity entity = when(mock(Entity.class).getIdValue()).thenReturn(Integer.valueOf(1)).getMock(); entityListenerRepositoryDecorator.update(entity); verify(decoratedRepository).update(entity); }
@SuppressWarnings("resource") @Test public void updateEntityWithoutListener() { Repository decoratedRepository = mock(Repository.class); EntityListenerRepositoryDecorator entityListenerRepositoryDecorator = new EntityListenerRepositoryDecorator(decoratedRepository); EntityListener entityListener0 = when(mock(EntityListener.class).getEntityId()).thenReturn(Integer.valueOf(-1)).getMock(); entityListenerRepositoryDecorator.addEntityListener(entityListener0); Entity entity = when(mock(Entity.class).getIdValue()).thenReturn(Integer.valueOf(1)).getMock(); entityListenerRepositoryDecorator.update(entity); verify(decoratedRepository).update(entity); verify(entityListener0, times(0)).postUpdate(entity); }