5 changed files with 172 additions and 5 deletions
@ -0,0 +1,57 @@
@@ -0,0 +1,57 @@
|
||||
/* |
||||
* Copyright 2002-2014 the original author or authors. |
||||
* |
||||
* 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 org.springframework.security.context; |
||||
|
||||
import org.springframework.context.ApplicationEvent; |
||||
import org.springframework.context.ApplicationListener; |
||||
import org.springframework.context.event.SmartApplicationListener; |
||||
import org.springframework.util.Assert; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
/** |
||||
* Used for delegating to a number of SmartApplicationListener instances. This is useful when needing to register an |
||||
* SmartApplicationListener with the ApplicationContext programmatically. |
||||
* |
||||
* @author Rob Winch |
||||
*/ |
||||
public final class DelegatingApplicationListener implements ApplicationListener<ApplicationEvent> { |
||||
private List<SmartApplicationListener> listeners = new ArrayList<SmartApplicationListener>(); |
||||
|
||||
@Override |
||||
public void onApplicationEvent(ApplicationEvent event) { |
||||
if(event == null) { |
||||
return; |
||||
} |
||||
for(SmartApplicationListener listener : listeners) { |
||||
Object source = event.getSource(); |
||||
if(source != null && listener.supportsEventType(event.getClass()) && listener.supportsSourceType(source.getClass())) { |
||||
listener.onApplicationEvent(event); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Adds a new SmartApplicationListener to use. |
||||
* |
||||
* @param smartApplicationListener the SmartApplicationListener to use. Cannot be null. |
||||
*/ |
||||
public void addListener(SmartApplicationListener smartApplicationListener) { |
||||
Assert.notNull(smartApplicationListener, "smartApplicationListener cannot be null"); |
||||
listeners.add(smartApplicationListener); |
||||
} |
||||
} |
||||
@ -0,0 +1,72 @@
@@ -0,0 +1,72 @@
|
||||
package org.springframework.security.context; |
||||
|
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.mockito.Mock; |
||||
import org.mockito.runners.MockitoJUnitRunner; |
||||
import org.springframework.context.ApplicationEvent; |
||||
import org.springframework.context.ApplicationListener; |
||||
import org.springframework.context.event.SmartApplicationListener; |
||||
import org.springframework.security.core.session.SessionDestroyedEvent; |
||||
|
||||
import static org.junit.Assert.*; |
||||
import static org.mockito.Matchers.any; |
||||
import static org.mockito.Mockito.never; |
||||
import static org.mockito.Mockito.verify; |
||||
import static org.mockito.Mockito.when; |
||||
|
||||
@RunWith(MockitoJUnitRunner.class) |
||||
public class DelegatingApplicationListenerTests { |
||||
@Mock |
||||
SmartApplicationListener delegate; |
||||
|
||||
ApplicationEvent event; |
||||
|
||||
DelegatingApplicationListener listener; |
||||
|
||||
@Before |
||||
public void setup() { |
||||
event = new ApplicationEvent(this) {}; |
||||
listener = new DelegatingApplicationListener(); |
||||
listener.addListener(delegate); |
||||
} |
||||
|
||||
@Test |
||||
public void processEventNull() { |
||||
listener.onApplicationEvent(null); |
||||
|
||||
verify(delegate,never()).onApplicationEvent(any(ApplicationEvent.class)); |
||||
} |
||||
|
||||
@Test |
||||
public void processEventSuccess() { |
||||
when(delegate.supportsEventType(event.getClass())).thenReturn(true); |
||||
when(delegate.supportsSourceType(event.getSource().getClass())).thenReturn(true); |
||||
listener.onApplicationEvent(event); |
||||
|
||||
verify(delegate).onApplicationEvent(event); |
||||
} |
||||
|
||||
@Test |
||||
public void processEventEventTypeNotSupported() { |
||||
when(delegate.supportsSourceType(event.getSource().getClass())).thenReturn(true); |
||||
listener.onApplicationEvent(event); |
||||
|
||||
verify(delegate,never()).onApplicationEvent(any(ApplicationEvent.class)); |
||||
} |
||||
|
||||
@Test |
||||
public void processEventSourceTypeNotSupported() { |
||||
when(delegate.supportsEventType(event.getClass())).thenReturn(true); |
||||
listener.onApplicationEvent(event); |
||||
|
||||
verify(delegate,never()).onApplicationEvent(any(ApplicationEvent.class)); |
||||
} |
||||
|
||||
@Test(expected = IllegalArgumentException.class) |
||||
public void addNull() { |
||||
listener.addListener(null); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue