← Back to branch summary

~openstack-charmers-archive/charms/trusty/neutron-gateway/trunk

« back to all changes in this revision

Viewing changes to unit_tests/test_actions_openstack_upgrade.py

  • Committer: James Page
  • Date: 2015-10-22 13:23:58 UTC
  • Revision ID: james.page@ubuntu.com-20151022132358-qin1nvlnqn4aezaz
Tags: 15.10
15.10 Charm release

Show diffs side-by-side

added added

removed removed

Lines of Context:
unit_tests/test_actions_openstack_upgrade.py
 
1
from mock import patch
 
2
import os
 
3
 
 
4
from test_utils import (
 
5
    CharmTestCase
 
6
)
 
7
 
 
8
os.environ['JUJU_UNIT_NAME'] = 'neutron-gateway'
 
9
 
 
10
with patch('charmhelpers.core.hookenv.status_set'):
 
11
    with patch('neutron_utils.register_configs') as register_configs:
 
12
        import openstack_upgrade
 
13
 
 
14
TO_PATCH = [
 
15
    'do_openstack_upgrade',
 
16
    'config_changed',
 
17
]
 
18
 
 
19
 
 
20
class TestNeutronGWUpgradeActions(CharmTestCase):
 
21
 
 
22
    def setUp(self):
 
23
        super(TestNeutronGWUpgradeActions, self).setUp(openstack_upgrade,
 
24
                                                       TO_PATCH)
 
25
 
 
26
    @patch('charmhelpers.contrib.openstack.utils.config')
 
27
    @patch('charmhelpers.contrib.openstack.utils.action_set')
 
28
    @patch('charmhelpers.contrib.openstack.utils.git_install_requested')
 
29
    @patch('charmhelpers.contrib.openstack.utils.openstack_upgrade_available')
 
30
    def test_openstack_upgrade_true(self, upgrade_avail, git_requested,
 
31
                                    action_set, config):
 
32
        git_requested.return_value = False
 
33
        upgrade_avail.return_value = True
 
34
        config.return_value = True
 
35
 
 
36
        openstack_upgrade.openstack_upgrade()
 
37
 
 
38
        self.assertTrue(self.do_openstack_upgrade.called)
 
39
        self.assertTrue(self.config_changed.called)
 
40
 
 
41
    @patch('charmhelpers.contrib.openstack.utils.config')
 
42
    @patch('charmhelpers.contrib.openstack.utils.action_set')
 
43
    @patch('charmhelpers.contrib.openstack.utils.git_install_requested')
 
44
    @patch('charmhelpers.contrib.openstack.utils.openstack_upgrade_available')
 
45
    def test_openstack_upgrade_false(self, upgrade_avail, git_requested,
 
46
                                     action_set, config):
 
47
        git_requested.return_value = False
 
48
        upgrade_avail.return_value = True
 
49
        config.return_value = False
 
50
 
 
51
        openstack_upgrade.openstack_upgrade()
 
52
 
 
53
        self.assertFalse(self.do_openstack_upgrade.called)
 
54
        self.assertFalse(self.config_changed.called)