← Back to branch summary

~1chb1n/charms/trusty/cinder/15.10-stable-flip-tests-helper-syncs

« 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:19:13 UTC
  • Revision ID: james.page@ubuntu.com-20151022131913-02u2l9s2fa0xtbio
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
os.environ['JUJU_UNIT_NAME'] = 'cinder'
 
5
 
 
6
with patch('cinder_utils.register_configs') as register_configs:
 
7
    import openstack_upgrade
 
8
 
 
9
from test_utils import (
 
10
    CharmTestCase
 
11
)
 
12
 
 
13
TO_PATCH = [
 
14
    'config_changed',
 
15
    'do_openstack_upgrade',
 
16
    'relation_set',
 
17
    'relation_ids',
 
18
    'uuid'
 
19
]
 
20
 
 
21
 
 
22
class TestCinderUpgradeActions(CharmTestCase):
 
23
 
 
24
    def setUp(self):
 
25
        super(TestCinderUpgradeActions, self).setUp(openstack_upgrade,
 
26
                                                    TO_PATCH)
 
27
 
 
28
    @patch('charmhelpers.contrib.openstack.utils.juju_log')
 
29
    @patch('charmhelpers.contrib.openstack.utils.config')
 
30
    @patch('charmhelpers.contrib.openstack.utils.action_set')
 
31
    @patch('charmhelpers.contrib.openstack.utils.git_install_requested')
 
32
    @patch('charmhelpers.contrib.openstack.utils.openstack_upgrade_available')
 
33
    def test_openstack_upgrade_true(self, upgrade_avail, git_requested,
 
34
                                    action_set, config, log):
 
35
        git_requested.return_value = False
 
36
        upgrade_avail.return_value = True
 
37
        config.return_value = True
 
38
        self.relation_ids.return_value = ['relid1']
 
39
        self.uuid.uuid4.return_value = 12345
 
40
 
 
41
        openstack_upgrade.openstack_upgrade()
 
42
 
 
43
        self.assertTrue(self.do_openstack_upgrade.called)
 
44
        self.assertTrue(self.relation_ids.called)
 
45
        self.relation_set.assert_called_with(relation_id='relid1',
 
46
                                             upgrade_nonce=12345)
 
47
        self.assertTrue(self.config_changed.called)
 
48
 
 
49
    @patch('charmhelpers.contrib.openstack.utils.juju_log')
 
50
    @patch('charmhelpers.contrib.openstack.utils.config')
 
51
    @patch('charmhelpers.contrib.openstack.utils.action_set')
 
52
    @patch('charmhelpers.contrib.openstack.utils.git_install_requested')
 
53
    @patch('charmhelpers.contrib.openstack.utils.openstack_upgrade_available')
 
54
    def test_openstack_upgrade_false(self, upgrade_avail, git_requested,
 
55
                                     action_set, config, log):
 
56
        git_requested.return_value = False
 
57
        upgrade_avail.return_value = True
 
58
        config.return_value = False
 
59
 
 
60
        openstack_upgrade.openstack_upgrade()
 
61
 
 
62
        self.assertFalse(self.do_openstack_upgrade.called)
 
63
        self.assertFalse(self.relation_ids.called)
 
64
        self.assertFalse(self.relation_set.called)
 
65
        self.assertFalse(self.config_changed.called)