5c58463f85b429e7280215c9a5efc40ee0d0cf67
[doctor.git] / tests / run.sh
1 #!/bin/bash -ex
2 ##############################################################################
3 # Copyright (c) 2016 NEC Corporation and others.
4 #
5 # All rights reserved. This program and the accompanying materials
6 # are made available under the terms of the Apache License, Version 2.0
7 # which accompanies this distribution, and is available at
8 # http://www.apache.org/licenses/LICENSE-2.0
9 ##############################################################################
10
11 IMAGE_URL=https://launchpad.net/cirros/trunk/0.3.0/+download/cirros-0.3.0-x86_64-disk.img
12 IMAGE_NAME=cirros
13 IMAGE_FILE="${IMAGE_NAME}.img"
14 IMAGE_FORMAT=qcow2
15 VM_NAME=doctor_vm1
16 VM_FLAVOR=m1.tiny
17 ALARM_NAME=doctor_alarm1
18 INSPECTOR_PORT=12345
19 CONSUMER_PORT=12346
20
21 INSTALLER_TYPE=${INSTALLER_TYPE:-apex}
22 INSTALLER_IP=${INSTALLER_IP:-none}
23 COMPUTE_HOST=${COMPUTE_HOST:-overcloud-novacompute-0}
24 COMPUTE_IP=${COMPUTE_IP:-none}
25 ssh_opts="-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no"
26
27 if [[ "$INSTALLER_TYPE" != "apex" ]] ; then
28     echo "ERROR: INSTALLER_TYPE=$INSTALLER_TYPE is not supported."
29     exit 1
30 fi
31
32 if [[ "$INSTALLER_IP" == "none" ]] ; then
33     instack_mac=$(sudo virsh domiflist instack | awk '/default/{print $5}')
34     INSTALLER_IP=$(/usr/sbin/arp -e | grep ${instack_mac} | awk '{print $1}')
35 fi
36
37 if [[ "$COMPUTE_IP" == "none" ]] ; then
38     COMPUTE_IP=$(sudo ssh $ssh_opts $INSTALLER_IP \
39                  "source stackrc; \
40                   nova show $COMPUTE_HOST \
41                   | awk '/ ctlplane network /{print \$5}'")
42 fi
43
44 download_image() {
45     [ -e "$IMAGE_FILE" ] && return 0
46     wget "$IMAGE_URL" -o "$IMAGE_FILE"
47 }
48
49 register_image() {
50     glance image-list | grep -q " $IMAGE_NAME " && return 0
51     glance image-create --name "$IMAGE_NAME" \
52                         --visibility public \
53                         --disk-format "$IMAGE_FORMAT" \
54                         --container-format bare \
55                         --file "$IMAGE_FILE"
56 }
57
58 boot_vm() {
59     nova list | grep -q " $VM_NAME " && return 0
60     nova boot --flavor "$VM_FLAVOR" \
61               --image "$IMAGE_NAME" \
62               "$VM_NAME"
63     sleep 1
64 }
65
66 create_alarm() {
67     ceilometer alarm-list | grep -q " $ALARM_NAME " && return 0
68     vm_id=$(nova list | grep " $VM_NAME " | awk '{print $2}')
69     ceilometer alarm-event-create --name "$ALARM_NAME" \
70         --alarm-action "http://localhost:$CONSUMER_PORT/failure" \
71         --description "VM failure" \
72         --enabled True \
73         --repeat-actions False \
74         --severity "moderate" \
75         --event-type compute.instance.update \
76         -q "traits.state=string::error; traits.instance_id=string::$vm_id"
77 }
78
79 start_monitor() {
80     pgrep -f "python monitor.py" && return 0
81     sudo python monitor.py "$COMPUTE_HOST" "$COMPUTE_IP" \
82         "http://127.0.0.1:$INSPECTOR_PORT/events" > monitor.log 2>&1 &
83 }
84
85 stop_monitor() {
86     pgrep -f "python monitor.py" || return 0
87     sudo kill $(pgrep -f "python monitor.py")
88     cat monitor.log
89 }
90
91 start_inspector() {
92     pgrep -f "python inspector.py" && return 0
93     python inspector.py "$INSPECTOR_PORT" > inspector.log 2>&1 &
94 }
95
96 stop_inspector() {
97     pgrep -f "python inspector.py" || return 0
98     kill $(pgrep -f "python inspector.py")
99     cat inspector.log
100 }
101
102 start_consumer() {
103     pgrep -f "python consumer.py" && return 0
104     python consumer.py "$CONSUMER_PORT" > consumer.log 2>&1 &
105 }
106
107 stop_consumer() {
108     pgrep -f "python consumer.py" || return 0
109     kill $(pgrep -f "python consumer.py")
110     cat consumer.log
111 }
112
113 wait_for_vm_launch() {
114     echo "waiting for vm launch..."
115     while true
116     do
117         state=$(nova list | grep " $VM_NAME " | awk '{print $6}')
118         [[ "$state" == "ACTIVE" ]] && return 0
119         sleep 1
120     done
121 }
122
123 inject_failure() {
124     echo "disabling network of comupte host [$COMPUTE_HOST] for 3 mins..."
125     cat > disable_network.sh << 'END_TXT'
126 #!/bin/bash -x
127 dev=$(/usr/sbin/ip route | awk '/^default/{print $5}')
128 sleep 1
129 echo sudo ip link set $dev down
130 sleep 120
131 echo sudo ip link set $dev up
132 sleep 1
133 END_TXT
134     chmod +x disable_network.sh
135     sudo scp $ssh_opts disable_network.sh $INSTALLER_IP:
136     ssh_opts_cpu="$ssh_opts -i /home/stack/.ssh/id_rsa"
137     sudo ssh $ssh_opts $INSTALLER_IP \
138         "scp $ssh_opts_cpu disable_network.sh heat-admin@$COMPUTE_HOST: && \
139          ssh $ssh_opts_cpu 'nohup ./disable_network.sh > c 2>&1 &'"
140 }
141
142 calculate_notification_time() {
143     detect=$(grep "doctor monitor detected at" monitor.log | awk '{print $5}')
144     notified=$(grep "doctor consumer notified at" consumer.log | awk '{print $5}')
145     duration=$(echo "$notified $detect" | awk '{print $1 - $2 }')
146     echo "$notified $detect" | \
147         awk '{d = $1 - $2; if (d < 1 ) print d " OK"; else print d " NG"}'
148 }
149
150 cleanup() {
151     set +e
152     echo "cleanup..."
153     stop_monitor
154     stop_inspector
155     stop_consumer
156
157     nova service-force-down --unset "$COMPUTE_HOST" nova-compute
158     sleep 1
159     nova delete "$VM_NAME"
160     sleep 1
161     alarm_id=$(ceilometer alarm-list | grep " $ALARM_NAME " | awk '{print $2}')
162     sleep 1
163     [ -n "$alarm_id" ] && ceilometer alarm-delete "$alarm_id"
164     sleep 1
165     image_id=$(glance image-list | grep " $IMAGE_NAME " | awk '{print $2}')
166     sleep 1
167     [ -n "$image_id" ] && glance image-delete "$image_id"
168     #TODO: add host status check via nova admin api
169     echo "waiting disabled compute host back to be enabled..."
170     sleep 180
171 }
172
173
174 echo "Note: doctor/tests/run.sh has been executed."
175
176 ping -c 1 "$COMPUTE_IP"
177
178 trap cleanup ERR
179
180 echo "preparing VM image..."
181 download_image
182 register_image
183
184 echo "starting doctor sample components..."
185 start_monitor
186 start_inspector
187 start_consumer
188
189 echo "creating VM and alarm..."
190 boot_vm
191 create_alarm
192 wait_for_vm_launch
193
194 sleep 60
195 echo "injecting host failure..."
196 inject_failure
197 sleep 10
198
199 calculate_notification_time
200
201 cleanup
202
203 echo "done"