/
RemoteQueueBindImplTest.java
106 lines (92 loc) · 5.14 KB
/
RemoteQueueBindImplTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.activemq.artemis.tests.unit.core.server.cluster.impl;
import java.util.ArrayList;
import java.util.List;
import java.util.function.IntFunction;
import org.apache.activemq.artemis.api.core.SimpleString;
import org.apache.activemq.artemis.core.server.Queue;
import org.apache.activemq.artemis.core.server.cluster.impl.MessageLoadBalancingType;
import org.apache.activemq.artemis.core.server.cluster.impl.RemoteQueueBindingImpl;
import org.apache.activemq.artemis.tests.unit.core.postoffice.impl.fakes.FakeQueue;
import org.apache.activemq.artemis.tests.util.ActiveMQTestBase;
import org.apache.activemq.artemis.utils.RandomUtil;
import org.junit.Test;
public class RemoteQueueBindImplTest extends ActiveMQTestBase {
private void testAddRemoveConsumerWithFilter(IntFunction<SimpleString> filterFactory,
int size,
int expectedSize) throws Exception {
final long id = RandomUtil.randomLong();
final SimpleString address = RandomUtil.randomSimpleString();
final SimpleString uniqueName = RandomUtil.randomSimpleString();
final SimpleString routingName = RandomUtil.randomSimpleString();
final Long remoteQueueID = RandomUtil.randomLong();
final SimpleString filterString = new SimpleString("A>B");
final Queue storeAndForwardQueue = new FakeQueue(null);
final SimpleString bridgeName = RandomUtil.randomSimpleString();
final int distance = 0;
RemoteQueueBindingImpl binding = new RemoteQueueBindingImpl(id, address, uniqueName, routingName, remoteQueueID, filterString, storeAndForwardQueue, bridgeName, distance, MessageLoadBalancingType.ON_DEMAND);
final List<SimpleString> filters = new ArrayList<>(size);
for (int i = 0; i < size; i++) {
final SimpleString filter = filterFactory.apply(i);
filters.add(filter);
binding.addConsumer(filter);
}
assertEquals(expectedSize, binding.getFilters().size());
for (int i = 0; i < size; i++) {
binding.removeConsumer(filters.get(i));
}
assertEquals(0, binding.getFilters().size());
}
@Test
public void testAddRemoveConsumer() throws Exception {
testAddRemoveConsumerWithFilter(i -> new SimpleString("B" + i + "<A"), 100, 100);
}
@Test
public void testAddRemoveConsumerUsingSameFilter() throws Exception {
testAddRemoveConsumerWithFilter(i -> new SimpleString("B" + 0 + "<A"), 100, 1);
}
@Test
public void testAddRemoveConsumerUsingEmptyFilters() throws Exception {
testAddRemoveConsumerWithFilter(i -> new SimpleString(""), 1, 0);
}
@Test
public void testAddRemoveConsumerUsingNullFilters() throws Exception {
testAddRemoveConsumerWithFilter(i -> null, 1, 0);
}
@Test
public void testIsHighAcceptPriority() throws Exception {
final long id = RandomUtil.randomLong();
final SimpleString address = RandomUtil.randomSimpleString();
final SimpleString uniqueName = RandomUtil.randomSimpleString();
final SimpleString routingName = RandomUtil.randomSimpleString();
final Long remoteQueueID = RandomUtil.randomLong();
final SimpleString filterString = new SimpleString("A>B");
final Queue storeAndForwardQueue = new FakeQueue(null);
final SimpleString bridgeName = RandomUtil.randomSimpleString();
final int distance = 0;
RemoteQueueBindingImpl bindingOff = new RemoteQueueBindingImpl(id, address, uniqueName, routingName, remoteQueueID, filterString, storeAndForwardQueue, bridgeName, distance, MessageLoadBalancingType.OFF);
bindingOff.addConsumer(null);
assertFalse(bindingOff.isHighAcceptPriority(null));
RemoteQueueBindingImpl bindingOffWithRedistribution = new RemoteQueueBindingImpl(id, address, uniqueName, routingName, remoteQueueID, filterString, storeAndForwardQueue, bridgeName, distance, MessageLoadBalancingType.OFF_WITH_REDISTRIBUTION);
bindingOffWithRedistribution.addConsumer(null);
// not really intuitive, but via getNextBinding (initial routing) BindingsImpl.matchBinding() traps remote bindings
// with OFF_WITH_REDISTRIBUTION which makes the need for change in isHighAcceptPriority redundant
// and ensures that redistribution can occur as isHighAcceptPriority is invoked from redistribute
assertTrue(bindingOffWithRedistribution.isHighAcceptPriority(null));
}
}