functest/opnfv_tests/vnf/router/vrouter_base.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2017 Okinawa Open Laboratory 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 # pylint: disable=missing-docstring
  11
  12 """vrouter testing base class module"""
  13
  14 import datetime
  15 import json
  16 import logging
  17 import os
  18 import time
  19
  20 import pkg_resources
  21
  22 from functest.utils import config
  23 from functest.opnfv_tests.vnf.router.test_controller import function_test_exec
  24
  25 __author__ = "Shuya Nakama <shuya.nakama@okinawaopenlabs.org>"
  26
  27
  28 class VrouterOnBoardingBase(object):
  29     """vrouter testing base class"""
  30
  31     def __init__(self, case_name, util, util_info):
  32         self.logger = logging.getLogger(__name__)
  33         self.case_dir = pkg_resources.resource_filename(
  34             'functest', 'opnfv_tests/vnf/router')
  35         self.result_dir = os.path.join(
  36             getattr(config.CONF, 'dir_results'), case_name)
  37         self.util = util
  38         self.util_info = util_info
  39         self.vnf_list = []
  40
  41         if not os.path.exists(self.result_dir):
  42             os.makedirs(self.result_dir)
  43
  44     def test_vnf(self):
  45         """vrouter test execution"""
  46         result = False
  47         test_result_data_list = []
  48         test_scenario_file_path = os.path.join(self.case_dir,
  49                                                self.util.test_scenario_yaml)
  50         test_scenario_list = self.util.get_test_scenario(
  51             test_scenario_file_path)
  52         for test_scenario in test_scenario_list:
  53             if test_scenario["test_type"] == "function_test":
  54                 function_test_list = test_scenario["function_test_list"]
  55                 for function_test in function_test_list:
  56                     test_list = function_test["test_list"]
  57                     target_vnf_name = function_test["target_vnf_name"]
  58                     for test_info in test_list:
  59                         self.logger.info(
  60                             "%s %s test.", test_info["protocol"],
  61                             test_info["test_kind"])
  62                         (result, result_data) = self.function_test_vrouter(
  63                             target_vnf_name, test_info)
  64                         test_result_data_list.append(result_data)
  65                         if not result:
  66                             break
  67
  68         self.util.request_vm_delete(self.vnf_list)
  69
  70         test_result_data = json.dumps(test_result_data_list, indent=4)
  71
  72         return result, test_result_data
  73
  74     def function_test_vrouter(self, target_vnf_name, test_info):
  75         """function test execution"""
  76
  77         test_protocol = test_info["protocol"]
  78         test_list = test_info[test_protocol]
  79
  80         vnf_info_list = self.get_vnf_info_list(target_vnf_name)
  81         self.vnf_list = vnf_info_list
  82
  83         target_vnf = self.util.get_target_vnf(vnf_info_list)
  84
  85         reference_vnf_list = self.util.get_reference_vnf_list(vnf_info_list)
  86
  87         test_exec = function_test_exec.FunctionTestExec(self.util_info)
  88
  89         # start test
  90         start_time_ts = time.time()
  91         self.logger.info("vRouter test Start Time:'%s'", (
  92             datetime.datetime.fromtimestamp(start_time_ts).strftime(
  93                 '%Y-%m-%d %H:%M:%S')))
  94
  95         (result, test_result_data) = test_exec.run(target_vnf,
  96                                                    reference_vnf_list,
  97                                                    test_info,
  98                                                    test_list)
  99
 100         end_time_ts = time.time()
 101         duration = round(end_time_ts - start_time_ts, 1)
 102         self.logger.info("vRouter test duration :'%s'", duration)
 103
 104         return result, test_result_data
 105
 106     def get_vnf_info_list(self, target_vnf_name):
 107         return self.util.get_vnf_info_list(
 108             self.util_info["cfy_manager_ip"],
 109             self.util_info["deployment_name"],
 110             target_vnf_name)