/** * * Copyright the original author or authors * * Licensed 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.jivesoftware.util; import static org.junit.Assert.assertEquals; import java.io.StringReader; import java.io.StringWriter; import java.util.ArrayList; import java.util.LinkedList; import java.util.List; import java.util.Queue; import javax.xml.transform.OutputKeys; import javax.xml.transform.Source; import javax.xml.transform.Transformer; import javax.xml.transform.TransformerFactory; import javax.xml.transform.stream.StreamResult; import javax.xml.transform.stream.StreamSource; import org.jivesoftware.smack.packet.Stanza; /** * This class can be used in conjunction with a mocked XMPP connection ( * {@link ConnectionUtils#createMockedConnection(Protocol, String, String)}) to * verify an XMPP protocol. This can be accomplished in the following was: * * Example: * *
 * 
 * public void methodToTest() {
 *   Packet packet = new Packet(); // create an XMPP packet
 *   PacketCollector collector = connection.createPacketCollector(new PacketIDFilter());
 *   connection.sendPacket(packet);
 *   Packet reply = collector.nextResult();
 * }
 * 
 * public void testMethod() {
 *   // create protocol
 *   Protocol protocol = new Protocol();
 *   // create mocked connection
 *   XMPPConnection connection = ConnectionUtils.createMockedConnection(protocol, "user@xmpp-server", "xmpp-server");
 *   
 *   // add reply packet to protocol
 *   Packet reply = new Packet();
 *   protocol.add(reply);
 *   
 *   // call method to test
 *   methodToTest();
 *   
 *   // verify protocol
 *   protocol.verifyAll();
 * }
 * 
 * 
* * Additionally to adding the response to the protocol instance you can pass * verifications that will be executed when {@link #verifyAll()} is invoked. * (See {@link Verification} for more details.) *

* If the {@link #printProtocol} flag is set to true {@link #verifyAll()} will * also print out the XML messages in the order they are sent to the console. * This may be useful to inspect the whole protocol "by hand". * * @author Henning Staib */ public class Protocol { /** * Set to true to print XML messages to the console while * verifying the protocol. */ public boolean printProtocol = false; // responses to requests are taken form this queue Queue responses = new LinkedList(); // list of verifications List[]> verificationList = new ArrayList[]>(); // list of requests List requests = new ArrayList(); // list of all responses List responsesList = new ArrayList(); /** * Adds a responses and all verifications for the request/response pair to * the protocol. * * @param response the response for a request * @param verifications verifications for request/response pair */ public void addResponse(Stanza response, Verification... verifications) { responses.offer(response); verificationList.add(verifications); responsesList.add(response); } /** * Verifies the request/response pairs by checking if their numbers match * and executes the verification for each pair. */ @SuppressWarnings("unchecked") public void verifyAll() { assertEquals(requests.size(), responsesList.size()); if (printProtocol) System.out.println("=================== Start ===============\n"); for (int i = 0; i < requests.size(); i++) { Stanza request = requests.get(i); Stanza response = responsesList.get(i); if (printProtocol) { System.out.println("------------------- Request -------------\n"); System.out.println(prettyFormat(request.toXML().toString())); System.out.println("------------------- Response ------------\n"); if (response != null) { System.out.println(prettyFormat(response.toXML().toString())); } else { System.out.println("No response"); } } Verification[] verifications = (Verification[]) verificationList.get(i); if (verifications != null) { for (Verification verification : verifications) { verification.verify(request, response); } } } if (printProtocol) System.out.println("=================== End =================\n"); } /** * Returns the responses queue. * * @return the responses queue */ protected Queue getResponses() { return responses; } /** * Returns a list of all collected requests. * * @return list of requests */ public List getRequests() { return requests; } private String prettyFormat(String input, int indent) { try { Source xmlInput = new StreamSource(new StringReader(input)); StringWriter stringWriter = new StringWriter(); StreamResult xmlOutput = new StreamResult(stringWriter); Transformer transformer = TransformerFactory.newInstance().newTransformer(); transformer.setOutputProperty(OutputKeys.INDENT, "yes"); transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", String.valueOf(indent)); transformer.transform(xmlInput, xmlOutput); return xmlOutput.getWriter().toString(); } catch (Exception e) { return "error while formatting the XML: " + e.getMessage(); } } private String prettyFormat(String input) { return prettyFormat(input, 2); } }