#!/bin/bash
# Copyright 2015-2017 AT&T Intellectual Property, Inc
#  
# Licensed 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.
#

# What this is: An OpenStack Congress policy test. Sets up and validates policy
# creation and execution for:
# 1) Detecting that a reserved subnet has been created, by mistake. "Reserved"
#    in this example means e.g. not intended for use by VMs.
#
# Status: this is a work in progress, under test. 
#
# Prerequisite: 
# - OpenStack deployment with Congress service activated.
# - OpenStack CLI environment variables setup e.g. via admin-openrc.sh script.
# How to use:
#   # Create Congress policy and resources that exercise policy
#   $ bash reserved_subnet.sh
#   # After test, cleanup
#   $ bash reserved_subnet-clean.sh

pass() {
  echo "Hooray!"
  set +x #echo off
  exit 0
}

# Use this to trigger fail() at the right places
# if [ "$RESULT" == "Test Failed!" ]; then fail; fi
fail() {
  echo "Test Failed!"
  set +x
  exit 1
}

unclean() {
  echo "Unclean environment!"
  fail
}

if [  $# -eq 1 ]; then
  if [ $1 == "debug" ]; then 
    set -x #echo on
  fi
fi

echo "Create Congress policy 'test'"
if [[ $(openstack congress policy show test | awk "/ id / { print \$4 }") ]]; then unclean; fi
openstack congress policy create test

echo "Create reserved_subnet_error rules in policy 'test'"
openstack congress policy rule create test "reserved_subnet_error(x) :- neutronv2:subnets(id=x, cidr='10.7.1.0/24')" --name rsv_subnet_adm
openstack congress policy rule create test "reserved_subnet_error(x) :- neutronv2:subnets(id=x, cidr='10.7.12.0/24')" --name rsv_subnet_prv
openstack congress policy rule create test "reserved_subnet_error(x) :- neutronv2:subnets(id=x, cidr='10.7.13.0/24')" --name rsv_subnet_stg
openstack congress policy rule create test "reserved_subnet_error(x) :- neutronv2:subnets(id=x, cidr='10.7.14.0/24')" --name rsv_subnet_mgm

echo "Create external network"
if [[ $(neutron net-list | awk "/ test_public / { print \$2 }") ]]; then unclean; fi
neutron net-create test_public --router:external=true
#neutron net-create test_public --router:external=true --provider:network_type=flat --provider:physical_network=physnet1

echo "Create external subnet"
neutron subnet-create --disable-dhcp test_public 10.7.1.0/24 --name test_public

echo "Get the external subnet ID"
test_public_SUBNET=$(neutron subnet-list | awk "/ test_public / { print \$2 }")

echo "Create internal network"
if [[ $(neutron net-list | awk "/ test_internal / { print \$2 }") ]]; then unclean; fi
neutron net-create test_internal

echo "Create internal subnet"
neutron subnet-create test_internal 10.7.12.0/24 --name test_internal --gateway 10.7.12.1 --enable-dhcp --allocation-pool start=10.7.12.2,end=10.7.12.254 --dns-nameserver 8.8.8.8

echo "Get the internal subnet ID"
test_internal_SUBNET=$(neutron subnet-list | awk "/ test_internal / { print \$2 }")

echo "Verify test_public subnet ID is in the Congress policy 'test' table 'reserved_subnet_error'"
COUNTER=5
RESULT="Test Failed!"
until [[ $COUNTER -eq 0  || $RESULT == "Test Success!" ]]; do
  openstack congress policy row list test reserved_subnet_error
  test_public_ID=$(openstack congress policy row list test reserved_subnet_error | awk "/ $test_public_SUBNET / { print \$2 }")
  if [ "$test_public_SUBNET" == "$test_public_ID" ]; then RESULT="Test Success!"
  fi
  let COUNTER-=1
  sleep 5
done
echo "Verify test_public subnet ID is in the Congress policy 'test' table 'reserved_subnet_error':" $RESULT
if [ "$RESULT" == "Test Failed!" ]; then fail; fi

echo "Verify test_internal subnet ID is in the Congress policy 'test' table 'reserved_subnet_error'"
COUNTER=5
RESULT="Test Failed!"
until [[ $COUNTER -eq 0  || $RESULT == "Test Success!" ]]; do
  openstack congress policy row list test reserved_subnet_error
  test_internal_ID=$(openstack congress policy row list test reserved_subnet_error | awk "/ $test_internal_SUBNET / { print \$2 }")
  if [ "$test_internal_SUBNET" == "$test_internal_ID" ]; then RESULT="Test Success!"
  fi
  let COUNTER-=1
  sleep 5
done
echo "Verify test_internal subnet ID is in the Congress policy 'test' table 'reserved_subnet_error':" $RESULT
if [ "$RESULT" == "Test Failed!" ]; then fail; fi

echo "Create reactive 'deleted_reserved_subnet_error' rule in policy 'test'"
openstack congress policy rule create test "execute[neutronv2:delete_subnet(x)] :- reserved_subnet_error(x)" --name deleted_reserved_subnet_error

echo "Verify test_internal subnet is deleted"
COUNTER=5
RESULT="Test Failed!"
until [[ $COUNTER -eq 0  || $RESULT == "Test Success!" ]]; do
  neutron subnet-list 
  test_internal_ID=$(neutron subnet-list | awk "/ test_internal / { print \$2 }")
  if [ "$test_internal_SUBNET" != "$test_internal_ID" ]; then RESULT="Test Success!"
  fi
  let COUNTER-=1
  sleep 5
done
echo "Verify test_internal subnet is deleted:" $RESULT
if [ "$RESULT" == "Test Failed!" ]; then fail; fi

echo "Verify test_public subnet is deleted"
COUNTER=5
RESULT="Test Failed!"
until [[ $COUNTER -eq 0  || $RESULT == "Test Success!" ]]; do
  neutron subnet-list 
  test_public_ID=$(neutron subnet-list | awk "/ test_public / { print \$2 }")
  if [ "$test_public_SUBNET" != "$test_public_ID" ]; then RESULT="Test Success!"
  fi
  let COUNTER-=1
  sleep 5
done
echo "Verify test_public subnet is deleted:" $RESULT
if [ "$RESULT" == "Test Failed!" ]; then fail; fi
pass