1 package org.apache.commons.jcs3.utils.discovery;
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 import java.io.IOException;
23 import java.util.ArrayList;
24 import java.util.concurrent.Callable;
25 import java.util.concurrent.ExecutionException;
26 import java.util.concurrent.ExecutorService;
27 import java.util.concurrent.Executors;
28 import java.util.concurrent.Future;
29 import java.util.concurrent.TimeUnit;
30 import java.util.concurrent.TimeoutException;
31
32 import org.apache.commons.jcs3.utils.discovery.UDPDiscoveryMessage.BroadcastType;
33 import org.apache.commons.jcs3.utils.serialization.EncryptingSerializer;
34
35 import junit.framework.TestCase;
36
37
38
39
40 public class UDPDiscoverySenderEncryptedUnitTest
41 extends TestCase
42 {
43
44 private static final String ADDRESS = "228.4.5.9";
45
46
47 private static final int PORT = 5556;
48
49
50 private static final String SENDING_HOST = "imaginary host address";
51
52
53 private static final int SENDING_PORT = 1;
54
55
56 private UDPDiscoveryReceiver receiver;
57
58
59 private UDPDiscoverySender sender;
60
61
62
63
64
65
66
67
68 @Override
69 protected void setUp()
70 throws Exception
71 {
72 super.setUp();
73
74 EncryptingSerializer serializer = new EncryptingSerializer();
75 serializer.setPreSharedKey("my_key");
76
77 receiver = new UDPDiscoveryReceiver( null, null, ADDRESS, PORT );
78 receiver.setSerializer(serializer);
79 final Thread t = new Thread( receiver );
80 t.start();
81
82 sender = new UDPDiscoverySender(null, ADDRESS, PORT, 1, serializer);
83 }
84
85
86
87
88
89
90 @Override
91 protected void tearDown()
92 throws Exception
93 {
94 receiver.shutdown();
95 sender.close();
96 super.tearDown();
97 }
98
99
100
101
102
103
104 public void testPassiveBroadcast()
105 throws Exception
106 {
107
108 final ArrayList<String> cacheNames = new ArrayList<>();
109
110
111 sender.passiveBroadcast( SENDING_HOST, SENDING_PORT, cacheNames, 1L );
112
113
114
115 final UDPDiscoveryMessage msg = getMessage();
116 assertNotNull("message not received", msg);
117 assertEquals( "wrong port", SENDING_PORT, msg.getPort() );
118 assertEquals( "wrong message type", BroadcastType.PASSIVE, msg.getMessageType() );
119 }
120
121
122
123
124
125
126 public void testRemoveBroadcast()
127 throws Exception
128 {
129
130 final ArrayList<String> cacheNames = new ArrayList<>();
131
132
133 sender.removeBroadcast( SENDING_HOST, SENDING_PORT, cacheNames, 1L );
134
135
136
137 final UDPDiscoveryMessage msg = getMessage();
138 assertNotNull("message not received", msg);
139 assertEquals( "wrong port", SENDING_PORT, msg.getPort() );
140 assertEquals( "wrong message type", BroadcastType.REMOVE, msg.getMessageType() );
141 }
142
143
144
145
146
147
148 public void testRequestBroadcast()
149 throws Exception
150 {
151
152 sender.requestBroadcast();
153
154
155
156 final UDPDiscoveryMessage msg = getMessage();
157 assertNotNull("message not received", msg);
158 assertEquals( "wrong message type", BroadcastType.REQUEST, msg.getMessageType() );
159
160
161 }
162
163
164
165
166
167
168 private UDPDiscoveryMessage getMessage() {
169 ExecutorService executor = Executors.newCachedThreadPool();
170 Callable<Object> task = new Callable<Object>() {
171 public Object call() throws IOException {
172 return receiver.waitForMessage();
173 }
174 };
175 Future<Object> future = executor.submit(task);
176 try {
177 Object obj = future.get(3, TimeUnit.SECONDS);
178
179 assertTrue( "unexpected crap received", obj instanceof UDPDiscoveryMessage );
180
181 return (UDPDiscoveryMessage) obj;
182 } catch (InterruptedException | ExecutionException | TimeoutException ex) {
183 return null;
184 }
185 }
186 }