Skip to content

Commit

Permalink
Test content params of FIle resources in Class[prometheus]
Browse files Browse the repository at this point in the history
Following voxpupuli#83, and preparing tests for voxpupuli#48.
  • Loading branch information
sathieu committed Oct 23, 2017
1 parent 1698b87 commit 255b8b9
Show file tree
Hide file tree
Showing 6 changed files with 405 additions and 13 deletions.
35 changes: 22 additions & 13 deletions spec/classes/prometheus_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -88,28 +88,37 @@
}

# prometheus::config
if ['debian-7-x86_64'].include?(os) ||
['centos-6-x86_64', 'redhat-6-x86_64'].include?(os)
# init_style = 'debian' or
if ['debian-7-x86_64'].include?(os)
# init_style = 'debian'

it {
is_expected.to contain_file('/etc/init.d/prometheus').with(
'mode' => '0555',
'owner' => 'root',
'group' => 'root',
'content' => File.read(fixtures('files', 'prometheus.debian'))
)
}
elsif ['centos-6-x86_64', 'redhat-6-x86_64'].include?(os)
# init_style = 'sysv'

it {
is_expected.to contain_file('/etc/init.d/prometheus').with(
'mode' => '0555',
'owner' => 'root',
'group' => 'root',
# 'content' => ...
'content' => File.read(fixtures('files', 'prometheus.sysv'))
)
}
elsif ['centos-7-x86_64', 'debian-8-x86_64', 'redhat-7-x86_64', 'ubuntu-16.04-x86_64'].include?(os)
# init_style = 'systemd'

it {
is_expected.to contain_file('/etc/systemd/system/prometheus.service').with(
'mode' => '0644',
'owner' => 'root',
'group' => 'root',
# 'content' => ...
'mode' => '0644',
'owner' => 'root',
'group' => 'root',
'content' => File.read(fixtures('files', 'prometheus.systemd'))
).that_notifies(['Exec[prometheus-systemd-reload]', 'Class[Prometheus::Run_service]'])
}

Expand All @@ -125,10 +134,10 @@

it {
is_expected.to contain_file('/etc/init/prometheus.conf').with(
'mode' => '0444',
'owner' => 'root',
'group' => 'root',
# 'content' => ...
'mode' => '0444',
'owner' => 'root',
'group' => 'root',
'content' => File.read(fixtures('files', 'prometheus.upstart'))
)
}

Expand Down Expand Up @@ -164,7 +173,7 @@
'owner' => 'prometheus',
'group' => 'prometheus',
'mode' => '0660',
# 'content' => ...,
'content' => File.read(fixtures('files', 'prometheus.yaml'))
).that_notifies('Class[prometheus::service_reload]')
}

Expand Down
177 changes: 177 additions & 0 deletions spec/fixtures/files/prometheus.debian
Original file line number Diff line number Diff line change
@@ -0,0 +1,177 @@
#! /bin/sh
### BEGIN INIT INFO
# Provides: prometheus
# Required-Start: $local_fs $remote_fs
# Required-Stop: $local_fs $remote_fs
# Default-Start: 2 3 4 5
# Default-Stop: S 0 1 6
# Short-Description: Prometheus monitoring system
# Description: The Prometheus monitoring system and time series database. https://prometheus.io/
#
### END INIT INFO

# Do NOT "set -e"

# PATH should only include /usr/* if it runs after the mountnfs.sh script
PATH=/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/bin
DESC="Prometheus monitoring framework"
NAME=prometheus
DAEMON=/usr/local/bin/$NAME
PIDFILE=/var/run/$NAME/$NAME.pid
DAEMON_ARGS=" -config.file=/etc/prometheus/prometheus.yaml
-storage.local.path=/var/lib/prometheus
-web.console.templates=/usr/local/share/prometheus/consoles
-web.console.libraries=/usr/local/share/prometheus/console_libraries
"
USER=prometheus
SCRIPTNAME=/etc/init.d/$NAME

# Exit if the package is not installed
[ -x "$DAEMON" ] || exit 0

# Read configuration variable file if it is present
[ -r /etc/default/$NAME ] && . /etc/default/$NAME

# Load the VERBOSE setting and other rcS variables
[ -f /etc/default/rcS ] && . /etc/default/rcS

# Define LSB log_* functions.
# Depend on lsb-base (>= 3.0-6) to ensure that this file is present.
. /lib/lsb/init-functions

#
# Function to create run directory
#
mkrundir() {
[ ! -d /var/run/prometheus ] && mkdir -p /var/run/prometheus
chown $USER /var/run/prometheus
}

