package de.cinovo.cloudconductor.server.test; /* * #%L * cloudconductor-server * %% * Copyright (C) 2013 - 2014 Cinovo AG * %% * 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. * #L% */ import java.util.Set; import org.junit.Assert; import org.junit.Test; import org.junit.runner.RunWith; import de.cinovo.cloudconductor.api.lib.exceptions.CloudConductorException; import de.cinovo.cloudconductor.api.lib.manager.SSHKeyHandler; import de.cinovo.cloudconductor.api.model.SSHKey; import de.cinovo.cloudconductor.server.APITest; import de.taimos.daemon.spring.SpringDaemonTestRunner; /** * * Copyright 2013 Cinovo AG<br> * <br> * * @author hoegertn * */ @RunWith(SpringDaemonTestRunner.class) @SuppressWarnings("javadoc") public class SSHKeyTest extends APITest { @Test public void test() throws CloudConductorException { String firstContent = "SSH key content"; String firstOwner = "foobar"; String newContent = "key content 2"; String newOwner = "admin"; SSHKeyHandler handler = new SSHKeyHandler(this.getCSApi()); { Set<SSHKey> keys = handler.get(); Assert.assertEquals(1, keys.size()); SSHKey key = keys.iterator().next(); Assert.assertEquals(firstOwner, key.getOwner()); Assert.assertEquals(firstContent, key.getKey()); } { SSHKey load = handler.get(firstOwner); Assert.assertEquals(firstOwner, load.getOwner()); Assert.assertEquals(firstContent, load.getKey()); } { SSHKey newKey = new SSHKey(newContent, newOwner); handler.save(newKey); } { Set<SSHKey> keys2 = handler.get(); Assert.assertEquals(2, keys2.size()); for (SSHKey sk : keys2) { if (sk.getOwner().equals(firstOwner)) { Assert.assertEquals(firstContent, sk.getKey()); } else if (sk.getOwner().equals(newOwner)) { Assert.assertEquals(newContent, sk.getKey()); } else { Assert.fail(); } } } { SSHKey load2 = handler.get(newOwner); Assert.assertEquals(newOwner, load2.getOwner()); Assert.assertEquals(newContent, load2.getKey()); } { handler.delete(newOwner); } { Set<SSHKey> keys = handler.get(); Assert.assertEquals(1, keys.size()); SSHKey key = keys.iterator().next(); Assert.assertEquals(firstOwner, key.getOwner()); Assert.assertEquals(firstContent, key.getKey()); } } }