@Test // SPR-13185 public void findSubscriptionsReturnsMapSafeToIterateIncludingValues() throws Exception { this.registry.registerSubscription(subscribeMessage("sess1", "1", "/foo")); this.registry.registerSubscription(subscribeMessage("sess1", "2", "/foo")); MultiValueMap<String, String> allSubscriptions = this.registry.findSubscriptions(createMessage("/foo")); assertNotNull(allSubscriptions); assertEquals(1, allSubscriptions.size()); Iterator<String> iteratorValues = allSubscriptions.get("sess1").iterator(); iteratorValues.next(); this.registry.unregisterSubscription(unsubscribeMessage("sess1", "2")); iteratorValues.next(); // no ConcurrentModificationException }
@Test public void unregisterSubscription() { List<String> sessIds = Arrays.asList("sess01", "sess02", "sess03"); List<String> subscriptionIds = Arrays.asList("subs01", "subs02", "subs03"); String dest = "/foo"; for (String sessId : sessIds) { for (String subsId : subscriptionIds) { this.registry.registerSubscription(subscribeMessage(sessId, subsId, dest)); } } this.registry.unregisterSubscription(unsubscribeMessage(sessIds.get(0), subscriptionIds.get(0))); this.registry.unregisterSubscription(unsubscribeMessage(sessIds.get(0), subscriptionIds.get(1))); this.registry.unregisterSubscription(unsubscribeMessage(sessIds.get(0), subscriptionIds.get(2))); MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals("Expected two elements: " + actual, 2, actual.size()); assertEquals(subscriptionIds, sort(actual.get(sessIds.get(1)))); assertEquals(subscriptionIds, sort(actual.get(sessIds.get(2)))); }
@Test // SPR-12665 public void findSubscriptionsReturnsMapSafeToIterate() throws Exception { this.registry.registerSubscription(subscribeMessage("sess1", "1", "/foo")); this.registry.registerSubscription(subscribeMessage("sess2", "1", "/foo")); MultiValueMap<String, String> subscriptions = this.registry.findSubscriptions(createMessage("/foo")); assertNotNull(subscriptions); assertEquals(2, subscriptions.size()); Iterator<Map.Entry<String, List<String>>> iterator = subscriptions.entrySet().iterator(); iterator.next(); this.registry.registerSubscription(subscribeMessage("sess3", "1", "/foo")); iterator.next(); // no ConcurrentModificationException }
@Test public void registerSubscriptionMultipleSessions() { List<String> sessIds = Arrays.asList("sess01", "sess02", "sess03"); List<String> subscriptionIds = Arrays.asList("subs01", "subs02", "subs03"); String dest = "/foo"; for (String sessId : sessIds) { for (String subsId : subscriptionIds) { this.registry.registerSubscription(subscribeMessage(sessId, subsId, dest)); } } MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals(3, actual.size()); assertEquals(subscriptionIds, sort(actual.get(sessIds.get(0)))); assertEquals(subscriptionIds, sort(actual.get(sessIds.get(1)))); assertEquals(subscriptionIds, sort(actual.get(sessIds.get(2)))); }
private Message<?> subscribeMessage(String sessionId, String subscriptionId, String destination) { return subscribeMessage(sessionId, subscriptionId, destination, null); }
@Test public void findSubscriptionsNoMatches() { MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage("/foo")); assertNotNull(actual); assertEquals("Expected no elements " + actual, 0, actual.size()); }
@Test // SPR-13555 public void cacheLimitExceeded() throws Exception { this.registry.setCacheLimit(1); this.registry.registerSubscription(subscribeMessage("sess1", "1", "/foo")); this.registry.registerSubscription(subscribeMessage("sess1", "2", "/bar")); assertEquals(1, this.registry.findSubscriptions(createMessage("/foo")).size()); assertEquals(1, this.registry.findSubscriptions(createMessage("/bar")).size()); this.registry.registerSubscription(subscribeMessage("sess2", "1", "/foo")); this.registry.registerSubscription(subscribeMessage("sess2", "2", "/bar")); assertEquals(2, this.registry.findSubscriptions(createMessage("/foo")).size()); assertEquals(2, this.registry.findSubscriptions(createMessage("/bar")).size()); }
@Test public void registerSubscriptionOneSession() { String sessId = "sess01"; List<String> subscriptionIds = Arrays.asList("subs01", "subs02", "subs03"); String dest = "/foo"; for (String subId : subscriptionIds) { this.registry.registerSubscription(subscribeMessage(sessId, subId, dest)); } MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals(1, actual.size()); assertEquals(subscriptionIds, sort(actual.get(sessId))); }
@Test public void registerSubscriptionWithSelectorNotSupported() { String sessionId = "sess01"; String subscriptionId = "subs01"; String destination = "/foo"; String selector = "headers.foo == 'bar'"; this.registry.setSelectorHeaderName(null); this.registry.registerSubscription(subscribeMessage(sessionId, subscriptionId, destination, selector)); SimpMessageHeaderAccessor accessor = SimpMessageHeaderAccessor.create(); accessor.setDestination(destination); accessor.setNativeHeader("foo", "bazz"); Message<?> message = MessageBuilder.createMessage("", accessor.getMessageHeaders()); MultiValueMap<String, String> actual = this.registry.findSubscriptions(message); assertNotNull(actual); assertEquals(1, actual.size()); assertEquals(Collections.singletonList(subscriptionId), actual.get(sessionId)); }
@Test // SPR-11755 public void registerAndUnregisterMultipleDestinations() { String sess1 = "sess01"; String sess2 = "sess02"; String subs1 = "subs01"; String subs2 = "subs02"; String subs3 = "subs03"; String subs4 = "subs04"; String subs5 = "subs05"; this.registry.registerSubscription(subscribeMessage(sess1, subs1, "/topic/PRICE.STOCK.NASDAQ.IBM")); this.registry.registerSubscription(subscribeMessage(sess1, subs2, "/topic/PRICE.STOCK.NYSE.IBM")); this.registry.registerSubscription(subscribeMessage(sess1, subs3, "/topic/PRICE.STOCK.NASDAQ.GOOG")); this.registry.findSubscriptions(createMessage("/topic/PRICE.STOCK.NYSE.IBM")); this.registry.findSubscriptions(createMessage("/topic/PRICE.STOCK.NASDAQ.GOOG")); this.registry.findSubscriptions(createMessage("/topic/PRICE.STOCK.NASDAQ.IBM")); this.registry.unregisterSubscription(unsubscribeMessage(sess1, subs1)); this.registry.unregisterSubscription(unsubscribeMessage(sess1, subs2)); this.registry.unregisterSubscription(unsubscribeMessage(sess1, subs3)); this.registry.registerSubscription(subscribeMessage(sess1, subs1, "/topic/PRICE.STOCK.NASDAQ.IBM")); this.registry.registerSubscription(subscribeMessage(sess1, subs2, "/topic/PRICE.STOCK.NYSE.IBM")); this.registry.registerSubscription(subscribeMessage(sess1, subs3, "/topic/PRICE.STOCK.NASDAQ.GOOG")); this.registry.registerSubscription(subscribeMessage(sess1, subs4, "/topic/PRICE.STOCK.NYSE.IBM")); this.registry.registerSubscription(subscribeMessage(sess2, subs5, "/topic/PRICE.STOCK.NASDAQ.GOOG")); this.registry.unregisterAllSubscriptions(sess1); this.registry.unregisterAllSubscriptions(sess2); }
@Test public void registerSubscription() { String sessId = "sess01"; String subsId = "subs01"; String dest = "/foo"; this.registry.registerSubscription(subscribeMessage(sessId, subsId, dest)); MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals("Expected one element " + actual, 1, actual.size()); assertEquals(Collections.singletonList(subsId), actual.get(sessId)); }
@Test public void unregisterAllSubscriptions() { List<String> sessIds = Arrays.asList("sess01", "sess02", "sess03"); List<String> subscriptionIds = Arrays.asList("subs01", "subs02", "subs03"); String dest = "/foo"; for (String sessId : sessIds) { for (String subsId : subscriptionIds) { this.registry.registerSubscription(subscribeMessage(sessId, subsId, dest)); } } this.registry.unregisterAllSubscriptions(sessIds.get(0)); this.registry.unregisterAllSubscriptions(sessIds.get(1)); MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals("Expected one element: " + actual, 1, actual.size()); assertEquals(subscriptionIds, sort(actual.get(sessIds.get(2)))); }
@Test // SPR-11931 public void registerSubscriptionTwiceAndUnregister() { this.registry.registerSubscription(subscribeMessage("sess01", "subs01", "/foo")); this.registry.registerSubscription(subscribeMessage("sess01", "subs02", "/foo")); MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage("/foo")); assertNotNull(actual); assertEquals("Expected 1 element", 1, actual.size()); assertEquals(Arrays.asList("subs01", "subs02"), actual.get("sess01")); this.registry.unregisterSubscription(unsubscribeMessage("sess01", "subs01")); actual = this.registry.findSubscriptions(createMessage("/foo")); assertNotNull(actual); assertEquals("Expected 1 element", 1, actual.size()); assertEquals(Collections.singletonList("subs02"), actual.get("sess01")); this.registry.unregisterSubscription(unsubscribeMessage("sess01", "subs02")); actual = this.registry.findSubscriptions(createMessage("/foo")); assertNotNull(actual); assertEquals("Expected no element", 0, actual.size()); }
@Test public void registerSubscriptionInvalidInput() { String sessId = "sess01"; String subsId = "subs01"; String dest = "/foo"; this.registry.registerSubscription(subscribeMessage(null, subsId, dest)); MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals(0, actual.size()); this.registry.registerSubscription(subscribeMessage(sessId, null, dest)); actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals(0, actual.size()); this.registry.registerSubscription(subscribeMessage(sessId, subsId, null)); actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals(0, actual.size()); }
Message<?> destNasdaqIbmMessage = createMessage(destNasdaqIbm); this.registry.registerSubscription(subscribeMessage(sess1, subs2, destNasdaqIbm)); this.registry.registerSubscription(subscribeMessage(sess1, subs1, "/topic/PRICE.STOCK.*.IBM")); assertEquals(Arrays.asList(subs2, subs1), actual.get(sess1)); this.registry.registerSubscription(subscribeMessage(sess2, subs1, destNasdaqIbm)); this.registry.registerSubscription(subscribeMessage(sess2, subs2, "/topic/PRICE.STOCK.NYSE.IBM")); this.registry.registerSubscription(subscribeMessage(sess2, subs3, "/topic/PRICE.STOCK.NASDAQ.GOOG")); assertEquals(Collections.singletonList(subs1), actual.get(sess2)); this.registry.registerSubscription(subscribeMessage(sess1, subs1, "/topic/PRICE.STOCK.*.IBM")); this.registry.registerSubscription(subscribeMessage(sess1, subs2, destNasdaqIbm)); assertEquals(Collections.singletonList(subs1), actual.get(sess2)); this.registry.unregisterSubscription(unsubscribeMessage(sess1, subs2)); assertEquals(Collections.singletonList(subs1), actual.get(sess2)); this.registry.unregisterSubscription(unsubscribeMessage(sess1, subs1)); assertEquals(Collections.singletonList(subs1), actual.get(sess2)); this.registry.unregisterSubscription(unsubscribeMessage(sess2, subs1));
@Test public void registerSubscriptionWithDestinationPattern() { String sessId = "sess01"; String subsId = "subs01"; String destPattern = "/topic/PRICE.STOCK.*.IBM"; String dest = "/topic/PRICE.STOCK.NASDAQ.IBM"; this.registry.registerSubscription(subscribeMessage(sessId, subsId, destPattern)); MultiValueMap<String, String> actual = this.registry.findSubscriptions(createMessage(dest)); assertNotNull(actual); assertEquals("Expected one element " + actual, 1, actual.size()); assertEquals(Collections.singletonList(subsId), actual.get(sessId)); }
@Test public void registerSubscriptionWithDestinationPatternRegex() { String sessId = "sess01"; String subsId = "subs01"; String destPattern = "/topic/PRICE.STOCK.*.{ticker:(IBM|MSFT)}"; this.registry.registerSubscription(subscribeMessage(sessId, subsId, destPattern)); Message<?> message = createMessage("/topic/PRICE.STOCK.NASDAQ.IBM"); MultiValueMap<String, String> actual = this.registry.findSubscriptions(message); assertNotNull(actual); assertEquals("Expected one element " + actual, 1, actual.size()); assertEquals(Collections.singletonList(subsId), actual.get(sessId)); message = createMessage("/topic/PRICE.STOCK.NASDAQ.MSFT"); actual = this.registry.findSubscriptions(message); assertNotNull(actual); assertEquals("Expected one element " + actual, 1, actual.size()); assertEquals(Collections.singletonList(subsId), actual.get(sessId)); message = createMessage("/topic/PRICE.STOCK.NASDAQ.VMW"); actual = this.registry.findSubscriptions(message); assertNotNull(actual); assertEquals("Expected no elements " + actual, 0, actual.size()); }
@Test public void registerSubscriptionWithSelector() { String sessionId = "sess01"; String subscriptionId = "subs01"; String destination = "/foo"; String selector = "headers.foo == 'bar'"; this.registry.registerSubscription(subscribeMessage(sessionId, subscriptionId, destination, selector)); // First, try with selector header SimpMessageHeaderAccessor accessor = SimpMessageHeaderAccessor.create(); accessor.setDestination(destination); accessor.setNativeHeader("foo", "bar"); Message<?> message = MessageBuilder.createMessage("", accessor.getMessageHeaders()); MultiValueMap<String, String> actual = this.registry.findSubscriptions(message); assertNotNull(actual); assertEquals(1, actual.size()); assertEquals(Collections.singletonList(subscriptionId), actual.get(sessionId)); // Then without actual = this.registry.findSubscriptions(createMessage(destination)); assertNotNull(actual); assertEquals(0, actual.size()); }