82 lines
2 KiB
Bash
Executable file
82 lines
2 KiB
Bash
Executable file
set -e
|
|
|
|
up_mysql() {
|
|
# Run a SQL database on port 3306 inside of Docker.
|
|
docker run --name mysql -p 3306:3306 -e MYSQL_ROOT_PASSWORD=password -d mysql
|
|
|
|
# Sleep for 5s to get MySQL get started.
|
|
echo 'Sleeping for 5...'
|
|
sleep 5
|
|
|
|
# Add the database to mysql.
|
|
docker run --rm --link mysql:mysql mysql sh -c 'echo "create database genschema" | mysql -h"$MYSQL_PORT_3306_TCP_ADDR" -P"$MYSQL_PORT_3306_TCP_PORT" -uroot -ppassword'
|
|
}
|
|
|
|
down_mysql() {
|
|
docker kill mysql
|
|
docker rm mysql
|
|
}
|
|
|
|
up_postgres() {
|
|
# Run a SQL database on port 5432 inside of Docker.
|
|
docker run --name postgres -p 5432:5432 -d postgres
|
|
|
|
# Sleep for 5s to get SQL get started.
|
|
echo 'Sleeping for 5...'
|
|
sleep 5
|
|
|
|
# Add the database to postgres.
|
|
docker run --rm --link postgres:postgres postgres sh -c 'echo "create database genschema" | psql -h "$POSTGRES_PORT_5432_TCP_ADDR" -p "$POSTGRES_PORT_5432_TCP_PORT" -U postgres'
|
|
}
|
|
|
|
down_postgres() {
|
|
docker kill postgres
|
|
docker rm postgres
|
|
}
|
|
|
|
gen_migrate() {
|
|
# Generate a SQLite database with the schema as defined by the existing alembic model.
|
|
GENMIGRATE=$1 PYTHONPATH=. alembic upgrade head
|
|
|
|
# Generate the migration to the current model.
|
|
GENMIGRATE=$1 PYTHONPATH=. alembic revision --autogenerate -m "$2"
|
|
}
|
|
|
|
test_migrate() {
|
|
# Generate a SQLite database with the schema as defined by the existing alembic model.
|
|
GENMIGRATE=$1 PYTHONPATH=. alembic upgrade head
|
|
|
|
# Downgrade to verify it works in both directions.
|
|
COUNT=`ls data/migrations/versions/*.py | wc -l | tr -d ' '`
|
|
GENMIGRATE=$1 PYTHONPATH=. alembic downgrade "-$COUNT"
|
|
}
|
|
|
|
# Test (and generate, if requested) via MySQL.
|
|
echo '> Starting MySQL'
|
|
up_mysql
|
|
|
|
if [ ! -z "$@" ]
|
|
then
|
|
set +e
|
|
echo '> Generating Migration'
|
|
gen_migrate "mysql" "$@"
|
|
set -e
|
|
fi
|
|
|
|
echo '> Testing Migration (mysql)'
|
|
set +e
|
|
test_migrate "mysql"
|
|
set -e
|
|
down_mysql
|
|
|
|
# Test via Postgres.
|
|
echo '> Starting Postgres'
|
|
up_postgres
|
|
|
|
echo '> Testing Migration (postgres)'
|
|
set +e
|
|
test_migrate "postgres"
|
|
set -e
|
|
down_postgres
|
|
|
|
|