InstanceInteractionIntegTest.java

/**
 * Copyright 2010 the original author or authors.
 *
 * 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 datameer.awstasks.aws.ec2;

import static org.junit.Assert.*;

import java.io.File;
import java.io.FileWriter;
import java.io.OutputStream;

import org.junit.Test;

import datameer.awstasks.aws.ec2.ssh.SshClient;
import datameer.awstasks.util.Ec2Util;
import datameer.awstasks.util.IoUtil;

public class InstanceInteractionIntegTest extends AbstractEc2IntegrationInteractionTest {

    private OutputStream _sysOutStream = IoUtil.closeProtectedStream(System.out);

    @Test
    public void testScpUploadToAllInstances() throws Exception {
        File privateKeyFile = new File(_ec2Conf.getPrivateKeyFile());
        SshClient sshClient = _instanceGroup.createSshClient(TEST_USERNAME, privateKeyFile);
        File localFile = new File("README.markdown");
        String remoteDir = "~/";
        sshClient.uploadFile(localFile, remoteDir);

        File localDestinationFolder = _tempFolder.newFolder("localDestinationFolder");
        sshClient.downloadFile(remoteDir + localFile.getName(), localDestinationFolder, false);
        assertEquals(1, localDestinationFolder.list().length);
        assertEquals(localFile.length(), new File(localDestinationFolder, localFile.getName()).length());
    }

    @Test
    public void testScpUploadToSpecificInstances() throws Exception {
        File privateKeyFile = new File(_ec2Conf.getPrivateKeyFile());
        SshClient sshClient = _instanceGroup.createSshClient(TEST_USERNAME, privateKeyFile);
        File localFile = new File("build.xml");
        String remoteDir = "~/";
        sshClient.uploadFile(localFile, remoteDir, new int[] { 0 });

        File localDestinationFolder = _tempFolder.newFolder("localDestinationFolder");
        sshClient.downloadFile(remoteDir + localFile.getName(), localDestinationFolder, false, new int[] { 0 });
        assertEquals(1, localDestinationFolder.list().length);
        assertEquals(localFile.length(), new File(localDestinationFolder, localFile.getName()).length());

        try {
            sshClient.downloadFile(remoteDir + localFile.getName(), localDestinationFolder, false, new int[] { 1 });
            fail("should throw exception");
        } catch (Exception e) {
            // expected
        }
    }

    @Test
    public void testSshExecutionToAllInstances() throws Exception {
        File privateKeyFile = new File(_ec2Conf.getPrivateKeyFile());
        SshClient sshClient = _instanceGroup.createSshClient(TEST_USERNAME, privateKeyFile);
        sshClient.executeCommand("ls -l", _sysOutStream);
        String noneExistingFile = "abcfi";
        try {
            sshClient.executeCommand("rm " + noneExistingFile, _sysOutStream);
            fail("should throw exception");
        } catch (Exception e) {
            // expected
        }
        sshClient.executeCommand("touch " + noneExistingFile, _sysOutStream);
        sshClient.executeCommand("rm " + noneExistingFile, _sysOutStream);
    }

    @Test
    public void testSshExecutionToSpecificInstances() throws Exception {
        File privateKeyFile = new File(_ec2Conf.getPrivateKeyFile());
        SshClient sshClient1 = _instanceGroup.createSshClient(TEST_USERNAME, privateKeyFile);

        String noneExistingFile = "abcfi";
        sshClient1.executeCommand("touch " + noneExistingFile, _sysOutStream, new int[] { 0 });
        sshClient1.executeCommand("rm " + noneExistingFile, _sysOutStream, new int[] { 0 });
        try {
            sshClient1.executeCommand("rm " + noneExistingFile, _sysOutStream, new int[] { 1 });
            fail("should throw exception");
        } catch (Exception e) {
            // expected
        }
    }

    @Test
    public void testSshExecutionFromFile() throws Exception {
        File privateKeyFile = new File(_ec2Conf.getPrivateKeyFile());
        SshClient sshClient = _instanceGroup.createSshClient(TEST_USERNAME, privateKeyFile);
        File commandFile = _tempFolder.newFile("commands.txt");
        FileWriter fileWriter = new FileWriter(commandFile);
        fileWriter.write("ls -l\n");
        String noneExistingFile = "abcfi";
        fileWriter.write("touch " + noneExistingFile + "\n");
        fileWriter.write("rm " + noneExistingFile + "\n");
        fileWriter.close();
        sshClient.executeCommandFile(commandFile, _sysOutStream, new int[] { 0 });
    }

    @Test
    public void testGroupExists() throws Exception {
        assertTrue(Ec2Util.groupExists(_ec2, TEST_SECURITY_GROUP));
        assertFalse(Ec2Util.groupExists(_ec2, "wegszdfsf"));
    }

}