kdeconnect-android/tests/org/kde/kdeconnect/NetworkPackageTest.java

/*
 * Copyright 2015 Vineet Garg <grg.vineet@gmail.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of
 * the License or (at your option) version 3 or any later version
 * accepted by the membership of KDE e.V. (or its successor approved
 * by the membership of KDE e.V.), which shall act as a proxy
 * defined in Section 14 of version 3 of the license.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

package org.kde.kdeconnect;

import android.test.AndroidTestCase;
import android.util.Log;

import org.json.JSONException;
import org.kde.kdeconnect.Helpers.SecurityHelpers.RsaHelper;
import org.skyscreamer.jsonassert.JSONAssert;

import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.PrivateKey;
import java.security.PublicKey;

public class NetworkPackageTest extends AndroidTestCase{

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        // Called before each test
    }

    @Override
    protected void tearDown() throws Exception {
        super.tearDown();
        // Called after each test
    }

    public void testNetworkPackage() throws JSONException {
        NetworkPackage np = new NetworkPackage("com.test");

        np.set("hello", "hola");
        assertEquals(np.getString("hello", "bye"), "hola");

        np.set("hello", "");
        assertEquals(np.getString("hello", "bye"), "");

        assertEquals(np.getString("hi", "bye"), "bye");

        np.set("foo", "bar");
        String serialized = np.serialize();
        NetworkPackage np2 = NetworkPackage.unserialize(serialized);

        assertEquals(np.getLong("id"), np2.getLong("id"));
        assertEquals(np.getString("type"), np2.getString("type"));
        assertEquals(np.getJSONArray("body"), np2.getJSONArray("body"));

        String json = "{\"id\":123,\"type\":\"test\",\"body\":{\"testing\":true}}";
        np2 = NetworkPackage.unserialize(json);
        assertEquals(np2.getId(),123);
        assertEquals(np2.getBoolean("testing"), true);
        assertEquals(np2.getBoolean("not_testing"), false);
        assertEquals(np2.getBoolean("not_testing", true), true);

    }

    public void testIdentity(){

        NetworkPackage np = NetworkPackage.createIdentityPackage(getContext());

        assertEquals(np.getInt("protocolVersion"), NetworkPackage.ProtocolVersion);

    }

    public void testEncryption() throws JSONException {
        NetworkPackage original = new NetworkPackage("com.test");
        original.set("hello", "hola");

        NetworkPackage copy = NetworkPackage.unserialize(original.serialize());

        NetworkPackage decrypted = new NetworkPackage("");

        KeyPair keyPair;
        try {
            KeyPairGenerator keyGen = KeyPairGenerator.getInstance("RSA");
            keyGen.initialize(2048);
            keyPair = keyGen.genKeyPair();
        } catch(Exception e) {
            e.printStackTrace();
            Log.e("KDE/initializeRsaKeys", "Exception");
            return;
        }

        PrivateKey privateKey = keyPair.getPrivate();
        assertNotNull(privateKey);

        PublicKey publicKey = keyPair.getPublic();
        assertNotNull(publicKey);

        // Encrypt and decrypt np
        assertEquals(original.getType(), "com.test");
        try {
            NetworkPackage encrypted = RsaHelper.encrypt(original,publicKey);
            assertEquals(encrypted.getType(), NetworkPackage.PACKAGE_TYPE_ENCRYPTED);

            decrypted = RsaHelper.decrypt(encrypted, privateKey);
            assertEquals(decrypted.getType(), "com.test");

        }catch (Exception e){
            e.printStackTrace();
        }

        // np should be equal to np2
        assertEquals(decrypted.getLong("id"), copy.getLong("id"));
        assertEquals(decrypted.getType(), copy.getType());
        assertEquals(decrypted.getJSONArray("body"), copy.getJSONArray("body"));

        String json = "{\"body\":{\"nowPlaying\":\"A really long song name - A really long artist name\",\"player\":\"A really long player name\",\"the_meaning_of_life_the_universe_and_everything\":\"42\"},\"id\":945945945,\"type\":\"kdeconnect.a_really_really_long_package_type\"}\n";
        NetworkPackage longJsonNp = NetworkPackage.unserialize(json);
        try {
            NetworkPackage encrypted = RsaHelper.encrypt(longJsonNp, publicKey);
            decrypted = RsaHelper.decrypt(encrypted, privateKey);
            String decryptedJson = decrypted.serialize();
            JSONAssert.assertEquals(json, decryptedJson, true);
        }catch (Exception e){
            e.printStackTrace();
        }
    }

}