def test_get_all_attributes_by_interface_with_enabled_plugin(self): nic_config = self.env.get_default_plugin_nic_config() plugin = self.env.create_plugin( name='plugin_a_with_nic_attributes', package_version='5.0.0', nic_attributes_metadata=nic_config) cluster = self._create_test_cluster() # create node with 1 interface for easy testing node = self.env.create_nodes_w_interfaces_count( 1, 1, **{"cluster_id": cluster.id})[0] interface = node.nic_interfaces[0] nic_plugin_id = node.node_nic_interface_cluster_plugins[0].id ClusterPlugin.set_attributes(cluster.id, plugin.id, enabled=True) attributes = NodeNICInterfaceClusterPlugin.\ get_all_enabled_attributes_by_interface(interface) expected_attributes = { 'plugin_a_with_nic_attributes': { 'metadata': { 'label': 'Test base plugin', 'nic_plugin_id': nic_plugin_id, 'class': 'plugin'}, 'plugin_name_text': { 'value': 'value', 'type': 'text', 'description': 'Some description', 'weight': 25, 'label': 'label'}}} self.assertEqual(expected_attributes, attributes)
def test_get_all_attributes_by_bond_with_enabled_plugin(self): plugin_bond_config = self.env.get_default_plugin_bond_config() plugin = self.env.create_plugin( name='plugin_a_with_bond_attributes', package_version='5.0.0', bond_attributes_metadata=plugin_bond_config) cluster = self._create_test_cluster() node = self.env.create_nodes_w_interfaces_count( 1, 2, **{"cluster_id": cluster.id})[0] bond_config = { 'type__': {'value': consts.BOND_TYPES.linux}, 'mode': {'value': {'value': consts.BOND_MODES.balance_rr}}} nic_names = [iface.name for iface in node.nic_interfaces] self.env.make_bond_via_api( 'lnx_bond', '', nic_names, node.id, attrs=bond_config) bond = node.bond_interfaces[0] bond_plugin_id = node.node_bond_interface_cluster_plugins[0].id ClusterPlugin.set_attributes(cluster.id, plugin.id, enabled=True) attributes = NodeBondInterfaceClusterPlugin.\ get_all_enabled_attributes_by_bond(bond) expected_attributes = { 'plugin_a_with_bond_attributes': { 'metadata': { 'label': 'Test base plugin', 'bond_plugin_id': bond_plugin_id, 'class': 'plugin'}, 'plugin_name_text': { 'value': 'value', 'type': 'text', 'description': 'Some description', 'weight': 25, 'label': 'label'}}} self.assertEqual(expected_attributes, attributes)
def setUp(self): super(TestNailgunReceiver, self).setUp() self.cluster = self.env.create( status=consts.CLUSTER_STATUSES.operational, nodes_kwargs=[{ 'roles': ['controller'], 'status': consts.NODE_STATUSES.ready }]) for i in range(2): meta = self.env.get_default_plugin_metadata( name='name{0}'.format(i), title='title{0}'.format(i), description='description{0}'.format(i)) self.plugin = Plugin.create(meta) self.cluster.plugins.append(self.plugin) ClusterPlugin.set_attributes(self.cluster.id, self.plugin.id, enabled=True) self.task = self.env.create_task(name=consts.TASK_NAMES.deployment, status=consts.TASK_STATUSES.ready, cluster_id=self.cluster.id)
def setUp(self): super(TestNailgunReceiver, self).setUp() self.cluster = self.env.create( status=consts.CLUSTER_STATUSES.operational, nodes_kwargs=[ {'roles': ['controller'], 'status': consts.NODE_STATUSES.ready}]) for i in range(2): meta = self.env.get_default_plugin_metadata( name='name{0}'.format(i), title='title{0}'.format(i), description='description{0}'.format(i)) self.plugin = Plugin.create(meta) self.cluster.plugins.append(self.plugin) ClusterPlugin.set_attributes(self.cluster.id, self.plugin.id, enabled=True) self.task = self.env.create_task( name=consts.TASK_NAMES.deployment, status=consts.TASK_STATUSES.ready, cluster_id=self.cluster.id)
def test_is_plugin_used(self): self._create_test_plugins() cluster = self._create_test_cluster() plugin = ClusterPlugin.get_connected_plugins(cluster).first() self.assertFalse(ClusterPlugin.is_plugin_used(plugin.id)) ClusterPlugin.set_attributes(cluster.id, plugin.id, enabled=True) self.assertTrue(ClusterPlugin.is_plugin_used(plugin.id))
def test_get_enabled(self): self._create_test_plugins() cluster = self._create_test_cluster() plugin = ClusterPlugin.get_connected_plugins(cluster).first() ClusterPlugin.set_attributes(cluster.id, plugin.id, enabled=True) enabled_plugin = ClusterPlugin.get_enabled(cluster.id).first() self.assertEqual(plugin.id, enabled_plugin.id)
def test_get_enabled_plugins(self): plugin_a = self._create_plugin(**self._compat_meta) plugin_b = self._create_plugin(**self._compat_meta) ClusterPlugin.set_attributes( self.cluster.id, plugin_a.id, enabled=True) compat_plugins = ClusterPlugin.get_compatible_plugins(self.cluster) self.assertItemsEqual(compat_plugins, [plugin_a, plugin_b]) enabled_plugins = ClusterPlugin.get_enabled(self.cluster.id) self.assertItemsEqual(enabled_plugins, [plugin_a])
def test_set_plugin_attributes(self): meta = base.reflect_db_metadata() self._create_test_plugins() cluster = self._create_test_cluster() plugin = ClusterPlugin.get_connected_plugins(cluster).first() ClusterPlugin.set_attributes(cluster.id, plugin.id, enabled=True) columns = meta.tables["cluster_plugins"].c enabled = self.db.execute( sa.select([columns.enabled]).where(columns.cluster_id == cluster.id).where(columns.plugin_id == plugin.id) ).fetchone() self.assertTrue(enabled[0])
def test_set_plugin_attributes(self): meta = base.reflect_db_metadata() self._create_test_plugins() cluster = self._create_test_cluster() plugin = ClusterPlugin.get_connected_plugins(cluster).first() ClusterPlugin.set_attributes(cluster.id, plugin.id, enabled=True) columns = meta.tables['cluster_plugins'].c enabled = self.db.execute( sa.select([columns.enabled ]).where(columns.cluster_id == cluster.id).where( columns.plugin_id == plugin.id)).fetchone() self.assertTrue(enabled[0])
def test_get_connected_for_specific_plugins(self): plugin_ids = self._create_test_plugins() cluster = self._create_test_cluster() number_of_connected_plugins =\ ClusterPlugin.get_connected_plugins( cluster, plugin_ids[1:]).count() self.assertEqual(6, number_of_connected_plugins)
def test_add_nodes_for_cluster_plugin_with_empty_attributes(self): meta = base.reflect_db_metadata() self.env.create_node( cluster_id=self.cluster.id, roles=['compute'] ) plugin = Plugin.create({ 'name': 'plugin_a_with_node_attributes', 'title': 'Test Plugin', 'package_version': '5.0.0', 'version': '1.0.0', 'node_attributes_metadata': {} }) cluster_plugin = ClusterPlugin.create({ 'cluster_id': self.cluster.id, 'plugin_id': plugin.id, 'enabled': False, 'attributes': plugin.node_attributes_metadata }) NodeClusterPlugin.add_nodes_for_cluster_plugin(cluster_plugin) node_cluster_plugins = self.db.execute( meta.tables['node_cluster_plugins'].select() ).fetchall() self.assertEqual(0, len(node_cluster_plugins))
def test_get_connected_clusters(self): plugin_id = self._create_test_plugins()[0] for _ in range(2): self._create_test_cluster() number_of_connected_clusters =\ ClusterPlugin.get_connected_clusters(plugin_id).count() self.assertEqual(2, number_of_connected_clusters)
def test_get_compatible_plugins_for_new_cluster(self): plugin_a = self._create_plugin(**self._compat_meta) plugin_b = self._create_plugin(**self._compat_meta) self._create_plugin(**self._uncompat_meta) cluster = self.env.create( cluster_kwargs={ 'release_id': self.cluster.release.id, 'mode': consts.CLUSTER_MODES.ha_compact, }) compat_plugins = ClusterPlugin.get_compatible_plugins(cluster) self.assertItemsEqual(compat_plugins, [plugin_a, plugin_b])
def test_enable_plugins_by_component(self): self.env.create_plugin( name='plugin_with_test_storage', package_version='4.0.0', fuel_version=['8.0'], releases=[{ 'repository_path': 'repositories/ubuntu', 'version': '2015.1-8.3', 'os': 'ubuntu', 'mode': ['ha'], 'deployment_scripts_path': 'deployment_scripts/' }], components_metadata=self.env.get_default_components( name='storage:test_storage')) plugin = self.env.create_plugin( version='1.0.0', name='plugin_with_test_storage', package_version='4.0.0', fuel_version=['8.0'], releases=[{ 'repository_path': 'repositories/ubuntu', 'version': '2015.1-8.3', 'os': 'ubuntu', 'mode': ['ha'], 'deployment_scripts_path': 'deployment_scripts/' }], components_metadata=self.env.get_default_components( name='storage:test_storage')) cluster = self.env.create( release_kwargs={ 'operating_system': consts.RELEASE_OS.ubuntu, 'version': '2015.1-8.3' }, cluster_kwargs={ 'mode': consts.CLUSTER_MODES.ha_compact, 'api': False, 'components': ['hypervisor:test_hypervisor', 'storage:test_storage'] }) enabled_plugins = ClusterPlugin.get_enabled(cluster.id) self.assertItemsEqual([plugin], enabled_plugins)
def test_enable_plugins_by_component(self): self.env.create_plugin( name='plugin_with_test_storage', package_version='4.0.0', fuel_version=['8.0'], releases=[{ 'repository_path': 'repositories/ubuntu', 'version': '2015.1-8.3', 'os': 'ubuntu', 'mode': ['ha'], 'deployment_scripts_path': 'deployment_scripts/'}], components_metadata=self.env.get_default_components( name='storage:test_storage')) plugin = self.env.create_plugin( version='1.0.0', name='plugin_with_test_storage', package_version='4.0.0', fuel_version=['8.0'], releases=[{ 'repository_path': 'repositories/ubuntu', 'version': '2015.1-8.3', 'os': 'ubuntu', 'mode': ['ha'], 'deployment_scripts_path': 'deployment_scripts/'}], components_metadata=self.env.get_default_components( name='storage:test_storage')) cluster = self.env.create( release_kwargs={ 'operating_system': consts.RELEASE_OS.ubuntu, 'version': '2015.1-8.3'}, cluster_kwargs={ 'mode': consts.CLUSTER_MODES.ha_compact, 'api': False, 'components': [ 'hypervisor:test_hypervisor', 'storage:test_storage']}) enabled_plugins = ClusterPlugin.get_enabled(cluster.id) self.assertItemsEqual([plugin], enabled_plugins)
def validate_delete(cls, data, instance): if ClusterPlugin.is_plugin_used(instance.id): raise errors.CannotDelete( "Can't delete plugin which is enabled " "for some environment." )
def validate_with_cluster(self, **kwargs): cluster = self.cluster_mock(**kwargs) return ClusterPlugin.is_compatible(cluster, self.plugin)
def test_get_compatible_plugins(self): plugin_a = self._create_plugin(**self._compat_meta) self._create_plugin(**self._uncompat_meta) compat_plugins = ClusterPlugin.get_compatible_plugins(self.cluster) self.assertItemsEqual(compat_plugins, [plugin_a])
def validate_delete(cls, data, instance): if ClusterPlugin.is_plugin_used(instance.id): raise errors.CannotDelete("Can't delete plugin which is enabled " "for some environment.")
def test_get_connected_plugins_data(self): self._create_test_plugins() cluster = self._create_test_cluster() number_of_connected_plugins_data_items =\ ClusterPlugin.get_connected_plugins_data(cluster.id).count() self.assertEqual(7, number_of_connected_plugins_data_items)
def test_get_all_connected_plugins(self): self._create_test_plugins() cluster = self._create_test_cluster() number_of_connected_plugins =\ ClusterPlugin.get_connected_plugins(cluster).count() self.assertEqual(7, number_of_connected_plugins)