Polish SessionSubscriptionRegistry types
This commit is contained in:
Rossen Stoyanchev
@@ -0,0 +1,116 @@
|
||||
/*
|
||||
* Copyright 2002-2013 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.web.messaging.support;
|
||||
|
||||
import java.util.Set;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.springframework.web.messaging.SessionSubscriptionRegistration;
|
||||
import org.springframework.web.messaging.SessionSubscriptionRegistry;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
|
||||
|
||||
/**
|
||||
* A test fixture for {@link AbstractSessionSubscriptionRegistry}.
|
||||
*
|
||||
* @author Rossen Stoyanchev
|
||||
*/
|
||||
public abstract class AbstractSessionSubscriptionRegistryTests {
|
||||
|
||||
protected SessionSubscriptionRegistry registry;
|
||||
|
||||
|
||||
@Before
|
||||
public void setup() {
|
||||
this.registry = createSessionSubscriptionRegistry();
|
||||
}
|
||||
|
||||
protected abstract SessionSubscriptionRegistry createSessionSubscriptionRegistry();
|
||||
|
||||
|
||||
@Test
|
||||
public void getRegistration() {
|
||||
String sessionId = "sess1";
|
||||
assertNull(this.registry.getRegistration(sessionId));
|
||||
|
||||
this.registry.getOrCreateRegistration(sessionId);
|
||||
assertNotNull(this.registry.getRegistration(sessionId));
|
||||
assertEquals(sessionId, this.registry.getRegistration(sessionId).getSessionId());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getOrCreateRegistration() {
|
||||
String sessionId = "sess1";
|
||||
assertNull(this.registry.getRegistration(sessionId));
|
||||
|
||||
SessionSubscriptionRegistration registration = this.registry.getOrCreateRegistration(sessionId);
|
||||
assertEquals(registration, this.registry.getOrCreateRegistration(sessionId));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void removeRegistration() {
|
||||
String sessionId = "sess1";
|
||||
this.registry.getOrCreateRegistration(sessionId);
|
||||
assertNotNull(this.registry.getRegistration(sessionId));
|
||||
assertEquals(sessionId, this.registry.getRegistration(sessionId).getSessionId());
|
||||
|
||||
this.registry.removeRegistration(sessionId);
|
||||
assertNull(this.registry.getRegistration(sessionId));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getSessionSubscriptions() {
|
||||
String sessionId = "sess1";
|
||||
SessionSubscriptionRegistration registration = this.registry.getOrCreateRegistration(sessionId);
|
||||
registration.addSubscription("/foo", "sub1");
|
||||
registration.addSubscription("/foo", "sub2");
|
||||
|
||||
Set<String> subscriptions = this.registry.getSessionSubscriptions(sessionId, "/foo");
|
||||
assertEquals("Wrong number of subscriptions " + subscriptions, 2, subscriptions.size());
|
||||
assertTrue(subscriptions.contains("sub1"));
|
||||
assertTrue(subscriptions.contains("sub2"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getRegistrationsByDestination() {
|
||||
|
||||
SessionSubscriptionRegistration reg1 = this.registry.getOrCreateRegistration("sess1");
|
||||
reg1.addSubscription("/foo", "sub1");
|
||||
|
||||
SessionSubscriptionRegistration reg2 = this.registry.getOrCreateRegistration("sess2");
|
||||
reg2.addSubscription("/foo", "sub1");
|
||||
|
||||
Set<SessionSubscriptionRegistration> actual = this.registry.getRegistrationsByDestination("/foo");
|
||||
assertEquals(2, actual.size());
|
||||
assertTrue(actual.contains(reg1));
|
||||
assertTrue(actual.contains(reg2));
|
||||
|
||||
reg1.removeSubscription("sub1");
|
||||
|
||||
actual = this.registry.getRegistrationsByDestination("/foo");
|
||||
assertEquals("Invalid set of registrations " + actual, 1, actual.size());
|
||||
assertTrue(actual.contains(reg2));
|
||||
|
||||
reg2.removeSubscription("sub1");
|
||||
|
||||
actual = this.registry.getRegistrationsByDestination("/foo");
|
||||
assertNull("Unexpected registrations " + actual, actual);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -16,56 +16,20 @@
|
||||
|
||||
package org.springframework.web.messaging.support;
|
||||
|
||||
import java.util.Set;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.springframework.web.messaging.SessionSubscriptionRegistration;
|
||||
import org.springframework.web.messaging.SessionSubscriptionRegistry;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
|
||||
|
||||
/**
|
||||
* Test fixture for {@link CachingSessionSubscriptionRegistry}.
|
||||
*
|
||||
* @author Rossen Stoyanchev
|
||||
*/
|
||||
public class CachingSessionSubscriptionRegistryTests {
|
||||
|
||||
private CachingSessionSubscriptionRegistry registry;
|
||||
public class CachingSessionSubscriptionRegistryTests extends AbstractSessionSubscriptionRegistryTests {
|
||||
|
||||
|
||||
@Before
|
||||
public void setup() {
|
||||
SessionSubscriptionRegistry delegate = new DefaultSessionSubscriptionRegistry();
|
||||
this.registry = new CachingSessionSubscriptionRegistry(delegate);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getRegistrationsByDestination() {
|
||||
|
||||
SessionSubscriptionRegistration reg1 = this.registry.getOrCreateRegistration("sess1");
|
||||
reg1.addSubscription("/foo", "sub1");
|
||||
|
||||
SessionSubscriptionRegistration reg2 = this.registry.getOrCreateRegistration("sess2");
|
||||
reg2.addSubscription("/foo", "sub1");
|
||||
|
||||
Set<SessionSubscriptionRegistration> actual = this.registry.getRegistrationsByDestination("/foo");
|
||||
assertEquals(2, actual.size());
|
||||
assertTrue(actual.contains(reg1));
|
||||
assertTrue(actual.contains(reg2));
|
||||
|
||||
reg1.removeSubscription("sub1");
|
||||
|
||||
actual = this.registry.getRegistrationsByDestination("/foo");
|
||||
assertEquals("Invalid set of registrations " + actual, 1, actual.size());
|
||||
assertTrue(actual.contains(reg2));
|
||||
|
||||
reg2.removeSubscription("sub1");
|
||||
|
||||
actual = this.registry.getRegistrationsByDestination("/foo");
|
||||
assertNull("Unexpected registrations " + actual, actual);
|
||||
@Override
|
||||
protected SessionSubscriptionRegistry createSessionSubscriptionRegistry() {
|
||||
return new CachingSessionSubscriptionRegistry(new DefaultSessionSubscriptionRegistry());
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -16,13 +16,7 @@
|
||||
|
||||
package org.springframework.web.messaging.support;
|
||||
|
||||
import java.util.Set;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.springframework.web.messaging.SessionSubscriptionRegistration;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
import org.springframework.web.messaging.SessionSubscriptionRegistry;
|
||||
|
||||
|
||||
/**
|
||||
@@ -30,57 +24,12 @@ import static org.junit.Assert.*;
|
||||
*
|
||||
* @author Rossen Stoyanchev
|
||||
*/
|
||||
public class DefaultSessionSubscriptionRegistryTests {
|
||||
|
||||
private DefaultSessionSubscriptionRegistry registry;
|
||||
public class DefaultSessionSubscriptionRegistryTests extends AbstractSessionSubscriptionRegistryTests {
|
||||
|
||||
|
||||
@Before
|
||||
public void setup() {
|
||||
this.registry = new DefaultSessionSubscriptionRegistry();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getRegistration() {
|
||||
String sessionId = "sess1";
|
||||
assertNull(this.registry.getRegistration(sessionId));
|
||||
|
||||
this.registry.getOrCreateRegistration(sessionId);
|
||||
assertNotNull(this.registry.getRegistration(sessionId));
|
||||
assertEquals(sessionId, this.registry.getRegistration(sessionId).getSessionId());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getOrCreateRegistration() {
|
||||
String sessionId = "sess1";
|
||||
assertNull(this.registry.getRegistration(sessionId));
|
||||
|
||||
SessionSubscriptionRegistration registration = this.registry.getOrCreateRegistration(sessionId);
|
||||
assertSame(registration, this.registry.getOrCreateRegistration(sessionId));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void removeRegistration() {
|
||||
String sessionId = "sess1";
|
||||
this.registry.getOrCreateRegistration(sessionId);
|
||||
assertNotNull(this.registry.getRegistration(sessionId));
|
||||
assertEquals(sessionId, this.registry.getRegistration(sessionId).getSessionId());
|
||||
|
||||
this.registry.removeRegistration(sessionId);
|
||||
assertNull(this.registry.getRegistration(sessionId));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getSessionSubscriptions() {
|
||||
String sessionId = "sess1";
|
||||
SessionSubscriptionRegistration registration = this.registry.getOrCreateRegistration(sessionId);
|
||||
registration.addSubscription("/foo", "sub1");
|
||||
registration.addSubscription("/foo", "sub2");
|
||||
|
||||
Set<String> subscriptions = this.registry.getSessionSubscriptions(sessionId, "/foo");
|
||||
assertEquals("Wrong number of subscriptions " + subscriptions, 2, subscriptions.size());
|
||||
assertTrue(subscriptions.contains("sub1"));
|
||||
assertTrue(subscriptions.contains("sub2"));
|
||||
@Override
|
||||
protected SessionSubscriptionRegistry createSessionSubscriptionRegistry() {
|
||||
return new DefaultSessionSubscriptionRegistry();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user