From 4781a7b06cdba9bcf4489b1cb4e94ffdeec677bd Mon Sep 17 00:00:00 2001
From: Ryota MIBU <r-mibu@cq.jp.nec.com>
Date: Fri, 15 Jan 2016 00:25:09 +0900
Subject: add doctor test case

Change-Id: Ia40face4cb32247542d5841e00caafb2cf927c2e
Signed-off-by: Ryota MIBU <r-mibu@cq.jp.nec.com>
---
 docker/run_tests.sh            |  7 ++++-
 testcases/config_functest.yaml |  1 +
 testcases/features/doctor.py   | 64 ++++++++++++++++++++++++++++++++++++++++++
 testcases/functest_utils.py    |  9 ++++--
 4 files changed, 77 insertions(+), 4 deletions(-)
 create mode 100644 testcases/features/doctor.py

diff --git a/docker/run_tests.sh b/docker/run_tests.sh
index d6faff23a..4fd505bbd 100755
--- a/docker/run_tests.sh
+++ b/docker/run_tests.sh
@@ -129,7 +129,12 @@ function run_test(){
         "promise")
             info "Running PROMISE test case..."
             # TODO
-   esac
+        ;;
+        "doctor")
+            info "Running Doctor test..."
+            python ${FUNCTEST_REPO_DIR}/testcases/features/doctor.py
+        ;;
+    esac
 }
 
 
diff --git a/testcases/config_functest.yaml b/testcases/config_functest.yaml
index 10637159a..14724c6cf 100644
--- a/testcases/config_functest.yaml
+++ b/testcases/config_functest.yaml
@@ -17,6 +17,7 @@ general:
         dir_repo_bgpvpn:    /home/opnfv/repos/bgpvpn
         dir_repo_onos:      /home/opnfv/repos/onos
         dir_repo_promise:   /home/opnfv/repos/promise
+        dir_repo_doctor:    /home/opnfv/repos/doctor
         dir_functest:       /home/opnfv/functest
         dir_results:        /home/opnfv/functest/results
         dir_functest_conf:  /home/opnfv/functest/conf
diff --git a/testcases/features/doctor.py b/testcases/features/doctor.py
new file mode 100644
index 000000000..a68c31cd0
--- /dev/null
+++ b/testcases/features/doctor.py
@@ -0,0 +1,64 @@
+#!/usr/bin/python
+#
+# Copyright (c) 2015 All rights reserved
+# This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# 0.1: This script boots the VM1 and allocates IP address from Nova
+# Later, the VM2 boots then execute cloud-init to ping VM1.
+# After successful ping, both the VMs are deleted.
+# 0.2: measure test duration and publish results under json format
+#
+#
+
+import os
+import time
+import sys
+import yaml
+
+
+with open('/home/opnfv/functest/conf/config_functest.yaml') as f:
+    functest_yaml = yaml.safe_load(f)
+
+dirs = functest_yaml.get('general').get('directories')
+FUNCTEST_REPO = dirs.get('dir_repo_functest')
+DOCTOR_REPO = dirs.get('dir_repo_doctor')
+TEST_DB_URL = functest_yaml.get('results').get('test_db_url')
+
+sys.path.append('%s/testcases' % FUNCTEST_REPO)
+import functest_utils
+
+
+def main():
+    cmd = 'cd %s/tests && ./run.sh' % DOCTOR_REPO
+    start_time_ts = time.time()
+
+    ret = functest_utils.execute_command(cmd, exit_on_error=False)
+
+    end_time_ts = time.time()
+    duration = round(end_time_ts - start_time_ts, 1)
+    if ret:
+        test_status = 'OK'
+    else:
+        test_status = 'NOK'
+
+    details = {
+        'timestart': start_time_ts,
+        'duration': duration,
+        'status': test_status,
+    }
+    pod_name = functest_utils.get_pod_name()
+    git_version = functest_utils.get_git_branch(DOCTOR_REPO)
+    functest_utils.push_results_to_db(TEST_DB_URL,
+                                      'doctor-notification',
+                                      None,
+                                      pod_name,
+                                      git_version,
+                                      details)
+
+
+if __name__ == '__main__':
+    main()
diff --git a/testcases/functest_utils.py b/testcases/functest_utils.py
index e7641b01d..a104182b5 100644
--- a/testcases/functest_utils.py
+++ b/testcases/functest_utils.py
@@ -577,7 +577,7 @@ def download_url(url, dest_path):
     return True
 
 
-def execute_command(cmd, logger=None):
+def execute_command(cmd, logger=None, exit_on_error=True):
     """
     Execute Linux command
     """
@@ -596,7 +596,9 @@ def execute_command(cmd, logger=None):
     else:
         if logger:
             logger.error("Error when executing command %s" % cmd)
-        exit(-1)
+        if exit_on_error:
+            exit(-1)
+        return False
 
 
 def get_git_branch(repo_path):
@@ -646,7 +648,8 @@ def push_results_to_db(db_url, case_name, logger, pod_name,
     headers = {'Content-Type': 'application/json'}
     try:
         r = requests.post(url, data=json.dumps(params), headers=headers)
-        logger.debug(r)
+        if logger:
+            logger.debug(r)
         return True
     except:
         print "Error:", sys.exc_info()[0]
-- 
cgit 1.2.3-korg