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
  16
  17 class CliTierTesting(unittest.TestCase):
  18
  19     def setUp(self):
  20         self.tiername = 'tiername'
  21         self.testnames = 'testnames'
  22         with mock.patch('functest.cli.commands.cli_tier.tb'):
  23             self.cli_tier = cli_tier.CliTier()
  24
  25     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  26     def test_list(self, mock_click_echo):
  27         with mock.patch.object(self.cli_tier.tiers, 'get_tiers',
  28                                return_value=[]):
  29             self.cli_tier.list()
  30             mock_click_echo.assert_called_with("")
  31
  32     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  33     def test_show_default(self, mock_click_echo):
  34         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
  35                                return_value=self.tiername):
  36             self.cli_tier.show(self.tiername)
  37             mock_click_echo.assert_called_with(self.tiername)
  38
  39     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  40     def test_show_missing_tier(self, mock_click_echo):
  41         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
  42                                return_value=None), \
  43             mock.patch.object(self.cli_tier.tiers, 'get_tier_names',
  44                               return_value='tiernames'):
  45             self.cli_tier.show(self.tiername)
  46             mock_click_echo.assert_called_with("The tier with name '%s' does "
  47                                                "not exist. Available tiers are"
  48                                                ":\n  %s\n" % (self.tiername,
  49                                                               'tiernames'))
  50
  51     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  52     def test_gettests_default(self, mock_click_echo):
  53         mock_obj = mock.Mock()
  54         attrs = {'get_test_names.return_value': self.testnames}
  55         mock_obj.configure_mock(**attrs)
  56
  57         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
  58                                return_value=mock_obj):
  59             self.cli_tier.gettests(self.tiername)
  60             mock_click_echo.assert_called_with("Test cases in tier "
  61                                                "'%s':\n %s\n" % (self.tiername,
  62                                                                  self.testnames
  63                                                                  ))
  64
  65     @mock.patch('functest.cli.commands.cli_tier.click.echo')
  66     def test_gettests_missing_tier(self, mock_click_echo):
  67         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
  68                                return_value=None), \
  69             mock.patch.object(self.cli_tier.tiers, 'get_tier_names',
  70                               return_value='tiernames'):
  71             self.cli_tier.gettests(self.tiername)
  72             mock_click_echo.assert_called_with("The tier with name '%s' does "
  73                                                "not exist. Available tiers are"
  74                                                ":\n  %s\n" % (self.tiername,
  75                                                               'tiernames'))
  76
  77     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
  78     def test_run_default(self, mock_ft_utils):
  79         cmd = "run_tests -n -r -t {}".format(self.tiername)
  80         self.cli_tier.run(self.tiername, noclean=True, report=True)
  81         mock_ft_utils.assert_called_with(cmd)
  82
  83     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
  84     def test_run_report_missing_noclean(self, mock_ft_utils):
  85         cmd = "run_tests -r -t {}".format(self.tiername)
  86         self.cli_tier.run(self.tiername, noclean=False, report=True)
  87         mock_ft_utils.assert_called_with(cmd)
  88
  89     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
  90     def test_run_noclean_missing_report(self, mock_ft_utils):
  91         cmd = "run_tests -n -t {}".format(self.tiername)
  92         self.cli_tier.run(self.tiername, noclean=True, report=False)
  93         mock_ft_utils.assert_called_with(cmd)
  94
  95     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
  96     def test_run_missing_noclean_report(self, mock_ft_utils):
  97         cmd = "run_tests -t {}".format(self.tiername)
  98         self.cli_tier.run(self.tiername, noclean=False, report=False)
  99         mock_ft_utils.assert_called_with(cmd)
 100
 101
 102 if __name__ == "__main__":
 103     logging.disable(logging.CRITICAL)
 104     unittest.main(verbosity=2)