[pitivi] tests: Move effects

[Alexandru Băluț <alexbalut src gnome org>]

commit 26f1a5d46855640e31eaa9f378c3377289cbf9ee
Author: Alexandru Băluț <alexandru balut gmail com>
Date:   Sat Jan 26 22:57:53 2019 +0100

    tests: Move effects

 tests/test_custom_effect_ui.py | 129 -----------------------------------------
 tests/test_effects.py          | 112 ++++++++++++++++++++++++++++++++++-
 2 files changed, 110 insertions(+), 131 deletions(-)
---
diff --git a/tests/test_effects.py b/tests/test_effects.py
index bf995625..35ebaa72 100644
--- a/tests/test_effects.py
+++ b/tests/test_effects.py
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Pitivi video editor
 # Copyright (c) 2016, Alex Băluț <alexandru balut gmail com>
+# Copyright (c) 2017, Suhas Nayak <suhas2go gmail com>
 #
 # This program is free software; you can redistribute it and/or
 # modify it under the terms of the GNU Lesser General Public
@@ -17,11 +18,20 @@
 # Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
 # Boston, MA 02110-1301, USA.
 """Tests for the effects module."""
+# pylint: disable=attribute-defined-outside-init,protected-access
+import os
+
 from gi.repository import GES
+from gi.repository import Gtk
 
 from pitivi.effects import AUDIO_EFFECT
 from pitivi.effects import EffectInfo
+from pitivi.effects import EffectsPropertiesManager
+from pitivi.effects import PROPS_TO_IGNORE
 from pitivi.effects import VIDEO_EFFECT
+from pitivi.utils.widgets import DynamicWidget
+from pitivi.utils.widgets import GstElementSettingsWidget
+from pitivi.utils.widgets import NumericWidget
 from tests import common
 
 
@@ -52,8 +62,8 @@ class EffectInfoTest(common.TestCase):
         ges_timeline.append_layer()
         ges_layer, = ges_timeline.get_layers()
         ges_layer.add_clip(ges_clip)
-        track_elements = dict([(track_element.get_track_type(), track_element)
-                               for track_element in ges_clip.get_children(recursive=True)])
+        track_elements = {track_element.get_track_type(): track_element
+                          for track_element in ges_clip.get_children(recursive=True)}
         audio_track_element = track_elements[GES.TrackType.AUDIO]
         video_track_element = track_elements[GES.TrackType.VIDEO]
 
@@ -64,3 +74,101 @@ class EffectInfoTest(common.TestCase):
         effect_info = EffectInfo(None, VIDEO_EFFECT, None, None, None)
         self.assertFalse(effect_info.good_for_track_element(audio_track_element))
         self.assertTrue(effect_info.good_for_track_element(video_track_element))
