Smack/smack-openpgp-bouncycastle/src/test/java/org/jivesoftware/smackx/ox/bouncycastle/BouncyCastleOpenPgpProviderTest.java

/**
 *
 * 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.bouncycastle;

import static junit.framework.TestCase.assertTrue;
import static org.custommonkey.xmlunit.XMLAssert.assertXMLEqual;

import java.security.Security;
import java.util.Collections;

import org.jivesoftware.smack.packet.ExtensionElement;
import org.jivesoftware.smack.packet.Message;
import org.jivesoftware.smack.test.util.SmackTestSuite;
import org.jivesoftware.smackx.ox.chat.OpenPgpMessage;
import org.jivesoftware.smackx.ox.element.OpenPgpContentElement;
import org.jivesoftware.smackx.ox.element.OpenPgpElement;
import org.jivesoftware.smackx.ox.element.PubkeyElement;
import org.jivesoftware.smackx.ox.element.SigncryptElement;

import org.bouncycastle.jce.provider.BouncyCastleProvider;
import org.junit.Ignore;
import org.jxmpp.jid.BareJid;
import org.jxmpp.jid.Jid;
import org.jxmpp.jid.impl.JidCreate;

public class BouncyCastleOpenPgpProviderTest extends SmackTestSuite {

    @Ignore
    public void encryptAndSign_decryptAndVerifyElementTest() throws Exception {
        Security.addProvider(new BouncyCastleProvider());

        // Create providers for alice and the cat
        BareJid alice = JidCreate.bareFrom("alice@wonderland.lit");
        BareJid cheshire = JidCreate.bareFrom("cheshire@wonderland.lit");
        BCOpenPgpProvider aliceProvider = new BCOpenPgpProvider(alice);
        BCOpenPgpProvider cheshireProvider = new BCOpenPgpProvider(cheshire);

        aliceProvider.createOpenPgpKeyPair();
        cheshireProvider.createOpenPgpKeyPair();

        // dry exchange keys

        PubkeyElement aliceKeys = aliceProvider.createPubkeyElement(aliceProvider.getPrimaryOpenPgpKeyPairFingerprint());
        PubkeyElement cheshireKeys = cheshireProvider.createPubkeyElement(cheshireProvider.getPrimaryOpenPgpKeyPairFingerprint());
        aliceProvider.storePublicKey(cheshire, cheshireProvider.getPrimaryOpenPgpKeyPairFingerprint(), cheshireKeys);
        cheshireProvider.storePublicKey(alice, aliceProvider.getPrimaryOpenPgpKeyPairFingerprint(), aliceKeys);

        // Create signed and encrypted message from alice to the cheshire cat
        SigncryptElement signcryptElement = new SigncryptElement(
                Collections.<Jid>singleton(cheshire),
                Collections.<ExtensionElement>singletonList(
                        new Message.Body("en", "How do you know I’m mad?")));
        OpenPgpElement encrypted = aliceProvider.signAndEncrypt(
                signcryptElement,
                aliceProvider.getPrimaryOpenPgpKeyPairFingerprint(),
                Collections.singleton(cheshireProvider.getPrimaryOpenPgpKeyPairFingerprint()));

        // Decrypt the message as the cheshire cat
        OpenPgpMessage decrypted = cheshireProvider.decryptAndVerify(encrypted, Collections.singleton(aliceProvider.getPrimaryOpenPgpKeyPairFingerprint()));
        OpenPgpContentElement content = decrypted.getOpenPgpContentElement();

        assertTrue(content instanceof SigncryptElement);
        assertXMLEqual(signcryptElement.toXML(null).toString(), content.toXML(null).toString());
    }
}