summaryrefslogtreecommitdiffstats
path: root/tests/unit/cmd/commands/test_env.py
blob: 73cd5af4777a6fc3733cf21b38165c01b6f83f45 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
##############################################################################
# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
#
# 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
##############################################################################
from __future__ import absolute_import
import unittest
import mock
import uuid

from yardstick.cmd.commands.env import EnvCommand


class EnvCommandTestCase(unittest.TestCase):

    @mock.patch('yardstick.cmd.commands.env.EnvCommand._start_async_task')
    @mock.patch('yardstick.cmd.commands.env.EnvCommand._check_status')
    def test_do_influxdb(self, check_status_mock, start_async_task_mock):
        env = EnvCommand()
        env.do_influxdb({})
        self.assertTrue(start_async_task_mock.called)
        self.assertTrue(check_status_mock.called)

    @mock.patch('yardstick.cmd.commands.env.EnvCommand._start_async_task')
    @mock.patch('yardstick.cmd.commands.env.EnvCommand._check_status')
    def test_do_grafana(self, check_status_mock, start_async_task_mock):
        env = EnvCommand()
        env.do_grafana({})
        self.assertTrue(start_async_task_mock.called)
        self.assertTrue(check_status_mock.called)

    @mock.patch('yardstick.cmd.commands.env.EnvCommand._start_async_task')
    @mock.patch('yardstick.cmd.commands.env.EnvCommand._check_status')
    def test_do_prepare(self, check_status_mock, start_async_task_mock):
        env = EnvCommand()
        env.do_prepare({})
        self.assertTrue(start_async_task_mock.called)
        self.assertTrue(check_status_mock.called)

    @mock.patch('yardstick.cmd.commands.env.HttpClient.post')
    def test_start_async_task(self, post_mock):
        data = {'action': 'create_grafana'}
        EnvCommand()._start_async_task(data)
        self.assertTrue(post_mock.called)

    @mock.patch('yardstick.cmd.commands.env.HttpClient.get')
    @mock.patch('yardstick.cmd.commands.env.EnvCommand._print_status')
    def test_check_status(self, print_mock, get_mock):
        task_id = str(uuid.uuid4())
        get_mock.return_value = {'status': 2, 'result': 'error'}
        status = EnvCommand()._check_status(task_id, 'hello world')
        self.assertEqual(status, 2)

    def test_print_status(self):
        try:
            EnvCommand()._print_status('hello', 'word')
        except Exception as e:
            self.assertIsInstance(e, IndexError)


def main():
    unittest.main()


if __name__ == '__main__':
    main()