@Test public void testOnlyOptional() { config.put("optional.foo", "ch1 ch2"); config.put("optional.zebra", "ch2 ch3"); selector = ChannelSelectorFactory.create(channels, config); Assert.assertTrue(selector instanceof MultiplexingChannelSelector); Event event1 = new MockEvent(); Map<String, String> header1 = new HashMap<String, String>(); header1.put("myheader", "foo"); // should match ch1 ch2 event1.setHeaders(header1); List<Channel> reqCh1 = selector.getRequiredChannels(event1); Assert.assertTrue(reqCh1.isEmpty()); List<Channel> optCh1 = selector.getOptionalChannels(event1); Assert.assertEquals(2, optCh1.size()); // ch2 should not be there -- since it is a required channel Event event4 = new MockEvent(); Map<String, String> header4 = new HashMap<String, String>(); header4.put("myheader", "zebra"); event4.setHeaders(header4); List<Channel> reqCh4 = selector.getRequiredChannels(event4); Assert.assertTrue(reqCh4.isEmpty()); List<Channel> optCh4 = selector.getOptionalChannels(event4); Assert.assertEquals(2, optCh4.size()); Assert.assertEquals("ch2", optCh4.get(0).getName()); Assert.assertEquals("ch3", optCh4.get(1).getName()); }
@Test public void testSelection() throws Exception { config.put("mapping.foo", "ch1 ch2"); config.put("mapping.bar", "ch2 ch3"); config.put("mapping.xyz", "ch1 ch2 ch3"); config.put("default", "ch1 ch3"); selector = ChannelSelectorFactory.create(channels, config); Assert.assertTrue(selector instanceof MultiplexingChannelSelector); Event event1 = new MockEvent(); Map<String, String> header1 = new HashMap<String, String>(); header1.put("myheader", "foo"); // should match ch1 ch2 event1.setHeaders(header1); List<Channel> reqCh1 = selector.getRequiredChannels(event1); Assert.assertEquals(2, reqCh1.size()); Assert.assertTrue(reqCh1.get(0).getName().equals("ch1")); Assert.assertTrue(reqCh1.get(1).getName().equals("ch2")); List<Channel> optCh1 = selector.getOptionalChannels(event1); Assert.assertTrue(optCh1.size() == 1); // ch2 should not be there -- since it is a required channel Assert.assertTrue(optCh1.get(0).getName().equals("ch3")); Event event2 = new MockEvent(); Map<String, String> header2 = new HashMap<String, String>(); header2.put("myheader", "bar"); // should match ch2 ch3 event2.setHeaders(header2); List<Channel> reqCh2 = selector.getRequiredChannels(event2); Assert.assertEquals(2, reqCh2.size()); Assert.assertTrue(reqCh2.get(0).getName().equals("ch2")); Assert.assertTrue(reqCh2.get(1).getName().equals("ch3")); List<Channel> optCh2 = selector.getOptionalChannels(event2); Assert.assertTrue(optCh2.isEmpty()); Event event3 = new MockEvent(); Map<String, String> header3 = new HashMap<String, String>(); header3.put("myheader", "xyz"); // should match ch1 ch2 ch3 event3.setHeaders(header3); List<Channel> reqCh3 = selector.getRequiredChannels(event3); Assert.assertEquals(3, reqCh3.size()); Assert.assertTrue(reqCh3.get(0).getName().equals("ch1")); Assert.assertTrue(reqCh3.get(1).getName().equals("ch2")); Assert.assertTrue(reqCh3.get(2).getName().equals("ch3")); List<Channel> optCh3 = selector.getOptionalChannels(event3); // All of the optional channels should go away. Assert.assertTrue(optCh3.size() == 0); }
// If the header information cannot map the event to any of the channels // it should always be mapped to the default channel(s). @Test public void testNoSelection() throws Exception { config.put("mapping.foo", "ch1 ch2"); config.put("mapping.bar", "ch2 ch3"); config.put("mapping.xyz", "ch1 ch2 ch3"); config.put("default", "ch1 ch3"); selector = ChannelSelectorFactory.create(channels, config); Assert.assertTrue(selector instanceof MultiplexingChannelSelector); Event noHeaderEvent = new MockEvent(); List<Channel> reqCh1 = selector.getRequiredChannels(noHeaderEvent); List<Channel> optCh1 = selector.getOptionalChannels(noHeaderEvent); Assert.assertEquals(2, reqCh1.size()); Assert.assertTrue(reqCh1.get(0).getName().equals("ch1")); Assert.assertTrue(reqCh1.get(1).getName().equals("ch3")); Assert.assertTrue(optCh1.isEmpty()); Map<String, String> header2 = new HashMap<String, String>(); header2.put("someheader", "foo"); Event invalidHeaderEvent = new MockEvent(); invalidHeaderEvent.setHeaders(header2); List<Channel> reqCh2 = selector.getRequiredChannels(invalidHeaderEvent); List<Channel> optCh2 = selector.getOptionalChannels(invalidHeaderEvent); Assert.assertEquals(2, reqCh2.size()); Assert.assertTrue(reqCh2.get(0).getName().equals("ch1")); Assert.assertTrue(reqCh2.get(1).getName().equals("ch3")); Assert.assertTrue(optCh2.isEmpty()); Map<String, String> header3 = new HashMap<String, String>(); header3.put("myheader", "bar1"); Event unmatchedHeaderEvent = new MockEvent(); unmatchedHeaderEvent.setHeaders(header3); List<Channel> reqCh3 = selector.getRequiredChannels(unmatchedHeaderEvent); List<Channel> optCh3 = selector.getOptionalChannels(unmatchedHeaderEvent); Assert.assertEquals(2, reqCh3.size()); Assert.assertTrue(reqCh3.get(0).getName().equals("ch1")); Assert.assertTrue(reqCh3.get(1).getName().equals("ch3")); Assert.assertTrue(optCh3.isEmpty()); Map<String, String> header4 = new HashMap<String, String>(); header4.put("myheader", "zebra"); Event zebraEvent = new MockEvent(); zebraEvent.setHeaders(header4); List<Channel> reqCh4 = selector.getRequiredChannels(zebraEvent); List<Channel> optCh4 = selector.getOptionalChannels(zebraEvent); Assert.assertEquals(2, reqCh4.size()); Assert.assertTrue(reqCh4.get(0).getName().equals("ch1")); Assert.assertTrue(reqCh4.get(1).getName().equals("ch3")); // Since ch1 is also in default list, it is removed. Assert.assertTrue(optCh4.size() == 1); Assert.assertTrue(optCh4.get(0).getName().equals("ch2")); List<Channel> allChannels = selector.getAllChannels(); Assert.assertTrue(allChannels.size() == 3); Assert.assertTrue(allChannels.get(0).getName().equals("ch1")); Assert.assertTrue(allChannels.get(1).getName().equals("ch2")); Assert.assertTrue(allChannels.get(2).getName().equals("ch3")); }