org.panbox.desktop.devicemgmt.TestAddDevice.java Source code

Introduction

Here is the source code for org.panbox.desktop.devicemgmt.TestAddDevice.java

Source

/*
 * 
 *               Panbox - encryption for cloud storage 
 *      Copyright (C) 2014-2015 by Fraunhofer SIT and Sirrix AG 
 *
 * 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 3 of the License, or
 * (at your option) any later version.
 * 
 * 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/>.
 * 
 * Additonally, third party code may be provided with notices and open source
 * licenses from communities and third parties that govern the use of those
 * portions, and any licenses granted hereunder do not alter any rights and
 * obligations you may have under such open source licenses, however, the
 * disclaimer of warranty and limitation of liability provisions of the GPLv3 
 * will apply to all the product.
 * 
 */
package org.panbox.desktop.devicemgmt;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

import java.io.File;
import java.security.KeyPair;
import java.security.cert.X509Certificate;

import org.apache.commons.io.FileUtils;
import org.junit.Before;
import org.junit.Test;
import org.panbox.Settings;
import org.panbox.core.crypto.CryptCore;
import org.panbox.core.devicemgmt.DeviceType;
import org.panbox.core.identitymgmt.Identity;
import org.panbox.core.identitymgmt.SimpleAddressbook;
import org.panbox.core.pairing.PairingIPersonDummy;
import org.panbox.desktop.common.devicemgmt.DeviceManagerException;
import org.panbox.desktop.common.devicemgmt.DeviceManagerImpl;
import org.panbox.desktop.common.identitymgmt.sqlightimpl.AddressbookManager;
import org.panbox.desktop.common.identitymgmt.sqlightimpl.IdentityManager;

public class TestAddDevice {

    private DeviceManagerImpl manager;

    @Before
    public void setUp() throws Exception {
        // test cleanup
        File confDir = new File(Settings.getInstance().getConfDir());
        if (confDir.exists()) {
            // clean config directory
            FileUtils.deleteDirectory(confDir);
        }
        confDir.mkdirs();

        File test123share = new File("test123share");
        if (test123share.exists()) {
            // clean test123share directory
            FileUtils.deleteDirectory(test123share);
        }

        // test preparation
        IdentityManager identMgmr = IdentityManager.getInstance();
        AddressbookManager aBookMgr = new AddressbookManager();
        // could not load identity -> create a new one!
        Identity id = new Identity(new SimpleAddressbook(), "testIdentity@example.org", "FirstName", "Lastname");
        KeyPair ownerKeySign = CryptCore.generateKeypair();
        KeyPair ownerKeyEnc = CryptCore.generateKeypair();
        KeyPair deviceKey = CryptCore.generateKeypair();

        id.setOwnerKeySign(ownerKeySign, "test".toCharArray());
        id.setOwnerKeyEnc(ownerKeyEnc, "test".toCharArray());
        id.addDeviceKey(deviceKey, "TestDevice");
        Settings.getInstance().setDeviceName("TestDevice");
        identMgmr.init(aBookMgr);
        identMgmr.storeMyIdentity(id);

        manager = DeviceManagerImpl.getInstance();
        manager.setIdentity(id);
    }

    @Test
    public void test() {
        try {
            int before = manager.getDeviceList().size();

            KeyPair devKeyPair = CryptCore.generateKeypair();
            X509Certificate deviceCert = CryptCore.createSelfSignedX509Certificate(devKeyPair.getPrivate(),
                    devKeyPair.getPublic(), new PairingIPersonDummy("email", "firstname", "lastname"));

            manager.addDevice("testdeviceDesktop", deviceCert, DeviceType.DESKTOP);

            assertEquals(before + 1, manager.getDeviceList().size());

            devKeyPair = CryptCore.generateKeypair();
            deviceCert = CryptCore.createSelfSignedX509Certificate(devKeyPair.getPrivate(), devKeyPair.getPublic(),
                    new PairingIPersonDummy("email", "firstname", "lastname"));

            manager.addDevice("testdeviceMobile", deviceCert, DeviceType.MOBILE);

            assertEquals(before + 2, manager.getDeviceList().size());
        } catch (DeviceManagerException e) {
            e.printStackTrace();
            fail();
        }
    }
}