def test_serialzize_rados_wo_ceph(self): task_config = {'id': 'restart_radosgw', 'type': 'shell', 'role': ['controller', 'primary-controller'], 'stage': 'post-deployment', 'parameters': {'cmd': '/cmd.sh', 'timeout': 60}} task = tasks_serializer.RestartRadosGW( task_config, self.cluster, self.nodes) self.assertFalse(task.should_execute())
def test_serialize_rados_with_ceph(self): task_config = {'id': 'restart_radosgw', 'type': 'shell', 'role': ['controller', 'primary-controller'], 'stage': 'post-deployment', 'parameters': {'cmd': '/cmd.sh', 'timeout': 60}} self.nodes.append(self.env.create_node( roles=['ceph-osd'], cluster_id=self.cluster.id)) task = tasks_serializer.RestartRadosGW( task_config, self.cluster, self.nodes) serialized = list(task.serialize()) self.assertEqual(len(serialized), 1) self.assertEqual(serialized[0]['type'], 'shell') self.assertEqual( serialized[0]['parameters']['cmd'], task_config['parameters']['cmd'])