Added an AppListener to collect events for use in tests

2010-09-13 14:20:21 +01:00
parent 62cbd51d54
commit 0217e98bdb
3 changed files with 65 additions and 1 deletions
--- a/config/src/test/java/org/springframework/security/CollectingAppListener.java
+++ b/config/src/test/java/org/springframework/security/CollectingAppListener.java
@@ -0,0 +1,53 @@
+package org.springframework.security;
+
+import java.util.*;
+
+import org.springframework.context.ApplicationEvent;
+import org.springframework.context.ApplicationListener;
+import org.springframework.security.access.event.AbstractAuthorizationEvent;
+import org.springframework.security.authentication.event.AbstractAuthenticationEvent;
+import org.springframework.security.authentication.event.AbstractAuthenticationFailureEvent;
+
+/**
+ * ApplicationListener which collects events for use in test assertions
+ *
+ * @author Luke Taylor
+ * @since 3.1
+ */
+public class CollectingAppListener implements ApplicationListener {
+    Set<ApplicationEvent> events = new HashSet<ApplicationEvent>();
+    Set<AbstractAuthenticationEvent> authenticationEvents = new HashSet<AbstractAuthenticationEvent>();
+    Set<AbstractAuthenticationFailureEvent> authenticationFailureEvents = new HashSet<AbstractAuthenticationFailureEvent>();
+    Set<AbstractAuthorizationEvent> authorizationEvents = new HashSet<AbstractAuthorizationEvent>();
+
+    public void onApplicationEvent(ApplicationEvent event) {
+        if (event instanceof AbstractAuthenticationEvent) {
+            events.add(event);
+            authenticationEvents.add((AbstractAuthenticationEvent) event);
+        }
+        if (event instanceof AbstractAuthenticationFailureEvent) {
+            events.add(event);
+            authenticationFailureEvents.add((AbstractAuthenticationFailureEvent) event);
+        }
+        if (event instanceof AbstractAuthorizationEvent) {
+            events.add(event);
+            authorizationEvents.add((AbstractAuthorizationEvent) event);
+        }
+    }
+
+    public Set<ApplicationEvent> getEvents() {
+        return events;
+    }
+
+    public Set<AbstractAuthenticationEvent> getAuthenticationEvents() {
+        return authenticationEvents;
+    }
+
+    public Set<AbstractAuthenticationFailureEvent> getAuthenticationFailureEvents() {
+        return authenticationFailureEvents;
+    }
+
+    public Set<AbstractAuthorizationEvent> getAuthorizationEvents() {
+        return authorizationEvents;
+    }
+}