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_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_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_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)