@@ -80,14 +80,19 @@ class SkipThroughSettingsWizardTestCase(tests.UnityTestCase):
8080 password_page = next_page
8181 wifi_connect_page = self._test_password_page(password_page)
8282
83− locationPageEnabled, next_page = self._test_wifi_connect_page(
84− wifi_connect_page)
83+ reporting_page = None
84+ locationPageEnabled, reportingPageEnabled, next_page = self._test_wifi_connect_page(wifi_connect_page)
8585 if locationPageEnabled:
8686 location_page = next_page
87− reporting_page = self._test_location_page(location_page)
87+ if reportingPageEnabled:
88+ reporting_page = self._test_location_page(location_page)
89+ else:
90+ finish_page = next_page
91+
92+ if reporting_page is not None:
93+ finish_page = self._test_reporting_page(reporting_page)
8894 else:
89− reporting_page = next_page
90− finish_page = self._test_reporting_page(reporting_page)
95+ finish_page = next_page
9196
9297 finish_page.finish()
9398 self.assertFalse(