#!/bin/bash -x
echo "Testing $1"

RES=0
SERVICE="cinder"

function prepare_vm(){
  if [[ `cat /etc/*-release | head -n 1 | awk '{print $1}'` =~ Ubuntu ]]
  then
      echo "mysql-server mysql-server/root_password select $MYSQL_PASSWORD" | sudo debconf-set-selections
      echo "mysql-server mysql-server/root_password_again select $MYSQL_PASSWORD" | sudo debconf-set-selections
      apt-get install -y mysql-server python-mysqldb rabbitmq-server curl screen
  else
      yum install net-tools mysql-server mysql-wsrep-client-5.6 MySQL-python -y

      # workaround for `sqlalchemy.exc.OperationalError: (_mysql_exceptions.OperationalError)
      # (2002, "Can't connect to local MySQL server through socket '/var/lib/mysql/mysql.sock' (2)")`
      rm -f /etc/my.cnf
      MYSQL_PASSWORD=$(cat /root/.mysql_secret | head -1 | awk -F': ' '{print $2}')
      service mysql start
      /usr/bin/mysqladmin -u root -p$MYSQL_PASSWORD password $SERVICE
  fi
  mysql -uroot -p$SERVICE <<MYSQL_SCRIPT 
CREATE DATABASE cinder;
CREATE USER 'cinder'@'localhost' IDENTIFIED BY 'cinder';
GRANT ALL PRIVILEGES ON * . * TO 'cinder'@'localhost';
FLUSH PRIVILEGES;
MYSQL_SCRIPT
}

function test_cinder_service(){
  echo "Starting Cinder $1..."
  service $1 start
  sleep 5

  service_result=0
  service_running=`netstat -nat | grep $2 | wc -l`
  if [ $service_running -eq 0 ]; then
    echo "Can not start Cinder $1 service"
    tail -n 40 /var/log/messages
    RES=1
  else
    echo "Cinder $1 successfully started"
    service $1 stop
  fi
}


function check_file_exists(){
  if [ ! -f "$1" ]; then
    RES=1
  fi
}


function test_cinder_configs(){
  declare -a files=("/etc/cinder/cinder.conf" "/etc/cinder/api-paste.ini" "/etc/cinder/policy.json")
  for f in "${files[@]}"
  do
    check_file_exists "$f"
 done
}


case $1 in
  python-cinder)
    declare -a imports=("cinder" "cinder.volume.drivers.lvm" "cinder.volume.drivers.rbd")
    for i in "${imports[@]}"
    do
      python -c "import $i"
      if [ $? -ne 0 ]; then
        RES=1
        break
      fi
    done
  ;;
  openstack-cinder)
    prepare_vm
    test_cinder_configs

    count=`ls -1 /usr/bin/cinder* 2>/dev/null | wc -l`
    if [ $count -eq 0 ]; then
        RES=1
    fi

    test_cinder_service openstack-cinder-api 8776
    check_file_exists "/usr/bin/cinder-volume"
    check_file_exists "/usr/bin/cinder-scheduler"
  ;;
  cinder-common)
    test_cinder_configs
    check_file_exists "/usr/bin/cinder-manage"
  ;;
  cinder-scheduler)
    check_file_exists "/usr/bin/cinder-scheduler"
  ;;
  cinder-api)
    check_file_exists "/usr/bin/cinder-api"
    test_cinder_service cinder-api 8776
  ;;
  cinder-volume)
    check_file_exists "/usr/bin/cinder-volume"
  ;;
  openstack-cinder-doc)
    echo "No test for docs defined"
  ;;
  *)
    echo "Test is not defined, skipping..."
  ;;
esac

exit $RES