def uiobj(*args, **kwargs): class _Object(object): pass o = _Object() o.info = uia_element_info(*args, **kwargs) return o
def set_element_find_result(mocks, **uia_attr): elem = MagicMock() elem.info = uia_element_info(**uia_attr) mocks.device.return_value = [elem] mocks.device.dump.return_value = create_dump_xml(elem.info) mocks.uiroot.nametowidget( 'sidebar.refresh_button').process_event(None, None) return elem