Add command to apply all hotfixes

upgrade/upgrade/UpgradeManager.py

@@ -173,6 +173,15 @@ def apply(self, name):
 			self.reboot()
 		return True
 
+	def applyAll(self):
+		for name, hotfix in self.list().items():
+			if hotfix['applied']:
+				continue
+			if hotfix['deployment'] != 'auto':
+				continue
+			yield name
+			self.apply(name)
+
 	def clearCache(self):
 		self.hotfixes = None
 
@@ -459,6 +468,10 @@ def runCli():
 				sys.stdout.write('%s %s\n' % ('*' if hfix['applied'] else ' ', hotfixName))
 			sys.stdout.flush()
 			sys.exit(0)
+		if hotfix == 'all':
+			for hotfixName in hotfixManager.applyAll():
+				logging.warning('Applying %s', hotfixName)
+			sys.exit(0)
 		if hotfix not in hotfixManager.list():
 			logging.error('Hotfix %s not found', hotfix)
 			sys.exit(1)

upgrade/upgrade/tests/UpgradeManagerTest.py

@@ -61,6 +61,24 @@ def testApply(self):
 		with open('/tmp/tests/helloworld', 'r') as fd:
 			self.assertEqual(fd.read(), 'Hello World!\n')
 
+	@patch.object(UpgradeManagerBase, 'reboot', reboot)
+	def testApplyAll(self):
+		autoHotfixes = ['default', 'deployAuto']
+		manualHotfixes = ['deployMalformed', 'deployManual', 'helloworld']
+		failedHotfixes = ['invalidSignature', 'scriptFailing']
+
+		self.assertEqual(self.hotfixManager.appliedHotfixes, [])
+		with self.assertRaises(RebootException):
+			for name in self.hotfixManager.applyAll():
+				# No manual hotfix may be applied
+				self.assertNotIn(name, manualHotfixes)
+		for name in autoHotfixes:
+			# Make sure all auto has been appled
+			self.assertIn(name, self.hotfixManager.appliedHotfixes)
+		for name in failedHotfixes:
+			# Make sure no failed has been marked as applied
+			self.assertNotIn(name, self.hotfixManager.appliedHotfixes)
+
 	def testDeployment(self):
 		self.assertEqual(self.__getHotFix('default')['deployment'], 'auto')
 		self.assertEqual(self.__getHotFix('deployAuto')['deployment'], 'auto')