← Back to branch summary

~landscape/charms/trusty/neutron-api-odl-vpp/trunk

« back to all changes in this revision

Viewing changes to unit_tests/test_odl_utils.py

  • Committer: Liam Young
  • Date: 2015-06-24 12:44:33 UTC
  • Revision ID: liam.young@canonical.com-20150624124433-7pqnnn8bn5ccvdb8
Fix unit tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
unit_tests/test_odl_utils.py
1
1
from test_utils import CharmTestCase
2
 
from mock import patch
3
 
import vpp_utils
 
2
import odl_utils
4
3
 
5
4
TO_PATCH = [
6
5
    'apt_install',
7
 
    'apt_purge',
8
6
    'config',
9
7
    'filter_installed_packages',
10
 
    'user_exists',
11
8
]
12
9
 
13
10
 
14
11
class VPPUtilsTest(CharmTestCase):
15
12
 
16
13
    def setUp(self):
17
 
        super(VPPUtilsTest, self).setUp(vpp_utils, TO_PATCH)
 
14
        super(VPPUtilsTest, self).setUp(odl_utils, TO_PATCH)
18
15
        self.config.side_effect = self.test_config.get
19
16
 
20
17
    def tearDown(self):
24
21
        def _filter(pkg_list):
25
22
            return pkg_list
26
23
        self.filter_installed_packages.side_effect = _filter
27
 
        vpp_utils.install_packages('neutron-api', 'api')
28
 
        self.apt_purge.assert_called_with([])
29
 
        self.apt_install.assert_called_with(['neutron-common', 'corekeeper'])
30
 
        vpp_utils.install_packages('neutron-api', 'compute')
31
 
        self.apt_purge.assert_called_with([])
32
 
        self.apt_install.assert_called_with(['neutron-common', 'corekeeper'])
33
 
        self.test_config.set('use-corekeeper', False)
34
 
        vpp_utils.install_packages('neutron-api', 'compute')
35
 
        self.apt_purge.assert_called_with([])
36
 
        self.apt_install.assert_called_with(['neutron-common', 'apport'])
 
24
        odl_utils.install_packages('neutron-api')
 
25
        self.apt_install.assert_called_with(['neutron-common',
 
26
                                             'neutron-plugin-ml2'])
37
27
 
38
28
    def test_determine_packages(self):
39
 
        self.assertEqual(vpp_utils.determine_packages(node_type='api'),
40
 
                         ['neutron-common', 'corekeeper'])
41
 
        self.assertEqual(vpp_utils.determine_packages(node_type='compute'),
42
 
                         ['neutron-common', 'corekeeper'])
43
 
        self.test_config.set('use-corekeeper', False)
44
 
        self.assertEqual(vpp_utils.determine_packages(node_type='api'),
45
 
                         ['neutron-common', 'apport'])
46
 
        self.assertEqual(vpp_utils.determine_packages(node_type='compute'),
47
 
                         ['neutron-common', 'apport'])
48
 
 
49
 
    @patch.object(vpp_utils, 'bug_1437891')
50
 
    def test_purge_packages(self, _bug1437891):
51
 
        def _filter(pkg_list):
52
 
            return []
53
 
        self.filter_installed_packages.side_effect = _filter
54
 
        self.test_config.set('use-corekeeper', True)
55
 
        vpp_utils.purge_packages('api')
56
 
        self.apt_purge.assert_called_with(['apport'])
57
 
        self.test_config.set('use-corekeeper', False)
58
 
        vpp_utils.purge_packages('compute')
59
 
        _bug1437891.assert_called_with()
60
 
        self.apt_purge.assert_called_with(['corekeeper'])
 
29
        self.assertEqual(odl_utils.determine_packages(),
 
30
                         ['neutron-common', 'neutron-plugin-ml2'])