Use Docker Machine for Test Environments

Usage:

   ./machines up [consul|etcd|zookeeper] [num_nodes]
   ./machines destroy

Uses Docker Machine to create test environments
We _could_ use these environments to run BATS tests against
This would ensure that all supported backends are working

Signed-off-by: Dave Tucker <dt@docker.com>
This commit is contained in:
Dave Tucker 2015-11-05 17:42:48 +00:00
parent 5305ea570b
commit b7d09b785f
1 changed files with 111 additions and 0 deletions

111
machines Executable file
View File

@ -0,0 +1,111 @@
#/bin/sh
set -e
usage()
{
cat << EOF
NAME:
machines - Create Test Environments for Docker Networking
VERSION:
0.1
USAGE:
$0 <command> [command_options] [arguments...]
COMMANDS:
help
Help and usage
up <kv-store> <scale>
Create environment with given KV store
zookeeper | etcd | consul (default)
Create N nodes, default = 2
destroy
Destroy Environment
EOF
}
step() {
printf "\033[0;36m-----> $@\033[0m\n"
}
up()
{
step "Creating KV Store Machine"
docker-machine create \
-d virtualbox \
mh-kv
step "KV Store is $1"
step "Starting KV Container"
case "$1" in
etcd)
cluster_store="cluster-store=etcd://$(docker-machine ip mh-kv):2379"
docker $(docker-machine config mh-kv) run -d \
-p "2379:2379" \
-h "etcd" \
--name "etcd" \
quay.io/coreos/etcd:v2.2.1 \
--listen-client-urls="http://0.0.0.0:2379" \
--advertise-client-urls="http://$(docker-machine ip mh-kv):2379"
;;
zookeeper)
cluster_store="cluster-store=zk://$(docker-machine ip mh-kv):2181"
docker $(docker-machine config mh-kv) run -d \
-p "2181:2181" \
-h "zookeeper" \
--name "zookeeper" \
tianon/zookeeper
;;
*)
cluster_store="cluster-store=consul://$(docker-machine ip mh-kv):8500"
docker $(docker-machine config mh-kv) run -d \
-p "8500:8500" \
-h "consul" \
--name "consul" \
progrium/consul -server -bootstrap-expect 1
;;
esac
machines=$2
if [ -z machines ]; then
machines=2
fi
step "Creating $machines Machines"
for i in $(seq $machines); do
step "Creating machine $i"
docker-machine create \
-d virtualbox \
--engine-opt="cluster-advertise=eth1:2376" \
--engine-opt="$cluster_store" \
mh-$i
done
}
destroy()
{
for x in $(docker-machine ls | grep mh- | awk '{ print $1 }'); do
docker-machine rm $x
done
}
case "$1" in
up)
shift
up $@
;;
destroy)
destroy $@
;;
help)
usage
;;
*)
usage
;;
esac