slapos/promise: Improve apachedex and slowquery promises

/reviewed-on !21

slapos/promise/check_apachedex_result/__init__.py

@@ -12,54 +12,53 @@ import time
 import datetime
 import argparse
 
-def checkApachedexResult(apachedex_file, apachedex_report_status_file, desired_threshold):
-
-  if not os.path.isfile(apachedex_file):
-    open(apachedex_file, 'a').close()
-
-  with open(apachedex_file, 'r') as content_file:
-    content = content_file.read()
- 
-  if len(content) == 0:
-    # File is empty
-    # Check the creation date of the file
-    # and if the date is greater than 30 hour throw an error
-    date_created = os.path.getmtime(apachedex_file)
-    current_date = time.mktime(datetime.datetime.now().timetuple())
-    if current_date - date_created > 108000:
-      with open(apachedex_report_status_file) as f:
-        json_content = f.read()
-
-      # Print the message from the monitor report
-      if len(json_content) > 0:
-        message = json.loads(json_content)["message"]
-        return message + "\nFile modification date is greater than 30 hours"
-      return "File modification date is greater than 30 hour"
+def checkApachedexResult(apachedex_path, apachedex_report_status_file, desired_threshold):
+  message = "No apachedex result for today or yesterday"
+  today = datetime.date.today()
+  today_or_yesterday = today, today - datetime.timedelta(1)
+  try:
+    apachedex_file = max(os.listdir(apachedex_path),
+      key=lambda x: os.stat(os.path.join(apachedex_path, x)).st_mtime)
+  except ValueError:
+    if datetime.date.fromtimestamp(os.stat(apachedex_path).st_mtime) in today_or_yesterday: 
+      return 0, "Instance has been just deployed. Skipping check.."
   else:
-    #TODO: this is old regex for bash, improve it
+    for date in today_or_yesterday:
+      if apachedex_file == date.strftime('ApacheDex-%Y-%m-%d.html'):
+
+        with open(os.path.join(apachedex_path, apachedex_file)) as f:
+          content = f.read()
+        if content:
+          # XXX: if not a lot of usage, skip
+          # XXX: too fragile, use lxml.html and use xpath
           regex = r"Overall<\/h2>.*\n<th>apdex<\/th><th>.*?\n<\/tr><tr>\n<td [^<]*>(.*?)%<\/td>"
           m = re.findall(regex, content)
-    if len(m) > 0:
-      result=int(m[0])
-      if result > desired_threshold:
-        return "Thanks for keeping it all clean, result is " + str(result)
+          if m:
+            apx_result=int(m[0])
+            if apx_result > desired_threshold:
+              return 0,  "Thanks for keeping it all clean, result is " + str(apx_result)
             else:
-        return "Threshold is lower than expected:  Expected was " + \
-               str(desired_threshold) +" and current result is " + str(result)
-  return "No result found in the apdex file or the file is corrupted"
+              return 1, "Threshold is lower than expected:  Expected was " + \
+                     str(desired_threshold) +" and current result is " + str(apx_result)
+        message = "No result found in the apdex file or the file is corrupted"
+        break
+
+  with open(apachedex_report_status_file) as f:
+    try:
+      json_content = json.load(f)
+    except ValueError, e:
+      json_content = ''
+  if json_content:
+    message += "\n" + json_content["message"]
+  return 1, message
 
 def main():
   parser = argparse.ArgumentParser()
-  parser.add_argument("--apachedex_file", required=True)
+  parser.add_argument("--apachedex_path", required=True)
   parser.add_argument("--status_file", required=True)
   parser.add_argument("--threshold", required=True)
   args = parser.parse_args()
 
-  if args.apachedex_file:
-    args.apachedex_file = args.apachedex_file + "/ApacheDex-" + datetime.date.today().strftime('%Y-%m-%d') + ".html"
-
-  result = checkApachedexResult(args.apachedex_file, args.status_file, args.threshold)
-
-  print result
-  if result != "OK":
-    sys.exit(1)
+  status, message = checkApachedexResult(args.apachedex_path, args.status_file, args.threshold)
+  print(message)
+  sys.exit(status)

slapos/promise/check_slow_queries_digest_result/__init__.py

@@ -12,34 +12,28 @@ import time
 import datetime
 import argparse
 
