summaryrefslogtreecommitdiffstats
path: root/tests/cli_test.py
blob: 255c2a37b94bae86f925174d4adecf507b86bff5 (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
import pytest
import mock
import os
from func.cli import Cli
from os.path import expanduser


class TestClass:
    @pytest.mark.parametrize("test_input, expected", [
        (['-l',
          'zte',
          '-f',
          'compute'], "You have specified a lab that is not present under test_plan"),
        (['-l',
          'default',
          '-f',
          'test'], "This suite file test doesn't exist under benchmarks/suite/")
    ])
    def test_cli_error(self, capfd, test_input, expected):
        k = mock.patch.dict(os.environ, {'INSTALLER_TYPE': 'fuel', 'PWD': '/home'})
        with pytest.raises(SystemExit):
            k.start()
            Cli(test_input)
            k.stop()
        with open(expanduser('~') + "/qtip/logs/cli.log", "r") as file:
            data = file.read()
        assert expected in data

    @pytest.mark.parametrize("test_input, expected", [
        (['-l',
          'default',
          '-f',
          'storage'], [('fuel', '/home', './test_plan/default/storage/fio_bm.yaml'),
                       ('fuel', '/home', './test_plan/default/storage/fio_vm.yaml')])
    ])
    @mock.patch('func.cli.args_handler.prepare_and_run_benchmark')
    def test_cli_successful(self, mock_args_handler, test_input, expected):
        k = mock.patch.dict(os.environ, {'INSTALLER_TYPE': 'fuel', 'PWD': '/home'})
        k.start()
        Cli(test_input)
        k.stop()
        call_list = map(lambda x: mock_args_handler.call_args_list[x][0], range(len(expected)))
        assert sorted(call_list) == sorted(expected)