+
+
+class EffectsPropertiesManagerTest(common.TestCase):
+    """Tests for the EffectsPropertiesManager class."""
+
+    def create_alpha_widget_cb(self, unused_manager, unused_container, unused_effect, widgets):
+        """Handles the request to create an effect widget."""
+        self.builder = Gtk.Builder()
+        path = os.path.join(os.path.dirname(__file__), "plugins", "test_alpha.ui")
+        self.builder.add_objects_from_file(path, widgets)
+        self.element_settings_widget.mapBuilder(self.builder)
+        return self.builder.get_object("GstAlpha::black-sensitivity")
+
+    def _register_alpha_widget(self, widgets):
+        """Sets up an EffectsPropertiesManager instance to create custom effect UI."""
+        self.alpha_effect = GES.Effect.new("alpha")
+        self.prop_name = "black-sensitivity"
+        _, _, self.prop = self.alpha_effect.lookup_child(self.prop_name)
+
+        self.effects_prop_manager = EffectsPropertiesManager(self.app)
+        self.effects_prop_manager.connect("create-widget", self.create_alpha_widget_cb, widgets)
+        self.element_settings_widget = GstElementSettingsWidget(self.alpha_effect, PROPS_TO_IGNORE)
+
+        self.effects_prop_manager.emit("create-widget", self.element_settings_widget, self.alpha_effect)
+        self.effects_prop_manager._connectAllWidgetCallbacks(self.element_settings_widget, self.alpha_effect)
+        self.effects_prop_manager._postConfiguration(self.alpha_effect, self.element_settings_widget)
+
+    def test_wrapping(self):
+        """Checks UI updating results in updating the effect."""
+        self.app = common.create_pitivi_mock()
+        self._register_alpha_widget(("black_sens_adjustment", "GstAlpha::black-sensitivity"))
+
+        # Check if the widget is wrapped correctly
+        wrapped_spin_button = self.element_settings_widget.properties[self.prop]
+        self.assertTrue(isinstance(wrapped_spin_button, DynamicWidget))
+        self.assertTrue(isinstance(wrapped_spin_button, NumericWidget))
+
+        # Check if the wrapper has the correct default value
+        self.assertEqual(self.prop.default_value, wrapped_spin_button.getWidgetDefault())
+
+        # Check if the callbacks are functioning
+        value = (1 + self.prop.default_value) % self.prop.maximum
+        wrapped_spin_button.setWidgetValue(value)
+        self.assertEqual(wrapped_spin_button.getWidgetValue(), value)
+        _, prop_value = self.alpha_effect.get_child_property(self.prop_name)
+        self.assertEqual(prop_value, value)
+
+    def test_prop_keyframe(self):
+        """Checks the keyframe button effect."""
+        uri = common.get_sample_uri("tears_of_steel.webm")
+        asset = GES.UriClipAsset.request_sync(uri)
+        ges_clip = asset.extract()
+
+        # Add the clip to a timeline so it gets tracks.
+        timeline = common.create_timeline_container()
+        self.app = timeline.app
+        ges_timeline = timeline.ges_timeline
+        ges_timeline.append_layer()
+        ges_layer, = ges_timeline.get_layers()
+        ges_layer.add_clip(ges_clip)
+
+        self._register_alpha_widget(
+            ("black_sens_adjustment", "GstAlpha::black-sensitivity", 
"GstAlpha::black-sensitivity::keyframe"))
+        ges_clip.add(self.alpha_effect)
+        track_element = 
self.element_settings_widget._GstElementSettingsWidget__get_track_element_of_same_type(
+            self.alpha_effect)
+        prop_keyframe_button = \
+            
list(self.element_settings_widget._GstElementSettingsWidget__widgets_by_keyframe_button.keys())[0]
+
+        # Control the self.prop property on the timeline
+        prop_keyframe_button.set_active(True)
+        self.assertEqual(track_element.ui_element._TimelineElement__controlledProperty, self.prop)
+        # Revert to controlling the default property
+        prop_keyframe_button.set_active(False)
+        self.assertNotEqual(track_element.ui_element._TimelineElement__controlledProperty, self.prop)
+
+    def test_prop_reset(self):
+        """Checks the reset button resets the property."""
+        self.app = common.create_pitivi_mock()
+        self._register_alpha_widget(
+            ("black_sens_adjustment", "GstAlpha::black-sensitivity", "GstAlpha::black-sensitivity::reset", 
"image1"))
+        wrapped_spin_button = self.element_settings_widget.properties[self.prop]
+        _, prop_value = self.alpha_effect.get_child_property(self.prop_name)
+        self.assertEqual(self.prop.default_value, prop_value)
+        self.assertEqual(self.prop.default_value, wrapped_spin_button.getWidgetValue())
+
+        # Set the property value to a different value than the default
+        wrapped_spin_button.setWidgetValue((1 + self.prop.default_value) % self.prop.maximum)
+        _, prop_value = self.alpha_effect.get_child_property(self.prop_name)
+        self.assertEqual(prop_value, (1 + self.prop.default_value) % self.prop.maximum)
+
+        # Reset the value of the property to default
+        prop_reset_button = \
+            list(self.element_settings_widget._GstElementSettingsWidget__widgets_by_reset_button.keys())[0]
+        prop_reset_button.clicked()
+        _, prop_value = self.alpha_effect.get_child_property(self.prop_name)
+        self.assertEqual(self.prop.default_value, prop_value)
+        self.assertEqual(self.prop.default_value, wrapped_spin_button.getWidgetValue())