functest/tests/unit/cli/commands/test_cli_tier.py

   1 #!/usr/bin/env python
   2
   3 # All rights reserved. This program and the accompanying materials
   4 # are made available under the terms of the Apache License, Version 2.0
   5 # which accompanies this distribution, and is available at
   6 # http://www.apache.org/licenses/LICENSE-2.0
   7
   8
   9 import logging
  10 import unittest
  11
  12 import mock
  13
  14 from functest.cli.commands import cli_tier
  15 from functest.utils.constants import CONST
  16
  17
  18 class CliTierTesting(unittest.TestCase):
  19
  20     def setUp(self):
  21         self.tiername = 'tiername'
  22         self.testnames = 'testnames'
  23         with mock.patch('functest.cli.commands.cli_tier.tb'):
  24             self.cli_tier = cli_tier.CliTier()
  25
  26     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  27     def test_list(self, mock_click_echo):
  28         with mock.patch.object(self.cli_tier.tiers, 'get_tiers',
  29                                return_value=[]):
  30             self.cli_tier.list()
  31             mock_click_echo.assert_called_with("")
  32
  33     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  34     def test_show_default(self, mock_click_echo):
  35         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
  36                                return_value=self.tiername):
  37             self.cli_tier.show(self.tiername)
  38             mock_click_echo.assert_called_with(self.tiername)
  39
  40     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  41     def test_show_missing_tier(self, mock_click_echo):
  42         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
  43                                return_value=None), \
  44             mock.patch.object(self.cli_tier.tiers, 'get_tier_names',
  45                               return_value='tiernames'):
  46             self.cli_tier.show(self.tiername)
  47             mock_click_echo.assert_called_with("The tier with name '%s' does "
  48                                                "not exist. Available tiers are"
  49                                                ":\n  %s\n" % (self.tiername,
  50                                                               'tiernames'))
  51
  52     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  53     def test_gettests_default(self, mock_click_echo):
  54         mock_obj = mock.Mock()
  55         attrs = {'get_test_names.return_value': self.testnames}
  56         mock_obj.configure_mock(**attrs)
  57
  58         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
  59                                return_value=mock_obj):
  60             self.cli_tier.gettests(self.tiername)
  61             mock_click_echo.assert_called_with("Test cases in tier "
  62                                                "'%s':\n %s\n" % (self.tiername,
  63                                                                  self.testnames
  64                                                                  ))
  65
  66     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  67     def test_gettests_missing_tier(self, mock_click_echo):
  68         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
  69                                return_value=None), \
  70             mock.patch.object(self.cli_tier.tiers, 'get_tier_names',
  71                               return_value='tiernames'):
  72             self.cli_tier.gettests(self.tiername)
  73             mock_click_echo.assert_called_with("The tier with name '%s' does "
  74                                                "not exist. Available tiers are"
  75                                                ":\n  %s\n" % (self.tiername,
  76                                                               'tiernames'))
  77
  78     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
  79                 return_value=False)
  80     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  81     def test_run_missing_env_file(self, mock_click_echo, mock_os):
  82         self.cli_tier.run(self.tiername)
  83         mock_click_echo.assert_called_with("Functest environment is not ready."
  84                                            " Run first 'functest env prepare'")
  85
  86     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
  87                 return_value=True)
  88     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
  89     def test_run_default(self, mock_ft_utils, mock_os):
  90         cmd = ("python %s/functest/ci/run_tests.py "
  91                "%s -t %s" %
  92                (CONST.__getattribute__('dir_repo_functest'),
  93                 "-n -r ", self.tiername))
  94         self.cli_tier.run(self.tiername, noclean=True, report=True)
  95         mock_ft_utils.assert_called_with(cmd)
  96
  97     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
  98                 return_value=True)
  99     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
 100     def test_run_report_missing_noclean(self, mock_ft_utils, mock_os):
 101         cmd = ("python %s/functest/ci/run_tests.py "
 102                "%s -t %s" %
 103                (CONST.__getattribute__('dir_repo_functest'),
 104                 "-r ", self.tiername))
 105         self.cli_tier.run(self.tiername, noclean=False, report=True)
 106         mock_ft_utils.assert_called_with(cmd)
 107
 108     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
 109                 return_value=True)
 110     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
 111     def test_run_noclean_missing_report(self, mock_ft_utils, mock_os):
 112         cmd = ("python %s/functest/ci/run_tests.py "
 113                "%s -t %s" %
 114                (CONST.__getattribute__('dir_repo_functest'),
 115                 "-n ", self.tiername))
 116         self.cli_tier.run(self.tiername, noclean=True, report=False)
 117         mock_ft_utils.assert_called_with(cmd)
 118
 119     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
 120                 return_value=True)
 121     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
 122     def test_run_missing_noclean_report(self, mock_ft_utils, mock_os):
 123         cmd = ("python %s/functest/ci/run_tests.py "
 124                "%s -t %s" %
 125                (CONST.__getattribute__('dir_repo_functest'),
 126                 "", self.tiername))
 127         self.cli_tier.run(self.tiername, noclean=False, report=False)
 128         mock_ft_utils.assert_called_with(cmd)
 129
 130
 131 if __name__ == "__main__":
 132     logging.disable(logging.CRITICAL)
 133     unittest.main(verbosity=2)