← Back to branch summary

~openstack-charmers-next/charms/xenial/swift-storage/trunk

« back to all changes in this revision

Viewing changes to unit_tests/test_actions_openstack_upgrade.py

  • Committer: David Ames
  • Date: 2015-09-22 22:58:18 UTC
  • mto: This revision was merged to the branch mainline in revision 83.
  • Revision ID: david.ames@canonical.com-20150922225818-rfkmptgcgxajp0zx
Action Managed Upgrades

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'] = 'swift-storge'
 
5
 
 
6
with patch('charmhelpers.core.hookenv.config') as config:
 
7
    config.return_value = 'swift-storage'
 
8
    import actions.openstack_upgrade as openstack_upgrade
 
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 TestSwiftStorageUpgradeActions(CharmTestCase):
 
21
 
 
22
    def setUp(self):
 
23
        super(TestSwiftStorageUpgradeActions, self).setUp(openstack_upgrade,
 
24
                                                          TO_PATCH)
 
25
 
 
26
    @patch('actions.charmhelpers.contrib.openstack.utils.config')
 
27
    @patch('actions.charmhelpers.contrib.openstack.utils.action_set')
 
28
    @patch('actions.charmhelpers.contrib.openstack.utils.'
 
29
           'git_install_requested')
 
30
    @patch('actions.charmhelpers.contrib.openstack.utils.'
 
31
           'openstack_upgrade_available')
 
32
    def test_openstack_upgrade_true(self, upgrade_avail, git_requested,
 
33
                                    action_set, config):
 
34
        git_requested.return_value = False
 
35
        upgrade_avail.return_value = True
 
36
        config.return_value = True
 
37
 
 
38
        openstack_upgrade.openstack_upgrade()
 
39
 
 
40
        self.assertTrue(self.do_openstack_upgrade.called)
 
41
        self.assertTrue(self.config_changed.called)
 
42
 
 
43
    @patch('actions.charmhelpers.contrib.openstack.utils.config')
 
44
    @patch('actions.charmhelpers.contrib.openstack.utils.action_set')
 
45
    @patch('actions.charmhelpers.contrib.openstack.utils.'
 
46
           'git_install_requested')
 
47
    @patch('actions.charmhelpers.contrib.openstack.utils.'
 
48
           'openstack_upgrade_available')
 
49
    def test_openstack_upgrade_false(self, upgrade_avail, git_requested,
 
50
                                     action_set, config):
 
51
        git_requested.return_value = False
 
52
        upgrade_avail.return_value = True
 
53
        config.return_value = False
 
54
 
 
55
        openstack_upgrade.openstack_upgrade()
 
56
 
 
57
        self.assertFalse(self.do_openstack_upgrade.called)
 
58
        self.assertFalse(self.config_changed.called)