Fix wrong image name in test code

This change fix the name of the container image to be tested
from the past name "LocalAgent" to the current name "DMA".
This also fix python function names, comments and log output.

Change-Id: I8bd8d1d884f6ff59c7e52da31455bc7f042ee9eb
Signed-off-by: Toshiaki Takahashi <takahashi.tsc@ncos.nec.co.jp>

1 files changed, 265 insertions, 0 deletions

diff --git a/baro_tests/dma.py b/baro_tests/dma.py
new file mode 100644
index 00000000..4a44480b
--- /dev/null
+++ b/baro_tests/dma.py
@@ -0,0 +1,265 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright 2018 OPNFV
+#
+# 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.
+# Patch on October 10 2017
+
+"""Executing test of DMA"""
+
+import os
+import pika
+import requests
+import time
+
+import tests
+
+logger = None
+
+TEMP_DIR = '/root'
+
+
+class DMAClient(object):
+    """Client to request DMA"""
+    def __init__(self, host, port, user, passwd):
+        """
+        Keyword arguments:
+        host -- Host URL
+        port -- Host Port
+        user -- Username
+        passwd -- Password
+        """
+        self._host = host
+        self._port = port
+        self._user = user
+        self._passwd = passwd
+
+    def set(self, file):
+        logger.error('Do nothing to DMA')
+
+    def __str__(self):
+        return ('host: {0}, port: {1}, user: {2}, pass: {3}'
+                .format(self._host, self._port,
+                        self._user, (self._passwd and '<Filterd>')))
+
+
+class RestDMAClient(DMAClient):
+    """Client to request DMA using REST"""
+    def __init__(self, host, port, user, passwd):
+        super(self.__class__, self).__init__(host, port, user, passwd)
+
+    def set(self, file):
+        logger.debug('Send to DMA using REST -- {}'.format(str(self)))
+
+        if not os.path.isfile(file):
+            print '{} is not found'.format(file)
+            return False
+        filename = os.path.basename(file)
+
+        url = 'http://{0}:{1}/collectd/conf'.format(self._host, self._port)
+        config = {'file': (filename, open(file, 'r'))}
+        requests.post(url, files=config)
+
+        return True
+
+
+class PubDMAClient(DMAClient):
+    """Client to request DMA using AMQP Publish"""
+    def __init__(self, host, port, user, passwd):
+        super(self.__class__, self).__init__(host, port, user, passwd)
+
+    def set(self, file):
+        logger.debug('Send to DMA using AMQP Publish -- {}'
+                     .format(str(self)))
+
+        if not os.path.isfile(file):
+            print '{} is not found'.format(file)
+            return False
+        filename = os.path.basename(file)
+        filebody = open(file, 'r').read()
+        message = filename + '/' + filebody
+
+        credentials = pika.PlainCredentials(self._user, self._passwd)
+        connection = pika.BlockingConnection(pika.ConnectionParameters(
+                host=self._host, port=int(self._port),
+                credentials=credentials))
+        channel = connection.channel()
+        channel.exchange_declare(exchange='collectd-conf',
+                                 exchange_type='fanout')
+        channel.basic_publish(exchange='collectd-conf',
+                              routing_key='',
+                              body=message)
+
+        connection.close()
+        return True
+
+
+def _process_dma_result(compute_node, testfunc,
+                               result, results_list, node):
+    """Print DMA test result and append it to results list.
+
+    Keyword arguments:
+    testfunc -- DMA function name
+    result -- boolean test result
+    results_list -- results list
+    """
+    if result:
+        logger.info(
+            'Test case for {0} with DMA PASSED on {1}.'.format(
+                node, testfunc))
+    else:
+        logger.error(
+            'Test case for {0} with DMA FAILED on {1}.'.format(
+                node, testfunc))
+    results_list.append((compute_node, "DMA", testfunc, result))
+
+
+def _print_result_of_dma(compute_ids, results):
+    """Print results of DMA.
+
+    Keyword arguments:
+    compute_ids -- list of compute node IDs
+    results -- results list
+    """
+    compute_node_names = ['Node-{}'.format(i) for i in range(
+        len((compute_ids)))]
+    all_computes_in_line = ''
+    for compute in compute_node_names:
+        all_computes_in_line += '| ' + compute + (' ' * (7 - len(compute)))
+    line_of_nodes = '| Test           ' + all_computes_in_line + '|'
+    logger.info('=' * 70)
+    logger.info('+' + ('-' * ((9 * len(compute_node_names))+16)) + '+')
+    logger.info(
+        '|' + ' ' * ((9*len(compute_node_names))/2)
+        + ' DMA TEST       '
+        + ' ' * (
+            9*len(compute_node_names) - (9*len(compute_node_names))/2)
+        + '|')
+    logger.info(
+        '+' + ('-' * 16) + '+' + (('-' * 8) + '+') * len(compute_node_names))
+    logger.info(line_of_nodes)
+    logger.info(
+        '+' + ('-' * 16) + '+' + (('-' * 8) + '+') * len(compute_node_names))
+
+    testname = "DMA"
+    print_line = ''
+    for id in compute_ids:
+        all_result = \
+            'FAIL' if [
+                testfunc for comp_id, testname, testfunc, res in results
+                if comp_id == id and not res] else 'PASS'
+        print_line += '| ' + all_result + '   '
+    logger.info(
+        '| {}'.format(testname) + (' ' * (15 - len(testname)))
+        + print_line + '|')
+
+    for testfunc in ['Server', 'InfoFetch']:
+        print_line = ''
+        for id in compute_ids:
+            if (id, testname, testfunc, True) in results:
+                print_line += ' PASS   |'
+            elif (id, testname, testfunc, False) in results:
+                print_line += ' FAIL   |'
+            else:
+                print_line += ' SKIP   |'
+        logger.info(
+            '|  {}'.format(testfunc) + (' ' * (14-len(testfunc)))
+            + '|' + print_line)
+
+    logger.info(
+        '+' + ('-' * 16) + '+'
+        + (('-' * 8) + '+') * len(compute_node_names))
+    logger.info('=' * 70)
+
+
+def dma_main(bt_logger, conf, computes):
+    """Check DMA of each compute node.
+
+    Keyword arguments:
+    bt_logger -- logger instance
+    computes -- compute node list
+    """
+    global logger
+    logger = bt_logger
+
+    compute_ids = []
+    agent_results = []
+    for compute_node in computes:
+        node_id = compute_node.get_id()
+        compute_ids.append(node_id)
+
+        agent_server_running = conf.is_dma_server_running(compute_node)
+        agent_infofetch_running = (
+            conf.is_dma_infofetch_running(compute_node) and
+            conf.is_redis_running(compute_node))
+
+        if agent_server_running:
+            test_name = 'barotest'
+            tmpfile = TEMP_DIR + '/' + test_name + '.conf'
+
+            agent_config = conf.get_dma_config(compute_node)
+            listen_ip = compute_node.get_ip()
+            listen_port = agent_config.get('server').get('listen_port')
+            amqp_host = agent_config.get('server').get('amqp_host')
+            amqp_port = agent_config.get('server').get('amqp_port')
+            amqp_user = agent_config.get('server').get('amqp_user')
+            amqp_passwd = agent_config.get('server').get('amqp_password')
+            rest_client = RestDMAClient(
+                              listen_ip, listen_port, '', '')
+            pub_client = PubDMAClient(
+                             amqp_host, amqp_port, amqp_user,
+                             amqp_passwd)
+
+            all_res = True
+            for client in [rest_client, pub_client]:
+                tests.test_dma_server_set_collectd(
+                    compute_node, tmpfile, logger, client)
+                sleep_time = 1
+                logger.info(
+                    'Sleeping for {} seconds'.format(sleep_time)
+                    + ' before DMA server test...')
+                time.sleep(sleep_time)
+                res = conf.check_dma_dummy_included(
+                          compute_node, test_name)
+                all_res = all_res and res
+
+            _process_dma_result(
+                compute_node.get_id(), 'Server',
+                all_res, agent_results, compute_node.get_name())
+
+        if agent_infofetch_running:
+            test_name = 'barotest'
+            resources = conf.create_testvm(compute_node, test_name)
+            sleep_time = 5
+            logger.info(
+                'Sleeping for {} seconds'.format(sleep_time)
+                + ' before DMA infofetch test...')
+            time.sleep(sleep_time)
+            res = conf.test_dma_infofetch_get_data(
+                      compute_node, test_name)
+            conf.delete_testvm(resources)
+
+            _process_dma_result(
+                compute_node.get_id(), 'InfoFetch',
+                res, agent_results, compute_node.get_name())
+
+    _print_result_of_dma(compute_ids, agent_results)
+
+    for res in agent_results:
+        if not res[3]:
+            logger.error('Some tests have failed or have not been executed')
+            logger.error('DMA test is Fail')
+            return 1
+        else:
+            pass
+    return 0