Synnefo » snf-ganeti

#!/bin/bash

#

# Copyright (C) 2012 Google Inc.

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

# General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

set -e -u -o pipefail

# Disable any locales

export LC_ALL=C

readonly bash_completion=${TOP_BUILDDIR:-.}/doc/examples/bash_completion-debug

readonly default_wordbreaks=$' \t\n"'\''@><=;|&(:'

err() {

  echo "$@"

  echo 'Aborting'

  exit 1

}

contains() {

  local -r needle="$1"; shift

  for value; do

    if [[ "$value" = "$needle" ]]; then

      return 0

    fi

  done

  return 1

}

# Prepares a subshell for testing bash completion functions

setup() {

  local -r unused=UNUSED

  set +e +u -o pipefail

  . $bash_completion

  COMP_KEY=$unused

  COMP_LINE=$unused

  COMP_POINT=$unused

  COMP_TYPE=$unused

  COMP_WORDBREAKS="$default_wordbreaks"

  GANETI_COMPL_LOG=

  unset COMP_CWORD

  unset COMP_WORDS

}

# Check if default wordbreaks are still valid (this detects cases where Bash

# were to change its built-in default value)

# TODO: May need an update for older Bash versions; some didn't include the

# colon character (':') in COMP_WORDBREAKS

(

  bashdef=$(env - bash --noprofile --norc -c 'echo -n "$COMP_WORDBREAKS"')

  case "$bashdef" in

    $default_wordbreaks) ;;

    *)

      err 'Bash uses unknown value for COMP_WORDBREAKS'

      ;;

  esac

)

# Check for --help

for cmd in gnt-{instance,node,group,job}; do

  (

    setup

    COMP_CWORD=2

    COMP_WORDS=( $cmd list - )

    _${cmd/-/_}

    contains --help "${COMPREPLY[@]}" || \

      err "'$cmd list' did not list --help as an option"

  )

done

# Completing a yes/no option

(

  setup

  COMP_CWORD=3

  COMP_WORDS=( gnt-node modify --drained )

  _gnt_node

  if [[ "${COMPREPLY[*]}" != 'no yes' ]]; then

    err "Completing '${COMP_WORDS[@]}' did not give correct result"

  fi

)

# Completing a multiple-choice option

(

  setup

  COMP_CWORD=2

  COMP_WORDS=( gnt-debug allocator --disk-template=sh foo )

  _gnt_debug

  if [[ "${COMPREPLY[*]}" != sharedfile ]]; then

    err "Completing '${COMP_WORDS[*]}' did not give correct result"

  fi

)

# Completing a node name

(

  setup

  # Override built-in function

  _ganeti_nodes() {

    echo aanode1 bbnode2 aanode3

  }

  COMP_CWORD=4

  COMP_WORDS=( gnt-node modify --drained yes aa )

  _gnt_node

  if [[ "${COMPREPLY[*]}" != 'aanode1 aanode3' ]]; then

    err 'Completing node names failed'

  fi

)

# Completing an option when it's not at the end

(

  setup

  # Override built-in function

  _ganeti_instances() {

    echo inst{1..5}

  }

  # Completing word in the middle

  COMP_CWORD=2

  COMP_WORDS=( gnt-instance list --o inst3 inst inst5 )

  _gnt_node

  contains --output "${COMPREPLY[@]}" || err 'Did not complete parameter'

)

# Completing an instance name

(

  setup

  # Override built-in function

  _ganeti_instances() {

    echo inst{1..5}

  }

  # Completing word in the middle

  COMP_CWORD=5

  COMP_WORDS=( gnt-instance list -o foobar inst1 inst inst5 )

  _gnt_instance

  if [[ "${COMPREPLY[*]}" != "$(echo inst{1..5})" ]]; then

    err "Completing '${COMP_WORDS[*]}' did not give correct result"

  fi

)

# Need to test node expansion with different wordbreak settings

[[ "$default_wordbreaks" == *:* ]] || \

  err 'No colon in default wordbreak characters'

for wb in "$default_wordbreaks" "${default_wordbreaks/:/}"; do

  (

    setup

    # Override built-in function

    _ganeti_nodes() {

      echo node{A..C}

    }

    COMP_WORDBREAKS="$wb"

    # Completing nodes

    COMP_CWORD=3

    COMP_WORDS=( gnt-instance add -n )

    _gnt_instance

    if [[ "${COMPREPLY[*]}" != 'nodeA nodeA: nodeB nodeB: nodeC nodeC:' ]]; then

      err 'Got wrong node list'

    fi

    COMP_CWORD=3

    COMP_WORDS=( gnt-instance add -n nodeB )

    _gnt_instance

    if [[ "${COMPREPLY[*]}" != 'nodeB nodeB:' ]]; then

      err 'Got wrong node list'

    fi

    COMP_CWORD=3

    COMP_WORDS=( gnt-instance add -n nodeC: )

    _gnt_instance

    if [[ "$COMP_WORDBREAKS" == *:* ]]; then

      expected='nodeA nodeB'

    else

      expected='nodeC:nodeA nodeC:nodeB'

    fi

    if [[ "${COMPREPLY[*]}" != "$expected" ]]; then

      err 'Got wrong node list'

    fi

  )

done

# Need to test different settings for the extglob shell option

for opt in -u -s; do

  verify_extglob() {

    if [[ "$(shopt -p extglob)" != "shopt $opt extglob" ]]; then

      err 'The "extglob" shell option has an unexpected value'

    fi

  }

  (

    shopt $opt extglob

    verify_extglob

    setup

    verify_extglob

    # Completing nodes

    COMP_CWORD=4

    COMP_WORDS=( gnt-instance add --os-type busybox --no-n )

    _gnt_instance

    if [[ "${COMPREPLY[*]}" != '--no-name-check --no-nics' ]]; then

      err "Completing '${COMP_WORDS[*]}' did not give correct result"

    fi

    verify_extglob

  )

done

exit 0