def setUp(self): super(RegisterWidgetTests, self).setUp() self.exc_infos = [] self.excs = [] self.backend = StubBackend() expected_facts = {'fact1': 'one', 'fact2': 'two', 'system': '', 'system.uuid': 'MOCKUUID'} self.facts = StubFacts(fact_dict=expected_facts) self.reg_info = RegisterInfo() self.rs = RegisterWidget(backend=self.backend, reg_info=self.reg_info) self.rs._screens[CHOOSE_SERVER_PAGE] = Mock() self.rs._screens[CHOOSE_SERVER_PAGE].index = 0 self.rs._screens[CHOOSE_SERVER_PAGE].screens_index = 0 self.rs._screens[CHOOSE_SERVER_PAGE].button_label = "Dummy" self.rs._screens[CHOOSE_SERVER_PAGE].apply.return_value = CREDENTIALS_PAGE