/*
 * 
 *               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();
		}
	}
}