-def checkMariadbDigestResult(mariadbdex_file, mariadbdex_report_status_file,
+def checkMariadbDigestResult(mariadbdex_path, mariadbdex_report_status_file,
                              max_query_threshold, slowest_query_threshold):
+  message = "No mariadbdex result for today or yesterday"
+  today = datetime.date.today()
+  today_or_yesterday = today, today - datetime.timedelta(1)
 
-  if not os.path.isfile(mariadbdex_file):
-    open(mariadbdex_file, 'a').close()
-
-  with open(mariadbdex_file, 'r') as content_file:
-    content = content_file.read()
- 
-  if len(content) == 0:
-    # File is empty
-    # Check the creation date of the file
-    # and if the date is greater than 30 hour throw an error
-    date_created = os.path.getmtime(mariadbdex_file)
-    current_date = time.mktime(datetime.datetime.now().timetuple())
-    if current_date - date_created > 108000:
-      with open(mariadbdex_report_status_file) as f:
-        json_content = f.read()
-
-      # Print the message from the monitor report
-      if len(json_content) > 0:
-        message = json.loads(json_content)["message"]
-        return message + "\nFile modification date is greater than 30 hours"
-      return "File modification date is greater than 30 hour"
+  try:
+    mariadbdex_file = max(os.listdir(mariadbdex_path),
+      key=lambda x: os.stat(os.path.join(mariadbdex_path, x)).st_mtime)
+  except ValueError:
+    if datetime.date.fromtimestamp(os.stat(mariadbdex_path).st_mtime) in today_or_yesterday: 
+        return 0, "Instance has been just deployed. Skipping check.."
   else:
+    for date in today_or_yesterday:
+      if mariadbdex_file == date.strftime('slowquery_digest.txt-%Y-%m-%d'):
+        with open(os.path.join(mariadbdex_path, mariadbdex_file)) as f:
+          content = f.read()
+        if content:
+          # XXX: if not a lot of usage, skip this
           regex = r"Overall: (.*) total,[\S\s]*# Exec time( *([\d]+)m?s?){4}"
           m = re.findall(regex, content)
-    if len(m) > 0:
+          if m:
             total_queries_exec=m[0][0].strip()
             slowest_query_time=int(m[0][2].strip())
             has_k=total_queries_exec[-1:]
@@ -48,26 +42,37 @@ def checkMariadbDigestResult(mariadbdex_file, mariadbdex_report_status_file,
               total_queries_exec=float(pre)*1000
             else:
               total_queries_exec=int(total_queries_exec)
-
             if total_queries_exec < max_query_threshold and slowest_query_time < slowest_query_threshold:
-        return "Thanks for keeping it all clean, total queries are : " + str(total_queries_exec) + \
+              return 0,  "Thanks for keeping it all clean, total queries are : " + str(total_queries_exec) + \
                          " slowest query time is : " + str(slowest_query_time)
             else:
-        return "Threshold is lower than expected: \nExpected total queries : " + \
+              return 1, "Threshold is lower than expected: \nExpected total queries : " + \
                        str(max_query_threshold) +" and current is: " + str(total_queries_exec) + "\n"+ \
-               "Expected slowest query : " + str(slowest_query_threshold) + " and current is: " + str(slowest_query_time)
-  return "No result found in the apdex file or the file is corrupted"
+                       "Expected slowest query : " + str(slowest_query_threshold) + " and current is: " + \
+                       str(slowest_query_time)
+        message = "No result found in the slow query digest file or the file is corrupted"
+        break
+
+  with open(mariadbdex_report_status_file) as f:
+    try:
+      json_content = json.load(f)
+    except ValueError, e:
+      json_content = ''
+  if json_content:
+    message += "\n" + json_content["message"]
+  return 1, message
 
 def main():
   parser = argparse.ArgumentParser()
-  parser.add_argument("--ptdigest_file", required=True)
+  parser.add_argument("--ptdigest_path", required=True)
   parser.add_argument("--status_file", required=True)
   parser.add_argument("--max_queries_threshold", required=True)
   parser.add_argument("--slowest_query_threshold", required=True)
   args = parser.parse_args()
 
-  result = checkMariadbDigestResult(args.ptdigest_file, args.status_file, args.max_queries_threshold, args.slowest_query_threshold)
+  status, message = checkMariadbDigestResult(args.ptdigest_path, args.status_file,
+                                    args.max_queries_threshold, args.slowest_query_threshold
+                                    )
 
-  print result
-  if result != "OK":
-    sys.exit(1)
+  print message
+  sys.exit(status)

slapos/test/promise/test_check_apachedex_result.py

@@ -29,37 +29,77 @@ import unittest
 import os
 import time
 import tempfile
+import datetime
+import shutil
 
 from slapos.test.promise import data
 from slapos.promise.check_apachedex_result import checkApachedexResult
 
 class TestCheckApacheDigestResult(unittest.TestCase):
 
+  def _create_file(self, date, with_content):
+    content = ''
+    if with_content:
+      with open(self.base_path + "/apachedex.html") as f:
+        content = f.read()
+
+    name = date.strftime('ApacheDex-%Y-%m-%d.html')
+    oldtime = time.mktime(date.timetuple()) + 2000
+    with open( self.base_dir+name, 'a') as the_file:
+      the_file.write(content)
+    os.utime(self.base_dir+name, ( oldtime , oldtime ))
+    
+  def _remove_file(self, date):
+    name = date.strftime('ApacheDex-%Y-%m-%d.html')
+    os.remove(self.base_dir+name)
+
   def setUp(self):
 
-    _, self.empty_apachedex_file = tempfile.mkstemp()
-    _, self.old_apachedex_file = tempfile.mkstemp()
+    self.base_path = "/".join(data.__file__.split("/")[:-1])
+    self.base_dir = "/tmp/ap/"
+    if not os.path.exists(self.base_dir):
+      os.makedirs(self.base_dir)
+      os.utime(self.base_dir, (time.time() - 202800, time.time() - 202800))
+
+    # create test files
+    self.today = datetime.date.today()
+    self._create_file(self.today, True)
+
+    self.yesterday = (self.today - datetime.timedelta(1))
+    self._create_file(self.yesterday, False)
     _, self.status_file = tempfile.mkstemp()
 
-    self.base_path = "/".join(data.__file__.split("/")[:-1])
-    self.status = "ok"
+  def test_threshold_is_greater(self):
+    status, message = checkApachedexResult(self.base_dir, self.status_file, 60)
+    self.assertEquals("Thanks for keeping it all clean, result is 80", message)
+    self.assertEquals(0, status)
 
-  def test_pass(self):
-    self.assertEquals("Thanks for keeping it all clean, result is 80",
-      checkApachedexResult(self.base_path + "/apachedex.html", self.status_file, 60))
+  def test_no_today_file_and_empty_yesterday_file(self):
+    self._remove_file(self.today)
+    status, message = checkApachedexResult(self.base_dir, self.status_file, 60)
+    self.assertEquals("No result found in the apdex file or the file is corrupted", message)
+    self.assertEquals(1, status)
+    self._create_file(self.today, True)
 
-  def test_empty_file(self):
-    self.assertEquals("No result found in the apdex file or the file is corrupted",
-      checkApachedexResult(self.empty_apachedex_file, self.status_file, 60))
+  def test_threshold_is_lower(self):
+    status, message = checkApachedexResult(self.base_dir, self.status_file, 90)
+    self.assertEquals("Threshold is lower than expected:  Expected was 90 and current result is 80", message)
+    self.assertEquals(1, status)
 
-  def test_fail(self):
-    self.assertEquals("Threshold is lower than expected:  Expected was 90 and current result is 80",
-      checkApachedexResult(self.base_path + "/apachedex.html", self.status_file, 90))
+  def test_no_today_file_but_yesterday_file(self):
+    self._remove_file(self.today)
+    self._create_file(self.yesterday, True)
+    status, message = checkApachedexResult(self.base_dir, self.status_file, 60)
+    self.assertEquals("Thanks for keeping it all clean, result is 80", message)
+    self.assertEquals(0, status)
+    self._create_file(self.today, True)
+    self._remove_file(self.yesterday)
+    self._create_file(self.yesterday, False)
 
-  def test_old_file(self):
-    os.utime(self.old_apachedex_file, (time.time() - 202800, time.time() - 202800))
-    self.assertEquals("File modification date is greater than 30 hour",
-      checkApachedexResult(self.old_apachedex_file, self.status_file, 60))
+  def tearDown(self):
+    self._remove_file(self.today)
+    self._remove_file(self.yesterday)
+    shutil.rmtree('/tmp/ap')
 
 if __name__ == '__main__':
   unittest.main()

slapos/test/promise/test_check_slow_queries_digest_result.py

@@ -29,36 +29,77 @@ import unittest
 import os
 import time
 import tempfile
+import datetime
+import shutil
 
 from slapos.test.promise import data
 from slapos.promise.check_slow_queries_digest_result import checkMariadbDigestResult
 
 class TestCheckSlowQueriesDigestResult(unittest.TestCase):
 
-  def setUp(self):
+  def _create_file(self, date, with_content):
+    content = ''
+    if with_content:
+      with open(self.base_path + "/ptdigest.html") as f:
+        content = f.read()
 
-    _, self.empty_ptdigest_file = tempfile.mkstemp()
-    _, self.old_ptdigest_file = tempfile.mkstemp()
-    _, self.status_file = tempfile.mkstemp()
+    name = date.strftime('slowquery_digest.txt-%Y-%m-%d')
+    oldtime = time.mktime(date.timetuple()) + 2000
+    with open( self.base_dir+name, 'a') as the_file:
+      the_file.write(content)
+    os.utime(self.base_dir+name, ( oldtime , oldtime ))
+    
+  def _remove_file(self, date):
+    name = date.strftime('slowquery_digest.txt-%Y-%m-%d')
+    os.remove(self.base_dir+name)
+
+  def setUp(self):
 
     self.base_path = "/".join(data.__file__.split("/")[:-1])
+    self.base_dir = "/tmp/ap/"
+    if not os.path.exists(self.base_dir):
+      os.makedirs(self.base_dir)
+      os.utime(self.base_dir, (time.time() - 202800, time.time() - 202800))
 
-  def test_pass(self):
-    self.assertEquals("Thanks for keeping it all clean, total queries are : 3420.0 slowest query time is : 34",
-      checkMariadbDigestResult(self.base_path + "/ptdigest.html", self.status_file, 5000, 100))
+    # create test files
+    self.today = datetime.date.today()
+    self._create_file(self.today, True)
 
-  def test_empty_file(self):
-    self.assertEquals("No result found in the apdex file or the file is corrupted",
-      checkMariadbDigestResult(self.empty_ptdigest_file, self.status_file, 60, 100))
+    self.yesterday = (self.today - datetime.timedelta(1))
+    self._create_file(self.yesterday, False)
+    _, self.status_file = tempfile.mkstemp()
+
+  def test_threshold_is_greater(self):
+    status, message = checkMariadbDigestResult(self.base_dir, self.status_file, 5000, 100)
+    self.assertEquals("Thanks for keeping it all clean, total queries are : 3420.0 slowest query time is : 34", message)
+    self.assertEquals(0, status)
+
+  def test_no_today_file_and_empty_yesterday_file(self):
+    self._remove_file(self.today)
+    status, message = checkMariadbDigestResult(self.base_dir, self.status_file, 60, 100)
+    self.assertEquals("No result found in the slow query digest file or the file is corrupted", message)
+    self.assertEquals(1, status)
+    self._create_file(self.today, True)
 
   def test_fail(self):
-    self.assertEquals("Threshold is lower than expected: \nExpected total queries : 90 and current is: 3420.0\nExpected slowest query : 100 and current is: 34",
-      checkMariadbDigestResult(self.base_path + "/ptdigest.html", self.status_file, 90, 100))
+    status, message = checkMariadbDigestResult(self.base_dir, self.status_file, 90, 100)
+    self.assertEquals("Threshold is lower than expected: \nExpected total queries : 90 and current is: 3420.0\nExpected slowest query : 100 and current is: 34", message)
+    self.assertEquals(1, status)
+
+  def test_no_today_file_but_yesterday_file(self):
+    self._remove_file(self.today)
+    self._create_file(self.yesterday, True)
+    status, message = checkMariadbDigestResult(self.base_dir, self.status_file, 5000, 100)
+    self.assertEquals("Thanks for keeping it all clean, total queries are : 3420.0 slowest query time is : 34", message)
+    self.assertEquals(0, status)
+    self._create_file(self.today, True)
+    self._remove_file(self.yesterday)
+    self._create_file(self.yesterday, False)
     
-  def test_old_file(self):
-    os.utime(self.old_ptdigest_file, (time.time() - 202800, time.time() - 202800))
-    self.assertEquals("File modification date is greater than 30 hour",
-      checkMariadbDigestResult(self.old_ptdigest_file, self.status_file, 60, 100))
+  def tearDown(self):
+    self._remove_file(self.today)
+    self._remove_file(self.yesterday)
+    shutil.rmtree('/tmp/ap')
 
 if __name__ == '__main__':
   unittest.main()