summaryrefslogtreecommitdiffstats
path: root/tests/monitor.py
diff options
context:
space:
mode:
authorRyota MIBU <r-mibu@cq.jp.nec.com>2016-01-04 16:12:39 +0900
committerRyota MIBU <r-mibu@cq.jp.nec.com>2016-01-13 00:44:35 +0900
commit5aacaaf9f3fba523264aa7bec7bee5bbc12acdb1 (patch)
tree2e7e37ec3362cc2aff5d51d66ffa09299271f0ff /tests/monitor.py
parentc1603e905316e88450b2bf7b23e3864205d40e55 (diff)
add test scenario and sample components
DOCTOR-22 Change-Id: I82cb572fadd057151c55396dd3ab4114539c8fa7 Signed-off-by: Ryota MIBU <r-mibu@cq.jp.nec.com> (cherry picked from commit 116bee29f8f3d7d4a648569dc441f4fa8487a7c4) Note: this patch includes the following change: disable doctor functional test Change-Id: I3bd62b53651777fe020c3919c415e1f2d7b45164 Signed-off-by: Ryota MIBU <r-mibu@cq.jp.nec.com> (cherry picked from commit aec51550523c3326b37dfb79680ebd80188c6a56)
Diffstat (limited to 'tests/monitor.py')
-rw-r--r--tests/monitor.py79
1 files changed, 79 insertions, 0 deletions
diff --git a/tests/monitor.py b/tests/monitor.py
new file mode 100644
index 00000000..2a78cb66
--- /dev/null
+++ b/tests/monitor.py
@@ -0,0 +1,79 @@
+#
+# Copyright 2016 NEC Corporation.
+#
+# 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.
+
+import argparse
+import json
+import requests
+import socket
+import time
+
+
+# NOTE: icmp message with all zero data (checksum = 0xf7ff)
+# see https://tools.ietf.org/html/rfc792
+ICMP_ECHO_MESSAGE = '\x08\x00\xf7\xff\x00\x00\x00\x00'
+
+
+class DoctorMonitorSample(object):
+
+ interval = 0.1 # second
+ timeout = 0.1 # second
+ event_type = "compute.host.down"
+
+ def __init__(self, args):
+ self.hostname = args.hostname
+ self.inspector = args.inspector
+ self.ip_addr = socket.gethostbyname(self.hostname)
+
+ def start_loop(self):
+ print "start ping to host %s" % self.hostname
+ sock = socket.socket(socket.AF_INET, socket.SOCK_RAW,
+ socket.IPPROTO_ICMP)
+ sock.settimeout(self.timeout)
+ while True:
+ try:
+ sock.sendto(ICMP_ECHO_MESSAGE, (self.ip_addr, 0))
+ data = sock.recv(4096)
+ except socket.timeout:
+ print "doctor monitor detected at %s" % time.time()
+ self.report_error()
+ print "ping timeout, quit monitoring..."
+ return
+ time.sleep(self.interval)
+
+ def report_error(self):
+ payload = {"type": self.event_type, "hostname": self.hostname}
+ data = json.dumps(payload)
+ headers = {'content-type': 'application/json'}
+ requests.post(self.inspector, data=data, headers=headers)
+
+
+def get_args():
+ parser = argparse.ArgumentParser(description='Doctor Sample Monitor')
+ parser.add_argument('hostname', metavar='HOSTNAME', type=str, nargs='?',
+ help='a hostname to monitor connectivity')
+ parser.add_argument('inspector', metavar='INSPECTOR', type=str, nargs='?',
+ help='inspector url to report error',
+ default='http://127.0.0.1:12345/events')
+ return parser.parse_args()
+
+
+def main():
+ args = get_args()
+ monitor = DoctorMonitorSample(args)
+ monitor.start_loop()
+
+
+if __name__ == '__main__':
+ main()