#!/bin/bash
# A script to test the official CentOS docker-images.
# Author: Athmane Madjoudj <athmane@fedoraproject.org>

EPEL6_RELEASE_URL="http://dl.fedoraproject.org/pub/epel/6/x86_64/epel-release-6-8.noarch.rpm"
EPEL6_GPGKEY_URL="https://fedoraproject.org/static/0608B895.txt"

# A function to test an output of a command 
# arg1: Command to run 
# arg2: Expected output
#
expect_output(){
    if [ -z "$2" ]
    then
        printf "expect_output requires two args\n"
        return 2
    fi
    $1 | grep -q "$2"
    if [ $? -eq 0 ]
    then
        printf "==> Test PASSED.\n\n" 
        return 0
    else
        printf "==> Test \'$1\' FAILED: expecting \'$2\'\n\n"
        return 1
    fi
    
}

enable_epel() {
    printf "Installing EPEL and some tools ...\n"
    yum -y install wget openssh-clients
    cd /root 
    wget  "$EPEL6_RELEASE_URL"
    rpm --import  "$EPEL6_GPGKEY_URL"
    yum -y localinstall /root/epel-release*.rpm
}

setup_docker() {
    printf "Setting up docker ...\n"
    yum -y install docker-io
    service docker start
    chkconfig docker on
}

test_get_centos_img() {
    printf "TEST 1: Pulling CentOS image ...\n"
    docker pull centos
    printf "\n"
}

test_centos_img_installed(){
    printf "TEST 2: Checking if CentOS image was installed correctly ...\n"
    expect_output 'docker images' 'centos'

}

test_centos_img_runcmd() {
    printf "TEST 3: Running a command inside CentOS image ...\n"
    expect_output 'docker run centos cat /etc/centos-release' 'CentOS'
}

test_centos_img_hostname_match_cont_id() {
    printf "TEST 4: hostname inside CentOS image matches container id ...\n"
    hostname_output="`docker run centos hostname`"
    expect_output 'docker ps -a' "$hostname_output"

}

run_tests() {
    test_get_centos_img
    test_centos_img_installed
    test_centos_img_runcmd
    test_centos_img_hostname_match_cont_id
}

# Main

enable_epel
setup_docker
run_tests