Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 1 | #!/bin/bash |
| 2 | # |
| 3 | # Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| 4 | # Use of this source code is governed by a BSD-style license that can be |
| 5 | # found in the LICENSE file. |
Simran Basi | 79b1a7b | 2013-02-13 11:16:00 -0800 | [diff] [blame] | 6 | set -e |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 7 | |
| 8 | USAGE="Usage: setup_dev_autotest.sh [-p <password>] [-a </path/to/autotest>]" |
| 9 | HELP="${USAGE}\n\n\ |
| 10 | Install and configure software needed to run autotest locally.\n\ |
| 11 | If you're just working on tests, you do not need to run this.\n\n\ |
| 12 | Options:\n\ |
| 13 | -p Desired Autotest DB password\n\ |
Michael Liang | b6a78f4 | 2014-07-11 15:05:30 -0700 | [diff] [blame] | 14 | -a Absolute path to autotest source tree.\n |
Jakob Juelich | 1c53ae7 | 2014-09-10 11:32:16 -0700 | [diff] [blame] | 15 | -v Show info logging from build_externals.py and compile_gwt_clients.py \n |
| 16 | -n Non-interactive mode, doesn't ask for any user input. |
| 17 | Requires -p and -a to be set." |
| 18 | |
| 19 | function get_y_or_n_interactive { |
| 20 | local ret |
| 21 | while true; do |
| 22 | read -p "$2" yn |
| 23 | case $yn in |
| 24 | [Yy]* ) ret="y"; break;; |
| 25 | [Nn]* ) ret="n"; break;; |
| 26 | * ) echo "Please enter y or n.";; |
| 27 | esac |
| 28 | done |
| 29 | eval $1="'$ret'" |
| 30 | } |
| 31 | |
| 32 | function get_y_or_n { |
| 33 | local ret=$3 |
| 34 | if [ "${noninteractive}" = "FALSE" ]; then |
| 35 | get_y_or_n_interactive sub "$2" |
| 36 | ret=$sub |
| 37 | fi |
| 38 | eval $1="'$ret'" |
| 39 | } |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 40 | |
| 41 | AUTOTEST_DIR= |
| 42 | PASSWD= |
Michael Liang | b6a78f4 | 2014-07-11 15:05:30 -0700 | [diff] [blame] | 43 | verbose="FALSE" |
Jakob Juelich | 1c53ae7 | 2014-09-10 11:32:16 -0700 | [diff] [blame] | 44 | noninteractive="FALSE" |
| 45 | while getopts ":p:a:nvh" opt; do |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 46 | case $opt in |
| 47 | a) |
| 48 | AUTOTEST_DIR=$OPTARG |
| 49 | ;; |
| 50 | p) |
| 51 | PASSWD=$OPTARG |
| 52 | ;; |
Michael Liang | b6a78f4 | 2014-07-11 15:05:30 -0700 | [diff] [blame] | 53 | v) |
| 54 | verbose="TRUE" |
| 55 | ;; |
Jakob Juelich | 1c53ae7 | 2014-09-10 11:32:16 -0700 | [diff] [blame] | 56 | n) |
| 57 | noninteractive="TRUE" |
| 58 | ;; |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 59 | h) |
| 60 | echo -e "${HELP}" >&2 |
| 61 | exit 0 |
| 62 | ;; |
| 63 | \?) |
| 64 | echo "Invalid option: -$OPTARG" >&2 |
| 65 | echo "${USAGE}" >&2 |
| 66 | exit 1 |
| 67 | ;; |
| 68 | :) |
| 69 | echo "Option -$OPTARG requires an argument." >&2 |
| 70 | echo "${USAGE}" >&2 |
| 71 | exit 1 |
| 72 | ;; |
| 73 | esac |
| 74 | done |
| 75 | |
Jakob Juelich | 1c53ae7 | 2014-09-10 11:32:16 -0700 | [diff] [blame] | 76 | |
| 77 | if [ "${noninteractive}" = "TRUE" ]; then |
| 78 | if [ -z "${AUTOTEST_DIR}" ]; then |
| 79 | echo "-a must be specified in non-interactive mode." >&2 |
| 80 | exit 1 |
| 81 | fi |
| 82 | if [ -z "${PASSWD}" ]; then |
| 83 | echo "-p must be specified in non-interactive mode." >&2 |
| 84 | exit 1 |
| 85 | fi |
| 86 | fi |
| 87 | |
| 88 | |
Richard Barnette | 23ef442 | 2012-11-02 13:46:20 -0700 | [diff] [blame] | 89 | if [ -z "${PASSWD}" ]; then |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 90 | read -s -p "Autotest DB password: " PASSWD |
| 91 | echo |
Richard Barnette | 23ef442 | 2012-11-02 13:46:20 -0700 | [diff] [blame] | 92 | if [ -z "${PASSWD}" ]; then |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 93 | echo "Empty passwords not allowed." >&2 |
| 94 | exit 1 |
| 95 | fi |
| 96 | read -s -p "Re-enter password: " PASSWD2 |
| 97 | echo |
Richard Barnette | 23ef442 | 2012-11-02 13:46:20 -0700 | [diff] [blame] | 98 | if [ "${PASSWD}" != "${PASSWD2}" ]; then |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 99 | echo "Passwords don't match." >&2 |
| 100 | exit 1 |
| 101 | fi |
| 102 | fi |
| 103 | |
| 104 | if [ -z "${AUTOTEST_DIR}" ]; then |
| 105 | CANDIDATE=$(dirname "$(readlink -f "$0")" | egrep -o '(/[^/]+)*/files') |
| 106 | read -p "Enter autotest dir [${CANDIDATE}]: " AUTOTEST_DIR |
| 107 | if [ -z "${AUTOTEST_DIR}" ]; then |
| 108 | AUTOTEST_DIR="${CANDIDATE}" |
| 109 | fi |
| 110 | fi |
| 111 | |
Aviv Keshet | 2c603d8 | 2013-05-24 10:31:59 -0700 | [diff] [blame] | 112 | |
| 113 | # Sanity check AUTOTEST_DIR. If it's null, or doesn't exist on the filesystem |
| 114 | # then die. |
| 115 | if [ -z "${AUTOTEST_DIR}" ]; then |
| 116 | echo "No AUTOTEST_DIR. Aborting script." |
| 117 | exit 1 |
| 118 | fi |
| 119 | |
| 120 | if [ ! -d "${AUTOTEST_DIR}" ]; then |
| 121 | echo "Directory " ${AUTOTEST_DIR} " does not exist. Aborting script." |
| 122 | exit 1 |
| 123 | fi |
| 124 | |
| 125 | |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 126 | SHADOW_CONFIG_PATH="${AUTOTEST_DIR}/shadow_config.ini" |
| 127 | echo "Autotest supports local overrides of global configuration through a " |
| 128 | echo "'shadow' configuration file. Setting one up for you now." |
Chris Sosa | 18c70b3 | 2013-02-15 14:12:43 -0800 | [diff] [blame] | 129 | CLOBBER=0 |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 130 | if [ -f ${SHADOW_CONFIG_PATH} ]; then |
Jakob Juelich | 1c53ae7 | 2014-09-10 11:32:16 -0700 | [diff] [blame] | 131 | get_y_or_n clobber "Clobber existing shadow config? [Y/n]: " "n" |
| 132 | if [[ "${clobber}" = 'n' ]]; then |
Chris Sosa | 18c70b3 | 2013-02-15 14:12:43 -0800 | [diff] [blame] | 133 | CLOBBER=1 |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 134 | echo "Refusing to clobber existing shadow_config.ini." |
Chris Sosa | 18c70b3 | 2013-02-15 14:12:43 -0800 | [diff] [blame] | 135 | else |
| 136 | echo "Clobbering existing shadow_config.ini." |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 137 | fi |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 138 | fi |
| 139 | |
Aviv Keshet | 2c603d8 | 2013-05-24 10:31:59 -0700 | [diff] [blame] | 140 | CROS_CHECKOUT=$(readlink -f ${AUTOTEST_DIR}/../../../..) |
| 141 | |
Chris Sosa | 18c70b3 | 2013-02-15 14:12:43 -0800 | [diff] [blame] | 142 | # Create clean shadow config if we're replacing it/creating a new one. |
| 143 | if [ $CLOBBER -eq 0 ]; then |
| 144 | cat > "${SHADOW_CONFIG_PATH}" <<EOF |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 145 | [AUTOTEST_WEB] |
| 146 | host: localhost |
| 147 | password: ${PASSWD} |
Chris Masone | e61dd81 | 2012-01-10 18:04:21 -0800 | [diff] [blame] | 148 | readonly_host: localhost |
| 149 | readonly_user: chromeosqa-admin |
| 150 | readonly_password: ${PASSWD} |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 151 | |
| 152 | [SERVER] |
| 153 | hostname: localhost |
Chris Sosa | e418124 | 2012-09-10 16:54:30 -0700 | [diff] [blame] | 154 | |
| 155 | [SCHEDULER] |
| 156 | drones: localhost |
Aviv Keshet | 2c603d8 | 2013-05-24 10:31:59 -0700 | [diff] [blame] | 157 | |
| 158 | [CROS] |
| 159 | source_tree: ${CROS_CHECKOUT} |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 160 | EOF |
Chris Sosa | 18c70b3 | 2013-02-15 14:12:43 -0800 | [diff] [blame] | 161 | echo -e "Done!\n" |
| 162 | fi |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 163 | |
| 164 | echo "Installing needed Ubuntu packages..." |
Scott Zawalski | a5745e9 | 2013-05-09 10:42:27 -0400 | [diff] [blame] | 165 | PKG_LIST="mysql-server mysql-common libapache2-mod-wsgi python-mysqldb \ |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 166 | gnuplot apache2-mpm-prefork unzip python-imaging libpng12-dev libfreetype6-dev \ |
| 167 | sqlite3 python-pysqlite2 git-core pbzip2 openjdk-6-jre openjdk-6-jdk \ |
Simran Basi | 3d18208 | 2012-09-07 10:15:25 -0700 | [diff] [blame] | 168 | python-crypto python-dev subversion build-essential python-setuptools \ |
| 169 | python-numpy python-scipy" |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 170 | |
| 171 | if ! sudo apt-get install -y ${PKG_LIST}; then |
| 172 | echo "Could not install packages: $?" |
| 173 | exit 1 |
| 174 | fi |
| 175 | echo -e "Done!\n" |
| 176 | |
Dan Shi | 7f0c183 | 2014-10-27 16:05:57 -0700 | [diff] [blame^] | 177 | # Check if database exists, clobber existing database with user consent. |
| 178 | # |
| 179 | # Arguments: Name of the database |
| 180 | check_database() |
| 181 | { |
| 182 | local db_name=$1 |
| 183 | echo "Setting up Database: $db_name in MySQL..." |
| 184 | if mysql -u root -e ';' 2> /dev/null ; then |
| 185 | PASSWD_STRING= |
| 186 | elif mysql -u root -p"${PASSWD}" -e ';' 2> /dev/null ; then |
| 187 | PASSWD_STRING="-p${PASSWD}" |
| 188 | else |
| 189 | PASSWD_STRING="-p" |
| 190 | fi |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 191 | |
Dan Shi | 7f0c183 | 2014-10-27 16:05:57 -0700 | [diff] [blame^] | 192 | if ! mysqladmin ping ; then |
| 193 | sudo service mysql start |
| 194 | fi |
Simran Basi | 79b1a7b | 2013-02-13 11:16:00 -0800 | [diff] [blame] | 195 | |
Dan Shi | 7f0c183 | 2014-10-27 16:05:57 -0700 | [diff] [blame^] | 196 | local clobberdb='y' |
| 197 | local existing_database=$(mysql -u root "${PASSWD_STRING}" -e "SELECT \ |
| 198 | SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = '$db_name'") |
Jakob Juelich | 1c53ae7 | 2014-09-10 11:32:16 -0700 | [diff] [blame] | 199 | |
Dan Shi | 7f0c183 | 2014-10-27 16:05:57 -0700 | [diff] [blame^] | 200 | if [ -n "${existing_database}" ]; then |
| 201 | get_y_or_n clobberdb "Clobber existing MySQL database? [Y/n]: " "n" |
| 202 | fi |
Alex Miller | da3ecf4 | 2012-07-25 11:51:13 -0700 | [diff] [blame] | 203 | |
Dan Shi | 7f0c183 | 2014-10-27 16:05:57 -0700 | [diff] [blame^] | 204 | local sql_command="drop database if exists $db_name; \ |
| 205 | create database $db_name; \ |
| 206 | grant all privileges on $db_name.* TO \ |
| 207 | 'chromeosqa-admin'@'localhost' identified by '${PASSWD}'; \ |
| 208 | FLUSH PRIVILEGES;" |
Alex Miller | da3ecf4 | 2012-07-25 11:51:13 -0700 | [diff] [blame] | 209 | |
Dan Shi | 7f0c183 | 2014-10-27 16:05:57 -0700 | [diff] [blame^] | 210 | if [[ "${clobberdb}" = 'y' ]]; then |
| 211 | mysql -u root "${PASSWD_STRING}" -e "${sql_command}" |
| 212 | fi |
| 213 | echo -e "Done!\n" |
| 214 | } |
| 215 | |
| 216 | check_database 'chromeos_autotest_db' |
| 217 | check_database 'chromeos_lab_servers' |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 218 | |
| 219 | AT_DIR=/usr/local/autotest |
| 220 | echo -n "Bind-mounting your autotest dir at ${AT_DIR}..." |
| 221 | sudo mkdir -p "${AT_DIR}" |
| 222 | sudo mount --bind "${AUTOTEST_DIR}" "${AT_DIR}" |
| 223 | echo -e "Done!\n" |
| 224 | |
| 225 | EXISTING_MOUNT=$(egrep "/.+[[:space:]]${AT_DIR}" /etc/fstab || /bin/true) |
| 226 | if [ -n "${EXISTING_MOUNT}" ]; then |
| 227 | echo "${EXISTING_MOUNT}" | awk '{print $1 " already automounting at " $2}' |
| 228 | echo "We won't update /etc/fstab, but you should have a line line this:" |
| 229 | echo -e "${AUTOTEST_DIR}\t${AT_DIR}\tbind defaults,bind\t0\t0" |
| 230 | else |
| 231 | echo -n "Adding aforementioned bind-mount to /etc/fstab..." |
| 232 | # Is there a better way to elevate privs and do a redirect? |
| 233 | sudo su -c \ |
| 234 | "echo -e '${AUTOTEST_DIR}\t${AT_DIR}\tbind defaults,bind\t0\t0' \ |
| 235 | >> /etc/fstab" |
| 236 | echo -e "Done!\n" |
| 237 | fi |
| 238 | |
| 239 | echo -n "Reticulating splines..." |
Michael Liang | b6a78f4 | 2014-07-11 15:05:30 -0700 | [diff] [blame] | 240 | |
| 241 | if [ "${verbose}" = "TRUE" ]; then |
| 242 | "${AT_DIR}"/utils/build_externals.py |
| 243 | "${AT_DIR}"/utils/compile_gwt_clients.py -a |
| 244 | else |
| 245 | "${AT_DIR}"/utils/build_externals.py &> /dev/null |
| 246 | "${AT_DIR}"/utils/compile_gwt_clients.py -a &> /dev/null |
| 247 | fi |
| 248 | |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 249 | echo -e "Done!\n" |
| 250 | |
| 251 | echo "Populating autotest mysql DB..." |
Jakob Juelich | 1c53ae7 | 2014-09-10 11:32:16 -0700 | [diff] [blame] | 252 | "${AT_DIR}"/database/migrate.py sync -f |
| 253 | "${AT_DIR}"/frontend/manage.py syncdb --noinput |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 254 | # You may have to run this twice. |
Jakob Juelich | 1c53ae7 | 2014-09-10 11:32:16 -0700 | [diff] [blame] | 255 | "${AT_DIR}"/frontend/manage.py syncdb --noinput |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 256 | "${AT_DIR}"/utils/test_importer.py |
| 257 | echo -e "Done!\n" |
| 258 | |
Dan Shi | 7f0c183 | 2014-10-27 16:05:57 -0700 | [diff] [blame^] | 259 | echo "Initializing chromeos_lab_servers mysql DB..." |
| 260 | "${AT_DIR}"/database/migrate.py sync -f -d AUTOTEST_SERVER_DB |
| 261 | echo -e "Done!\n" |
| 262 | |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 263 | echo "Configuring apache to run the autotest web interface..." |
Simran Basi | 508ee92 | 2013-11-18 13:43:40 -0800 | [diff] [blame] | 264 | if [ ! -d /etc/apache2/run ]; then |
| 265 | sudo mkdir /etc/apache2/run |
| 266 | fi |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 267 | sudo ln -sf "${AT_DIR}"/apache/apache-conf \ |
| 268 | /etc/apache2/sites-available/autotest-server |
| 269 | # disable currently active default |
| 270 | sudo a2dissite default |
| 271 | # enable autotest server |
| 272 | sudo a2ensite autotest-server |
| 273 | # Enable rewrite module |
| 274 | sudo a2enmod rewrite |
Scott Zawalski | a5745e9 | 2013-05-09 10:42:27 -0400 | [diff] [blame] | 275 | # Enable wsgi |
| 276 | sudo a2enmod wsgi |
| 277 | # enable version |
| 278 | sudo a2enmod version |
Dan Shi | a85f6d8 | 2014-10-01 08:58:01 -0700 | [diff] [blame] | 279 | # enable headers |
| 280 | sudo a2enmod headers |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 281 | # Setup permissions so that Apache web user can read the proper files. |
| 282 | chmod -R o+r "${AT_DIR}" |
Alex Miller | 5e7b254 | 2013-02-13 14:24:04 -0800 | [diff] [blame] | 283 | find "${AT_DIR}"/ -type d -print0 | xargs --null chmod o+x |
Chris Masone | 2aa5533 | 2011-12-30 14:41:07 -0800 | [diff] [blame] | 284 | chmod o+x "${AT_DIR}"/tko/*.cgi |
| 285 | # restart server |
| 286 | sudo /etc/init.d/apache2 restart |
| 287 | |
| 288 | echo "Browse to http://localhost to see if Autotest is working." |
| 289 | echo "For further necessary set up steps, see https://sites.google.com/a/chromium.org/dev/chromium-os/testing/autotest-developer-faq/setup-autotest-server?pli=1" |