/** * * Copyright 2018 Paul Schaub. * * 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.smackx.ox; import static org.jivesoftware.smack.test.util.XmlAssertUtil.assertXmlSimilar; import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.junit.jupiter.api.Assertions.assertThrows; import java.util.ArrayList; import java.util.Collections; import java.util.Date; import java.util.HashSet; import java.util.List; import java.util.Set; import org.jivesoftware.smack.packet.ExtensionElement; import org.jivesoftware.smack.packet.Message; import org.jivesoftware.smack.test.util.SmackTestSuite; import org.jivesoftware.smack.test.util.TestUtils; import org.jivesoftware.smack.xml.XmlPullParser; import org.jivesoftware.smack.xml.XmlPullParserException; import org.jivesoftware.smackx.ox.element.CryptElement; import org.jivesoftware.smackx.ox.element.OpenPgpElement; import org.jivesoftware.smackx.ox.element.SignElement; import org.jivesoftware.smackx.ox.element.SigncryptElement; import org.jivesoftware.smackx.ox.provider.OpenPgpContentElementProvider; import org.jivesoftware.smackx.ox.provider.OpenPgpElementProvider; import org.junit.jupiter.api.Test; import org.jxmpp.jid.Jid; import org.jxmpp.jid.impl.JidCreate; import org.jxmpp.stringprep.XmppStringprepException; public class OpenPgpElementTest extends SmackTestSuite { private final Set recipients; // 2014-07-10T15:06:00.000+00:00 private static final Date testDate = new Date(1405004760000L); public OpenPgpElementTest() throws XmppStringprepException { super(); Set jids = new HashSet<>(); jids.add(JidCreate.bareFrom("alice@wonderland.lit")); jids.add(JidCreate.bareFrom("bob@builder.lit")); this.recipients = Collections.unmodifiableSet(jids); } @Test public void providerTest() throws Exception { String expected = "" + "BASE64_OPENPGP_MESSAGE" + ""; OpenPgpElement element = new OpenPgpElement("BASE64_OPENPGP_MESSAGE"); assertXmlSimilar(expected, element.toXML().toString()); XmlPullParser parser = TestUtils.getParser(expected); OpenPgpElement parsed = OpenPgpElementProvider.TEST_INSTANCE.parse(parser); assertEquals(element.getEncryptedBase64MessageContent(), parsed.getEncryptedBase64MessageContent()); } @Test public void simplifiedConstructorTest() { ArrayList payload = new ArrayList<>(); payload.add(new Message.Body("de", "Hallo Welt!")); CryptElement element = new CryptElement(recipients, payload); assertNotNull(element.getTimestamp()); } @SuppressWarnings("UndefinedEquals") @Test public void signElementProviderTest() throws Exception { String expected = "" + "" + "" + "" + "" + "Hello World!" + "" + ""; List payload = new ArrayList<>(); payload.add(new Message.Body("en", "Hello World!")); SignElement element = new SignElement(recipients, testDate, payload); assertXmlSimilar(expected, element.toXML().toString()); XmlPullParser parser = TestUtils.getParser(expected); SignElement parsed = (SignElement) OpenPgpContentElementProvider.parseOpenPgpContentElement(parser); assertEquals(element.getTimestamp(), parsed.getTimestamp()); assertEquals(element.getTo(), parsed.getTo()); assertEquals(element.getExtensions(), parsed.getExtensions()); } @SuppressWarnings("UndefinedEquals") @Test public void cryptElementProviderTest() throws Exception { String expected = "" + "" + "" + "" + "The cake is a lie." + "" + "f0rm1l4n4-mT8y33j!Y%fRSrcd^ZE4Q7VDt1L%WEgR!kv" + ""; List payload = new ArrayList<>(); payload.add(new Message.Body("en", "The cake is a lie.")); Set to = new HashSet<>(); to.add(JidCreate.bareFrom("alice@wonderland.lit")); CryptElement element = new CryptElement(to, "f0rm1l4n4-mT8y33j!Y%fRSrcd^ZE4Q7VDt1L%WEgR!kv", testDate, payload); assertXmlSimilar(expected, element.toXML().toString()); XmlPullParser parser = TestUtils.getParser(expected); CryptElement parsed = (CryptElement) OpenPgpContentElementProvider.parseOpenPgpContentElement(parser); assertEquals(element.getTimestamp(), parsed.getTimestamp()); assertEquals(element.getTo(), parsed.getTo()); assertEquals(element.getExtensions(), parsed.getExtensions()); } @SuppressWarnings("UndefinedEquals") @Test public void signcryptElementProviderTest() throws Exception { String expected = "" + "" + "" + "" + "This is a secret message." + "" + "f0rm1l4n4-mT8y33j!Y%fRSrcd^ZE4Q7VDt1L%WEgR!kv" + ""; List payload = new ArrayList<>(); payload.add(new Message.Body("en", "This is a secret message.")); Set jids = new HashSet<>(); jids.add(JidCreate.bareFrom("juliet@example.org")); SigncryptElement element = new SigncryptElement(jids, "f0rm1l4n4-mT8y33j!Y%fRSrcd^ZE4Q7VDt1L%WEgR!kv", testDate, payload); assertXmlSimilar(expected, element.toXML().toString()); XmlPullParser parser = TestUtils.getParser(expected); SigncryptElement parsed = (SigncryptElement) OpenPgpContentElementProvider.parseOpenPgpContentElement(parser); assertEquals(element.getTimestamp(), parsed.getTimestamp()); assertEquals(element.getTo(), parsed.getTo()); assertEquals(element.getExtensions(), parsed.getExtensions()); assertEquals(payload.get(0), element.getExtension(Message.Body.NAMESPACE)); assertEquals(payload.get(0), element.getExtension(Message.Body.ELEMENT, Message.Body.NAMESPACE)); } @Test public void openPgpContentElementProvider_invalidElementTest() { String invalidElementXML = "" + "This is a secret message." + ""; assertThrows(XmlPullParserException.class, () -> OpenPgpContentElementProvider.parseOpenPgpContentElement(invalidElementXML)); } }