From b2f81bd4105f0ba460afd8fc0c3fd362463ba214 Mon Sep 17 00:00:00 2001
From: bhavani <bhavani.cr831@gmail.com>
Date: Wed, 15 Mar 2017 10:19:23 +0530
Subject: [PATCH] Prepare for using standard python tests

Add simple script to setup mysql and postgresql databases, this script
can be run by users during testing and will be run by CI systems for
specific setup before running unit tests. This is exactly what is
currently done by OpenStack CI in project-config.

This allows to change in project-config the python-db jobs to
python-jobs since python-jobs will call this script initially.

See also
http://lists.openstack.org/pipermail/openstack-dev/2016-November/107784.html

Change-Id: I11fd4c8b0291779a2f2dd1526b8b33b2885319ee
---
 tools/test-setup.sh | 61 +++++++++++++++++++++++++++++++++++++++------
 1 file changed, 53 insertions(+), 8 deletions(-)

diff --git a/tools/test-setup.sh b/tools/test-setup.sh
index 8c8c32634..e703ff609 100755
--- a/tools/test-setup.sh
+++ b/tools/test-setup.sh
@@ -1,13 +1,58 @@
-#!/usr/bin/env bash
+#!/bin/bash -xe
 
-# Borrowed from mysql-prep in openstack-infra/project-config/macros.yaml
-DB_ROOT_PW=insecure_slave
+# This script will be run by OpenStack CI before unit tests are run,
+# it sets up the test system as needed.
+# Developers should setup their test systems in a similar way.
+
+# This setup needs to be run as a user that can run sudo.
+
+# The root password for the MySQL database; pass it in via
+# MYSQL_ROOT_PW.
+DB_ROOT_PW=${MYSQL_ROOT_PW:-insecure_slave}
+
+# This user and its password are used by the tests, if you change it,
+# your tests might fail.
 DB_USER=openstack_citest
 DB_PW=openstack_citest
+
 sudo -H mysqladmin -u root password $DB_ROOT_PW
-# note; we remove anonymous users first
+
+# It's best practice to remove anonymous users from the database.  If
+# a anonymous user exists, then it matches first for connections and
+# other connections from that host will not work.
 sudo -H mysql -u root -p$DB_ROOT_PW -h localhost -e "
-DELETE FROM mysql.user WHERE User='';
-FLUSH PRIVILEGES;
-GRANT ALL PRIVILEGES ON *.*
-TO '$DB_USER'@'%' identified by '$DB_PW' WITH GRANT OPTION;"
+    DELETE FROM mysql.user WHERE User='';
+    FLUSH PRIVILEGES;
+    GRANT ALL PRIVILEGES ON *.*
+        TO '$DB_USER'@'%' identified by '$DB_PW' WITH GRANT OPTION;"
+
+# Now create our database.
+mysql -u $DB_USER -p$DB_PW -h 127.0.0.1 -e "
+    SET default_storage_engine=MYISAM;
+    DROP DATABASE IF EXISTS openstack_citest;
+    CREATE DATABASE openstack_citest CHARACTER SET utf8;"
+
+# Same for PostgreSQL
+# The root password for the PostgreSQL database; pass it in via
+# POSTGRES_ROOT_PW.
+DB_ROOT_PW=${POSTGRES_ROOT_PW:-insecure_slave}
+
+# Setup user
+root_roles=$(sudo -H -u postgres psql -t -c "
+   SELECT 'HERE' from pg_roles where rolname='$DB_USER'")
+if [[ ${root_roles} == *HERE ]];then
+    sudo -H -u postgres psql -c "ALTER ROLE $DB_USER WITH SUPERUSER LOGIN PASSWORD '$DB_PW'"
+else
+    sudo -H -u postgres psql -c "CREATE ROLE $DB_USER WITH SUPERUSER LOGIN PASSWORD '$DB_PW'"
+fi
+
+# Store password for tests
+cat << EOF > $HOME/.pgpass
+*:*:*:$DB_USER:$DB_PW
+EOF
+chmod 0600 $HOME/.pgpass
+
+# Now create our database
+psql -h 127.0.0.1 -U $DB_USER -d template1 -c "DROP DATABASE IF EXISTS openstack_citest"
+createdb -h 127.0.0.1 -U $DB_USER -l C -T template0 -E utf8 openstack_citest
+