From 906056bb4e594e3c0a46cf11657b3561583699b4 Mon Sep 17 00:00:00 2001
From: dongwenjuan <dong.wenjuan@zte.com.cn>
Date: Thu, 17 Aug 2017 08:57:27 +0800
Subject: fix some bugs in monitor

1.send bytes data to the socket
see https://docs.python.org/3/library/socket.html
2.fix a endless loop

Change-Id: I880e713266347d8836cec45ebf9a500bb7c813f4
Signed-off-by: dongwenjuan <dong.wenjuan@zte.com.cn>
---
 tests/monitor/sample.py | 24 ++++++++++--------------
 tests/utils.py          |  2 +-
 2 files changed, 11 insertions(+), 15 deletions(-)

(limited to 'tests')

diff --git a/tests/monitor/sample.py b/tests/monitor/sample.py
index 4f1ce490..1333a2ec 100644
--- a/tests/monitor/sample.py
+++ b/tests/monitor/sample.py
@@ -78,7 +78,7 @@ class SampleMonitor(BaseMonitor):
 class Pinger(Thread):
     interval = 0.1  # second
     timeout = 0.1   # second
-    ICMP_ECHO_MESSAGE = '\x08\x00\xf7\xff\x00\x00\x00\x00'
+    ICMP_ECHO_MESSAGE = bytes([0x08, 0x00, 0xf7, 0xff, 0x00, 0x00, 0x00, 0x00])
 
     def __init__(self, host_name, host_ip, monitor, log):
         Thread.__init__(self)
@@ -89,18 +89,6 @@ class Pinger(Thread):
         self._stopped = False
 
     def run(self):
-        while True:
-            if self._stopped:
-                return
-            self._run()
-            time.sleep(self.interval)
-
-    def stop(self):
-        self.log.info("Stopping Pinger host_name(%s), host_ip(%s)"
-                      % (self.hostname, self.ip_addr))
-        self._stopped = True
-
-    def _run(self):
         self.log.info("Starting Pinger host_name(%s), host_ip(%s)"
                       % (self.hostname, self.ip_addr))
 
@@ -108,8 +96,10 @@ class Pinger(Thread):
                              socket.IPPROTO_ICMP)
         sock.settimeout(self.timeout)
         while True:
+            if self._stopped:
+                return
             try:
-                sock.sendto(self.ICMP_ECHO_MESSAGE.encode(), (self.ip_addr, 0))
+                sock.sendto(self.ICMP_ECHO_MESSAGE, (self.ip_addr, 0))
                 sock.recv(4096)
             except socket.timeout:
                 self.log.info("doctor monitor detected at %s" % time.time())
@@ -117,3 +107,9 @@ class Pinger(Thread):
                 self.log.info("ping timeout, quit monitoring...")
                 self._stopped = True
                 return
+            time.sleep(self.interval)
+
+    def stop(self):
+        self.log.info("Stopping Pinger host_name(%s), host_ip(%s)"
+                      % (self.hostname, self.ip_addr))
+        self._stopped = True
diff --git a/tests/utils.py b/tests/utils.py
index f57cd266..41e22353 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -56,7 +56,7 @@ class SSHClient(object):
         ret = stdout.channel.recv_exit_status()
         output = list()
         for line in stdout.read().splitlines():
-            output.append(line)
+            output.append(line.decode('utf-8'))
         if ret:
             if self.log:
                 self.log.debug("*** FAILED to run command %s (%s)" % (command, ret))
-- 
cgit 1.2.3-korg