Added state_advance test for buttons

Author: chaosphere2112

testing/vcs/vtk_ui/CMakeLists.txt

@@ -173,25 +173,25 @@ cdat_add_test(test_vtk_ui_textbox_blank_text
   ${TEST_DIR}/test_vtk_ui_textbox_blank_text.py
 )
 
-add_test(test_vtk_ui_toolbar_label
+cdat_add_test(test_vtk_ui_toolbar_label
   "${PYTHON_EXECUTABLE}"
   ${TEST_DIR}/test_vtk_ui_toolbar_label.py
   ${BASELINE_DIR}/test_vtk_ui_toolbar_label.png
 )
 
-add_test(test_vtk_ui_toolbar_open
+cdat_add_test(test_vtk_ui_toolbar_open
   "${PYTHON_EXECUTABLE}"
   ${TEST_DIR}/test_vtk_ui_toolbar_open.py
   ${BASELINE_DIR}/test_vtk_ui_toolbar_open.png
 )
 
-add_test(test_vtk_ui_toolbar_close
+cdat_add_test(test_vtk_ui_toolbar_close
   "${PYTHON_EXECUTABLE}"
   ${TEST_DIR}/test_vtk_ui_toolbar_close.py
   ${BASELINE_DIR}/test_vtk_ui_toolbar_close.png
 )
 
-add_test(test_vtk_ui_toggle_button_get_text
+cdat_add_test(test_vtk_ui_toggle_button_get_text
   "${PYTHON_EXECUTABLE}"
   ${TEST_DIR}/test_vtk_ui_toggle_button_get_text.py
 )
@@ -207,3 +207,8 @@ cdat_add_test(test_vtk_ui_toolbar_in_toolbar_closed
   ${TEST_DIR}/test_vtk_ui_toolbar_in_toolbar_closed.py
   ${BASELINE_DIR}/test_vtk_ui_toolbar_in_toolbar_closed.png
 )
+
+cdat_add_test(test_vtk_ui_button_state_advance
+  "${PYTHON_EXECUTABLE}"
+  ${TEST_DIR}/test_vtk_ui_button_state_advance.py
+)

testing/vcs/vtk_ui/test_vtk_ui_button_state_advance.py

@@ -0,0 +1,32 @@
+"""
+Test button state change
+"""
+import vcs.vtk_ui
+
+
+from vtk_ui_test import vtk_ui_test
+
+class test_vtk_ui_button_state_advance(vtk_ui_test):
+    def do_test(self):
+        self.win.SetSize(100, 100)
+        states = [vcs.vtk_ui.ButtonState(label="State %d" % i, fgcolor=(.1 * i + .5, .1 * i + .5, .1 * i + .5), bgcolor=(.5 - .1 * i,.5 - .1 * i,.5 - .1 * i)) for i in range(5)]
+
+        b = vcs.vtk_ui.Button(self.inter, states=states, action=self.pass_me, left=0, top=0)
+        b.show()
+
+        b.set_state(1)
+        b.repr.NextState()
+        b.widget.InvokeEvent("StateChangedEvent")
+
+
+    def pass_me(self, state):
+        if state == 2:
+            print "Button action executed"
+            self.passed = 0
+        else:
+            print state, "Action passed inaccurate state"
+            from sys import exit
+            sys.exit(1)
+
+if __name__ == "__main__":
+    test_vtk_ui_button_state_advance().test()