def test_create_repo_ubuntu(self):
task_config = {'id': 'upload_mos_repos',
'type': 'upload_file',
'role': '*'}
self.cluster.release.operating_system = consts.RELEASE_OS.ubuntu
task = tasks_serializer.UploadMOSRepo(
task_config, self.cluster, self.nodes)
serialized = list(task.serialize())
self.assertEqual(len(serialized), 17)
self.assertEqual(serialized[0]['type'], 'shell')
self.assertEqual(
serialized[0]['parameters']['cmd'], '> /etc/apt/sources.list')
self.assertEqual(serialized[1]['type'], 'upload_file')
self.assertEqual(serialized[2]['type'], 'upload_file')
self.assertEqual(serialized[3]['type'], 'upload_file')
self.assertEqual(serialized[4]['type'], 'upload_file')
self.assertEqual(serialized[5]['type'], 'upload_file')
self.assertEqual(serialized[6]['type'], 'upload_file')
self.assertEqual(serialized[7]['type'], 'upload_file')
self.assertEqual(serialized[8]['type'], 'upload_file')
self.assertEqual(serialized[9]['type'], 'upload_file')
self.assertEqual(serialized[10]['type'], 'upload_file')
self.assertEqual(serialized[11]['type'], 'upload_file')
self.assertEqual(serialized[12]['type'], 'upload_file')
self.assertEqual(serialized[13]['type'], 'upload_file')
self.assertEqual(serialized[14]['type'], 'upload_file')
self.assertEqual(serialized[15]['type'], 'upload_file')
self.assertEqual(serialized[16]['type'], 'shell')
self.assertEqual(serialized[16]['parameters']['cmd'], 'apt-get update')
self.assertItemsEqual(serialized[3]['uids'], self.all_uids)
def test_create_repo_centos(self):
"""Verify that repository is created with correct metadata."""
task_config = {'id': 'upload_mos_repos',
'type': 'upload_file',
'role': '*'}
self.cluster.release.operating_system = consts.RELEASE_OS.centos
task = tasks_serializer.UploadMOSRepo(
task_config, self.cluster, self.nodes)
serialized = list(task.serialize())
self.assertEqual(len(serialized), 2)
self.assertEqual(serialized[0]['type'], 'upload_file')
self.assertEqual(serialized[1]['type'], 'shell')
self.assertEqual(serialized[1]['parameters']['cmd'], 'yum clean all')
self.assertItemsEqual(serialized[1]['uids'], self.all_uids)