def _enroll(builder, oauth, meta, config_dict, secret, key, lets_connect): # type : (Gtk.Builder, oauth, Metadata, config_dict, secret, key, bool) -> None dialog = builder.get_object('totp-enroll-dialog') error_label = builder.get_object('totp-error-label') cancel_button = builder.get_object('totp-cancel-button') submit_button = builder.get_object('totp-submit-button') try: two_factor_enroll_totp(oauth, meta.api_base_uri, secret=secret, key=key) except Exception as e: error = e GLib.idle_add(lambda: error_label.set_markup( '<span color="red" size="large">{}</span>'.format(error))) GLib.idle_add(lambda: submit_button.set_sensitive(True)) GLib.idle_add(lambda: cancel_button.set_sensitive(True)) raise else: GLib.idle_add(lambda: finalizing_step(meta=meta, builder=builder, config_dict=config_dict, lets_connect=lets_connect)) GLib.idle_add(lambda: dialog.hide())
def test_2fa_enroll(self): username, password = online_tests disable_2fa(username, password, totp_secret=TOTP_SECRET, base_url=INSTANCE_URI) oauth, meta = get_oauth_token(username, password, instance_uri=INSTANCE_URI) two_factor_enroll_totp(oauth, meta.api_base_uri, secret=TOTP_SECRET, key=TOTP(TOTP_SECRET).now()) disable_2fa(username, password, totp_secret=TOTP_SECRET, base_url=INSTANCE_URI)