#
# Function that starts the daemon/service
#
do_start()
{
# Return
# 0 if daemon has been started
# 1 if daemon was already running
# 2 if daemon could not be started
mkrundir
start-stop-daemon --start --quiet --pidfile $PIDFILE --exec $DAEMON --chuid $USER --background --make-pidfile --test > /dev/null \
|| return 1
start-stop-daemon --start --quiet --pidfile $PIDFILE --exec $DAEMON --chuid $USER --background --make-pidfile -- \
$DAEMON_ARGS \
|| return 2

for i in `seq 1 30`; do
if ! start-stop-daemon --quiet --stop --test --pidfile $PIDFILE --exec $DAEMON --user $USER; then
RETVAL=2
sleep 1
continue
fi
if "$DAEMON" info ${RPC_ADDR} >/dev/null; then
return 0
fi
done
return "$RETVAL"
}


#
# Function that stops the daemon/service
#
do_stop()
{
# Return
# 0 if daemon has been stopped
# 1 if daemon was already stopped
# 2 if daemon could not be stopped
# other if a failure occurred
start-stop-daemon --stop --quiet --retry=TERM/30/KILL/5 --pidfile $PIDFILE --name $NAME
RETVAL="$?"
[ "$RETVAL" = 2 ] && return 2
# Wait for children to finish too if this is a daemon that forks
# and if the daemon is only ever run from this initscript.
# If the above conditions are not satisfied then add some other code
# that waits for the process to drop all resources that could be
# needed by services started subsequently. A last resort is to
# sleep for some time.
start-stop-daemon --stop --quiet --oknodo --retry=0/30/KILL/5 --exec $DAEMON
[ "$?" = 2 ] && return 2
# Many daemons don't delete their pidfiles when they exit.
rm -f $PIDFILE
return "$RETVAL"
}

#
# Function that sends a SIGHUP to the daemon/service
#
do_reload() {
#
# If the daemon can reload its configuration without
# restarting (for example, when it is sent a SIGHUP),
# then implement that here.
#
start-stop-daemon --stop --signal 1 --quiet --pidfile $PIDFILE --name $NAME
return 0
}

case "$1" in
start)
[ "$VERBOSE" != no ] && log_daemon_msg "Starting $DESC" "$NAME"
do_start
case "$?" in
0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;;
2) [ "$VERBOSE" != no ] && log_end_msg 1 ;;
esac
;;
stop)
[ "$VERBOSE" != no ] && log_daemon_msg "Stopping $DESC" "$NAME"
do_stop
case "$?" in
0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;;
2) [ "$VERBOSE" != no ] && log_end_msg 1 ;;
esac
;;
#reload|force-reload)
#
# If do_reload() is not implemented then leave this commented out
# and leave 'force-reload' as an alias for 'restart'.
#
#log_daemon_msg "Reloading $DESC" "$NAME"
#do_reload
#log_end_msg $?
#;;
restart|force-reload)
#
# If the "reload" option is implemented then remove the
# 'force-reload' alias
#
log_daemon_msg "Restarting $DESC" "$NAME"
do_stop
case "$?" in
0|1)
do_start
case "$?" in
0) log_end_msg 0 ;;
1) log_end_msg 1 ;; # Old process is still running
*) log_end_msg 1 ;; # Failed to start
esac
;;
*)
# Failed to stop
log_end_msg 1
;;
esac
;;
status)
status_of_proc -p $PIDFILE $DAEMON $NAME && exit 0 || exit $?
;;
*)
#echo "Usage: $SCRIPTNAME {start|stop|restart|reload|force-reload}" >&2
echo "Usage: $SCRIPTNAME {start|stop|status|restart|force-reload}" >&2
exit 3
;;
esac

:
21 changes: 21 additions & 0 deletions spec/fixtures/files/prometheus.systemd
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
[Unit]
Description=Prometheus Monitoring framework
Wants=basic.target
After=basic.target network.target

[Service]
User=prometheus
Group=prometheus
ExecStart=/usr/local/bin/prometheus \
-config.file=/etc/prometheus/prometheus.yaml\
-storage.local.path=/var/lib/prometheus \
-web.console.templates=/usr/local/share/prometheus/consoles \
-web.console.libraries=/usr/local/share/prometheus/console_libraries \

ExecReload=/bin/kill -HUP $MAINPID
KillMode=process
Restart=always
RestartSec=42s

[Install]
WantedBy=multi-user.target
Loading

0 comments on commit 255b8b9

Please sign in to comment.