#!/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