You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
3.5 KiB
87 lines
3.5 KiB
#!/usr/bin/env bash |
|
# Licensed to the Apache Software Foundation (ASF) under one or more |
|
# contributor license agreements. See the NOTICE file distributed with |
|
# this work for additional information regarding copyright ownership. |
|
# The ASF licenses this file to You under the Apache License, Version 2.0 |
|
# (the "License"); you may not use this file except in compliance with |
|
# the License. You may obtain a copy of the License at |
|
# |
|
# http://www.apache.org/licenses/LICENSE-2.0 |
|
# |
|
# Unless required by applicable law or agreed to in writing, software |
|
# distributed under the License is distributed on an "AS IS" BASIS, |
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
# See the License for the specific language governing permissions and |
|
# limitations under the License. |
|
|
|
# This script automates the process of setting up a local machine for running Kafka system tests |
|
export GREP_OPTIONS='--color=never' |
|
|
|
# Helper function which prints version numbers so they can be compared lexically or numerically |
|
function version { echo "$@" | awk -F. '{ printf("%03d%03d%03d%03d\n", $1,$2,$3,$4); }'; } |
|
|
|
base_dir=`dirname $0`/.. |
|
cd $base_dir |
|
|
|
echo "Checking Virtual Box installation..." |
|
bad_vb=false |
|
if [ -z `vboxmanage --version` ]; then |
|
echo "It appears that Virtual Box is not installed. Please install and try again (see https://www.virtualbox.org/ for details)" |
|
bad_vb=true |
|
else |
|
echo "Virtual Box looks good." |
|
fi |
|
|
|
echo "Checking Vagrant installation..." |
|
vagrant_version=`vagrant --version | egrep -o "[0-9]+\.[0-9]+\.[0-9]+"` |
|
bad_vagrant=false |
|
if [ "$(version $vagrant_version)" -lt "$(version 1.6.4)" ]; then |
|
echo "Found Vagrant version $vagrant_version. Please upgrade to 1.6.4 or higher (see http://www.vagrantup.com for details)" |
|
bad_vagrant=true |
|
else |
|
echo "Vagrant installation looks good." |
|
fi |
|
|
|
if [ "x$bad_vagrant" == "xtrue" -o "x$bad_vb" == "xtrue" ]; then |
|
exit 1 |
|
fi |
|
|
|
echo "Checking for necessary Vagrant plugins..." |
|
hostmanager_version=`vagrant plugin list | grep vagrant-hostmanager | egrep -o "[0-9]+\.[0-9]+\.[0-9]+"` |
|
if [ -z "$hostmanager_version" ]; then |
|
vagrant plugin install vagrant-hostmanager |
|
fi |
|
|
|
echo "Creating and packaging a reusable base box for Vagrant..." |
|
vagrant/package-base-box.sh |
|
|
|
# Set up Vagrantfile.local if necessary |
|
if [ ! -e Vagrantfile.local ]; then |
|
echo "Creating Vagrantfile.local..." |
|
cp vagrant/system-test-Vagrantfile.local Vagrantfile.local |
|
else |
|
echo "Found an existing Vagrantfile.local. Keeping without overwriting..." |
|
fi |
|
|
|
# Sanity check contents of Vagrantfile.local |
|
echo "Checking Vagrantfile.local..." |
|
vagrantfile_ok=true |
|
num_brokers=`egrep -o "num_brokers\s*=\s*[0-9]+" Vagrantfile.local | cut -d '=' -f 2 | xargs` |
|
num_zookeepers=`egrep -o "num_zookeepers\s*=\s*[0-9]+" Vagrantfile.local | cut -d '=' -f 2 | xargs` |
|
num_workers=`egrep -o "num_workers\s*=\s*[0-9]+" Vagrantfile.local | cut -d '=' -f 2 | xargs` |
|
if [ "x$num_brokers" == "x" -o "$num_brokers" != 0 ]; then |
|
echo "Vagrantfile.local: bad num_brokers. Update to: num_brokers = 0" |
|
vagrantfile_ok=false |
|
fi |
|
if [ "x$num_zookeepers" == "x" -o "$num_zookeepers" != 0 ]; then |
|
echo "Vagrantfile.local: bad num_zookeepers. Update to: num_zookeepers = 0" |
|
vagrantfile_ok=false |
|
fi |
|
if [ "x$num_workers" == "x" -o "$num_workers" == 0 ]; then |
|
echo "Vagrantfile.local: bad num_workers (size of test cluster). Set num_workers high enough to run your tests." |
|
vagrantfile_ok=false |
|
fi |
|
|
|
if [ "$vagrantfile_ok" == "true" ]; then |
|
echo "Vagrantfile.local looks good." |
|
fi
|
|
|