Merge lp:~smoser/cloud-init/trunk.1536706 into lp:~cloud-init-dev/cloud-init/trunk

Proposed by Scott Moser
Status: Merged
Merged at revision: 1205
Proposed branch: lp:~smoser/cloud-init/trunk.1536706
Merge into: lp:~cloud-init-dev/cloud-init/trunk
Diff against target: 71 lines (+17/-6)
2 files modified
cloudinit/config/cc_rh_subscription.py (+10/-4)
tests/unittests/test_rh_subscription.py (+7/-2)
To merge this branch: bzr merge lp:~smoser/cloud-init/trunk.1536706
Reviewer Review Type Date Requested Status
cloud-init Commiters Pending
Review via email: mp+291656@code.launchpad.net
To post a comment you must log in.

Preview Diff

[H/L] Next/Prev Comment, [J/K] Next/Prev File, [N/P] Next/Prev Hunk
=== modified file 'cloudinit/config/cc_rh_subscription.py'
--- cloudinit/config/cc_rh_subscription.py 2016-03-04 15:51:05 +0000
+++ cloudinit/config/cc_rh_subscription.py 2016-04-12 17:01:26 +0000
@@ -19,10 +19,14 @@
19from cloudinit import util19from cloudinit import util
2020
2121
22def handle(_name, cfg, _cloud, log, _args):22def handle(name, cfg, _cloud, log, _args):
23 sm = SubscriptionManager(cfg)23 sm = SubscriptionManager(cfg)
24 sm.log = log24 sm.log = log
25 if not sm.is_registered:25 if not sm.is_configured():
26 log.debug("%s: module not configured.", name)
27 return None
28
29 if not sm.is_registered():
26 try:30 try:
27 verify, verify_msg = sm._verify_keys()31 verify, verify_msg = sm._verify_keys()
28 if verify is not True:32 if verify is not True:
@@ -95,7 +99,6 @@
95 self.disable_repo = self.rhel_cfg.get('disable-repo')99 self.disable_repo = self.rhel_cfg.get('disable-repo')
96 self.servicelevel = self.rhel_cfg.get('service-level')100 self.servicelevel = self.rhel_cfg.get('service-level')
97 self.subman = ['subscription-manager']101 self.subman = ['subscription-manager']
98 self.is_registered = self._is_registered()
99102
100 def log_success(self, msg):103 def log_success(self, msg):
101 '''Simple wrapper for logging info messages. Useful for unittests'''104 '''Simple wrapper for logging info messages. Useful for unittests'''
@@ -134,7 +137,7 @@
134 return False, no_auto137 return False, no_auto
135 return True, None138 return True, None
136139
137 def _is_registered(self):140 def is_registered(self):
138 '''141 '''
139 Checks if the system is already registered and returns142 Checks if the system is already registered and returns
140 True if so, else False143 True if so, else False
@@ -400,3 +403,6 @@
400 self.log.debug("Disabled the following repos: %s" %403 self.log.debug("Disabled the following repos: %s" %
401 (", ".join(disable_list)).replace('--disable=', ''))404 (", ".join(disable_list)).replace('--disable=', ''))
402 return True405 return True
406
407 def is_configured(self):
408 return bool((self.userid and self.password) or self.activation_key)
403409
=== modified file 'tests/unittests/test_rh_subscription.py'
--- tests/unittests/test_rh_subscription.py 2015-05-29 14:18:49 +0000
+++ tests/unittests/test_rh_subscription.py 2016-04-12 17:01:26 +0000
@@ -126,7 +126,8 @@
126 'enable-repo': 'not_a_list'126 'enable-repo': 'not_a_list'
127 }}127 }}
128 config_badkey = {'rh_subscription':128 config_badkey = {'rh_subscription':
129 {'activation_key': 'abcdef1234',129 {'activation-key': 'abcdef1234',
130 'fookey': 'bar',
130 'org': '123',131 'org': '123',
131 }}132 }}
132133
@@ -138,7 +139,11 @@
138 '''139 '''
139 Attempt to register without the password key/value140 Attempt to register without the password key/value
140 '''141 '''
141 self.input_is_missing_data(self.config_no_password)142 self.SM._sub_man_cli = mock.MagicMock(
143 side_effect=[util.ProcessExecutionError, (self.reg, 'bar')])
144 self.handle(self.name, self.config_no_password, self.cloud_init,
145 self.log, self.args)
146 self.assertEqual(self.SM._sub_man_cli.call_count, 0)
142147
143 def test_no_org(self):148 def test_no_org(self):
144 '''149 '''