From 03f9f9065b5267fb09d0dcb148dd1481c330684d Mon Sep 17 00:00:00 2001
From: Dan Bode <dan@puppetlabs.com>
Date: Thu, 21 Mar 2013 13:11:08 -0700
Subject: [PATCH] Add conditional branches for unit testing

This commit adds conditional logic for running unit tests.
---
 test_scripts/openstack_test.sh | 62 +++++++++++++++++++---------------
 1 file changed, 35 insertions(+), 27 deletions(-)

diff --git a/test_scripts/openstack_test.sh b/test_scripts/openstack_test.sh
index 27c4978..892add7 100644
--- a/test_scripts/openstack_test.sh
+++ b/test_scripts/openstack_test.sh
@@ -5,7 +5,7 @@
 # the following build parameters:
 #   $operatingsystem - OS to test OS install on (accepts Redhat/Ubuntu)
 #   $openstack_version - openstack version to test (accepts folsom/grizzly)
-#   $test_mode type of test to run (accepts: tempest_full, tempest_smoke, puppet_openstack)
+#   $test_mode type of test to run (accepts: tempest_full, tempest_smoke, puppet_openstack, unit)
 #   $module_install_method - how to install modules (accepts librarian or pmt)
 #
 # # I am running it as follows:
@@ -18,18 +18,6 @@
 set -e
 set -u
 
-# set testing variables
-echo "operatingsystem: ${operatingsystem}" > config.yaml
-if [ $openstack_version = 'grizzly' ]; then
-  echo 'openstack_version: grizzly' > hiera_data/jenkins.yaml
-else
-  echo 'openstack_version: folsom' > hiera_data/jenkins.yaml
-fi
-
-mkdir .vendor
-export GEM_HOME=`pwd`/.vendor
-# install gem dependencies
-bundle install
 # install required modules
 if [ $module_install_method = 'librarian' ]; then
   bundle exec librarian-puppet install
@@ -38,25 +26,43 @@ elif [ $module_install_method = 'pmt' ]; then
   git clone https://github.com/ripienaar/hiera-puppet modules/hiera_puppet
   git clone git://github.com/puppetlabs/puppetlabs-swift modules/swift
   git clone git://github.com/puppetlabs/puppetlabs-tempest modules/tempest
+  git clone git://github.com/puppetlabs/puppetlabs-vcsrepo modules/vcsrepo
 fi
 # install a controller and compute instance
 
-# check that the VM is not currently running
-# if it is, stop that VM
-if VBoxManage list vms | grep openstack_controller.puppetlabs.lan; then
-  VBoxManage controlvm openstack_controller.puppetlabs.lan  poweroff || true
-  VBoxManage unregistervm openstack_controller.puppetlabs.lan  --delete
-fi
-bundle exec vagrant up openstack_controller
+# only build out integration test environment if we are not running unit tests
+if [ ! $test_mode = 'unit' ]; then
+  # set operatingsystem to use for integration tests tests
+  echo "operatingsystem: ${operatingsystem}" > config.yaml
+  if [ $openstack_version = 'grizzly' ]; then
+    echo 'openstack_version: grizzly' > hiera_data/jenkins.yaml
+  else
+    echo 'openstack_version: folsom' > hiera_data/jenkins.yaml
+  fi
 
-# check if the compute VM is running, if so stop the VM before launching ours
-if VBoxManage list vms | grep compute2.puppetlabs.lan; then
-  VBoxManage controlvm compute2.puppetlabs.lan  poweroff || true
-   VBoxManage unregistervm compute2.puppetlabs.lan --delete
+  # install gem dependencies
+  mkdir .vendor
+  export GEM_HOME=`pwd`/.vendor
+  # install gem dependencies
+  bundle install
+
+  # check that the VM is not currently running
+  # if it is, stop that VM
+  if VBoxManage list vms | grep openstack_controller.puppetlabs.lan; then
+    VBoxManage controlvm openstack_controller.puppetlabs.lan  poweroff || true
+    VBoxManage unregistervm openstack_controller.puppetlabs.lan  --delete
+  fi
+  bundle exec vagrant up openstack_controller
+
+  # check if the compute VM is running, if so stop the VM before launching ours
+  if VBoxManage list vms | grep compute2.puppetlabs.lan; then
+    VBoxManage controlvm compute2.puppetlabs.lan  poweroff || true
+     VBoxManage unregistervm compute2.puppetlabs.lan --delete
+  fi
+  bundle exec vagrant up compute2
+  # install tempest on the controller
+  bundle exec vagrant status
 fi
-bundle exec vagrant up compute2
-# install tempest on the controller
-bundle exec vagrant status
 
 if [ $test_mode = 'puppet_openstack' ]; then
   # run my simple puppet integration tests
@@ -68,6 +74,8 @@ elif [ $test_mode = 'tempest_smoke' ]; then
   bundle exec vagrant ssh -c 'cd /var/lib/tempest/;sudo ./jenkins_launch_script.sh --smoke;' openstack_controller
 elif [ $test_mode = 'tempest_full' ]; then
   bundle exec vagrant ssh -c 'cd /var/lib/tempest/;sudo ./jenkins_launch_script.sh;' openstack_controller
+elif [ $test_mode = 'unit' ]; then
+  echo 'we do not yet support running unit tests'
 else
   echo "Unsupported testnode ${test_mode}, this test matrix only support tempest_smoke and puppet_openstack tests"
 fi