src/ceph/qa/workunits/ceph-helpers-root.sh

   1 #!/bin/bash
   2 #
   3 # Copyright (C) 2015 Red Hat <contact@redhat.com>
   4 #
   5 # Author: Loic Dachary <loic@dachary.org>
   6 #
   7 # This program is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU Library Public License as published by
   9 # the Free Software Foundation; either version 2, or (at your option)
  10 # any later version.
  11 #
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU Library Public License for more details.
  16 #
  17
  18 #######################################################################
  19
  20 function install() {
  21     for package in "$@" ; do
  22         install_one $package
  23     done
  24     return 0
  25 }
  26
  27 function install_one() {
  28     case $(lsb_release -si) in
  29         Ubuntu|Debian|Devuan)
  30             sudo apt-get install -y "$@"
  31             ;;
  32         CentOS|Fedora|RedHatEnterpriseServer)
  33             sudo yum install -y "$@"
  34             ;;
  35         *SUSE*)
  36             sudo zypper --non-interactive install "$@"
  37             ;;
  38         *)
  39             echo "$(lsb_release -si) is unknown, $@ will have to be installed manually."
  40             ;;
  41     esac
  42 }
  43
  44 #######################################################################
  45
  46 function control_osd() {
  47     local action=$1
  48     local id=$2
  49
  50     local init=$(ceph-detect-init)
  51
  52     case $init in
  53         upstart)
  54             sudo service ceph-osd $action id=$id
  55             ;;
  56         systemd)
  57             sudo systemctl $action ceph-osd@$id
  58             ;;
  59         *)
  60             echo ceph-detect-init returned an unknown init system: $init >&2
  61             return 1
  62             ;;
  63     esac
  64     return 0
  65 }
  66
  67 #######################################################################
  68
  69 function pool_read_write() {
  70     local size=${1:-1}
  71     local dir=/tmp
  72     local timeout=360
  73     local test_pool=test_pool
  74
  75     ceph osd pool delete $test_pool $test_pool --yes-i-really-really-mean-it || return 1
  76     ceph osd pool create $test_pool 4 || return 1
  77     ceph osd pool set $test_pool size $size || return 1
  78     ceph osd pool set $test_pool min_size $size || return 1
  79     ceph osd pool application enable $test_pool rados
  80
  81     echo FOO > $dir/BAR
  82     timeout $timeout rados --pool $test_pool put BAR $dir/BAR || return 1
  83     timeout $timeout rados --pool $test_pool get BAR $dir/BAR.copy || return 1
  84     diff $dir/BAR $dir/BAR.copy || return 1
  85     ceph osd pool delete $test_pool $test_pool --yes-i-really-really-mean-it || return 1
  86 }
  87
  88 #######################################################################
  89
  90 set -x
  91
  92 "$@"