Skip to content

Commit 517e072

Browse files
author
Lloyd Watkin
committed
Add test for outgoing packet remapping
1 parent 5eb18cc commit 517e072

1 file changed

Lines changed: 29 additions & 3 deletions

File tree

src/test/java/org/buddycloud/channelserver/queue/FederatedQueueManagerTest.java

Lines changed: 29 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -183,9 +183,6 @@ public void testOutgoingFederatedPacketsAreRoutedBackToOriginalSender() throws E
183183
Assert.assertEquals(1, channelsEngine.size());
184184
Packet redirected = channelsEngine.poll();
185185

186-
System.out.println(packet);
187-
System.out.println(redirected);
188-
189186
Assert.assertEquals(packet.getFrom(), redirected.getTo());
190187
}
191188

@@ -229,4 +226,33 @@ public void testOutgoingFederatedPacketsFromDifferentClientsUsingSameIdAreRouted
229226
Assert.assertEquals(clientOnePacket.getFrom(), clientOneRedirected.getTo());
230227
Assert.assertEquals(clientTwoPacket.getFrom(), clientTwoRedirected.getTo());
231228
}
229+
230+
@Test
231+
public void testOutgoingIqPacketsGetIdMapped() throws Exception {
232+
channelsEngine.clear();
233+
234+
String originalId = "id:12345";
235+
IQ packet = new IQ();
236+
packet.setFrom(new JID("romeo@montague.lit/street"));
237+
packet.setTo(new JID("topics.capulet.lit"));
238+
packet.setType(IQ.Type.get);
239+
packet.setID(originalId);
240+
packet.getElement().addAttribute("remote-server-discover", "false");
241+
242+
queueManager.addChannelMap(new JID("topics.capulet.lit"));
243+
244+
queueManager.process(packet.createCopy());
245+
246+
IQ packetExternal = (IQ) channelsEngine.poll();
247+
248+
Assert.assertFalse(originalId.equals(packetExternal.getID()));
249+
250+
IQ response = IQ.createResultIQ(packetExternal);
251+
queueManager.passResponseToRequester(response);
252+
253+
IQ packetInternal = (IQ) channelsEngine.poll();
254+
255+
Assert.assertTrue(originalId.equals(packetInternal.getID()));
256+
}
257+
232258
}

0 commit comments

Comments
 (0)