Assistant: Add get_page()

Use get_page() instead of page specific methods

1 files changed, 10 insertions, 8 deletions

diff --git a/test/gtk/assistant.py b/test/gtk/assistant.py
index 8a049a72c..d988e87fa 100644
--- a/test/gtk/assistant.py
+++ b/test/gtk/assistant.py
@@ -20,6 +20,16 @@ class TestAssistant(Assistant):
         self.add_default_page('error')
         self.add_default_page('success')
 
+        success_page = self.get_page('success')
+        success_page.set_title('Success')
+        success_page.set_heading('Success Heading')
+        success_page.set_text('This is the success text')
+
+        error_page = self.get_page('error')
+        error_page.set_title('Error')
+        error_page.set_heading('Error Heading')
+        error_page.set_text('This is the error text')
+
         self.add_button('forward', 'Forward', 'suggested-action')
         self.add_button('close', 'Close', 'destructive-action')
         self.add_button('back', 'Back')
@@ -77,17 +87,9 @@ class TestAssistant(Assistant):
             self.set_default_button('forward')
 
         elif page_name == 'success':
-            success_page = self.get_success_page()
-            success_page.set_title('Success')
-            success_page.set_heading('Success Heading')
-            success_page.set_text('This is the success text')
             self.set_default_button('forward')
 
         elif page_name == 'error':
-            error_page = self.get_error_page()
-            error_page.set_title('Error')
-            error_page.set_heading('Error Heading')
-            error_page.set_text('This is the error text')
             self.set_default_button('back')