← Back to branch summary

~brad-marshall/charms/trusty/keystone/haproxy-add-timeout

« back to all changes in this revision

Viewing changes to unit_tests/test_actions_openstack_upgrade.py

  • Committer: David Ames
  • Date: 2015-10-08 22:29:33 UTC
  • mfrom: (185.2.1 keystone)
  • Revision ID: david.ames@canonical.com-20151008222933-y8gjtjhdg71kqfbr
[coreycb, r=thedac] Action managed upgrade

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
os.environ['JUJU_UNIT_NAME'] = 'keystone'
 
5
 
 
6
with patch('keystone_utils.register_configs') as register_configs:
 
7
    import openstack_upgrade
 
8
    import keystone_hooks as hooks
 
9
 
 
10
from test_utils import (
 
11
    CharmTestCase
 
12
)
 
13
 
 
14
TO_PATCH = [
 
15
    'config_changed',
 
16
    'do_openstack_upgrade',
 
17
]
 
18
 
 
19
 
 
20
class TestCinderUpgradeActions(CharmTestCase):
 
21
 
 
22
    def setUp(self):
 
23
        super(TestCinderUpgradeActions, self).setUp(openstack_upgrade,
 
24
                                                    TO_PATCH)
 
25
 
 
26
    @patch.object(hooks, 'register_configs')
 
27
    @patch('charmhelpers.contrib.openstack.utils.config')
 
28
    @patch('charmhelpers.contrib.openstack.utils.action_set')
 
29
    @patch('charmhelpers.contrib.openstack.utils.git_install_requested')
 
30
    @patch('charmhelpers.contrib.openstack.utils.openstack_upgrade_available')
 
31
    def test_openstack_upgrade_true(self, upgrade_avail, git_requested,
 
32
                                    action_set, config, reg_configs):
 
33
        git_requested.return_value = False
 
34
        upgrade_avail.return_value = True
 
35
        config.return_value = True
 
36
 
 
37
        openstack_upgrade.openstack_upgrade()
 
38
 
 
39
        self.assertTrue(self.do_openstack_upgrade.called)
 
40
        self.assertTrue(self.config_changed.called)
 
41
 
 
42
    @patch.object(hooks, 'register_configs')
 
43
    @patch('charmhelpers.contrib.openstack.utils.config')
 
44
    @patch('charmhelpers.contrib.openstack.utils.action_set')
 
45
    @patch('charmhelpers.contrib.openstack.utils.git_install_requested')
 
46
    @patch('charmhelpers.contrib.openstack.utils.openstack_upgrade_available')
 
47
    def test_openstack_upgrade_false(self, upgrade_avail, git_requested,
 
48
                                     action_set, config, reg_configs):
 
49
        git_requested.return_value = False
 
50
        upgrade_avail.return_value = True
 
51
        config.return_value = False
 
52
 
 
53
        openstack_upgrade.openstack_upgrade()
 
54
 
 
55
        self.assertFalse(self.do_openstack_upgrade.called)
 
56
        self.assertFalse(self.config_changed.called)