alibi_plugins/two_factor/

mod.rs

mod backup_codes;
mod challenge;
mod codec;
mod config;
mod cookies;
mod enrollment;
mod lockout;
mod otp_flow;
mod totp;

use backup_codes::generate_backup_codes;
use backup_codes::generate_backup_codes_core;
use backup_codes::generate_numeric_string;
use backup_codes::verify_backup_code_core;
use backup_codes::view_backup_codes_core;
use challenge::begin_factor_attempt;
pub(crate) use challenge::begin_sign_in_challenge;
use challenge::finalize_pending_two_factor;
pub(crate) use challenge::inspect_trusted_device;
use challenge::load_two_factor_record;
use challenge::rearm_factor_attempt;
use challenge::resolve_two_factor_state;
use challenge::verification_error_response;
use challenge::verify_existing_session_factor;
use codec::decrypt_value;
use codec::encrypt_value;
pub use config::AccountLockoutConfig;
pub use config::SendTwoFactorOtp;
pub use config::TwoFactorConfig;
use cookies::clear_cookie_header;
use cookies::cookie_expiry;
use cookies::create_session_cookie_for_dont_remember;
use cookies::create_signed_cookie_header;
use cookies::create_trust_device_cookie_header;
use cookies::get_factor_cookie;
use cookies::read_signed_cookie;
use cookies::sign_value;
use cookies::two_factor_cookie_max_age;
use cookies::verify_factor_cookie_value;
use enrollment::disable_core;
use enrollment::enable_core;
use enrollment::mark_factor_verified;
use enrollment::parse_password_body;
use enrollment::verify_user_password;
use lockout::assert_account_not_locked;
use lockout::record_account_failure;
use lockout::reset_account_failures;
use otp_flow::send_otp_core;
use otp_flow::verify_otp_core;
use totp::generate_secret;
use totp::generate_totp_at;
use totp::get_totp_uri_core;
use totp::totp_counter;
use totp::totp_uri;
use totp::verify_totp_core;
mod backup_storage;
mod endpoint;

mod otp_storage;

mod otp;

use super::StatusResponse;
use crate::helpers::{
    SessionIssueError, get_credential_password_hash, issue_user_session_record,
    issue_user_session_with_overrides_record,
};
use alibi_core::entity::{AuthSession, AuthTwoFactor, AuthUser};
use alibi_core::utils::cookie_utils::{
    create_clear_cookie, create_session_cookie, create_session_cookie_with_max_age,
    related_cookie_name,
};
use alibi_core::wire::UserView;
use alibi_core::{
    AuthContext, AuthError, AuthRequest, AuthResponse, AuthResult, CreateTwoFactor,
    CreateVerification, RequestMeta, TwoFactor, UpdateTwoFactor, UpdateUser,
};
use async_trait::async_trait;
pub use backup_storage::{TwoFactorBackupCipher, TwoFactorBackupStorage};
use base64::{Engine as _, engine::general_purpose::URL_SAFE_NO_PAD};
use chrono::{Duration, Utc};
pub use endpoint::{
    BackupCodeVerified, BackupCodesOutput, TotpOutput, TotpVerified, TwoFactorDisabled,
    TwoFactorOtpSent,
};
use hmac::{Hmac, KeyInit, Mac};
pub use otp_storage::{TwoFactorOtpCipher, TwoFactorOtpHasher, TwoFactorOtpStorage};
use rand::RngExt;
use rand::distr::Alphanumeric;
use serde::{Deserialize, Serialize};
use std::sync::Arc;
use validator::Validate;

const TWO_FACTOR_COOKIE_SUFFIX: &str = "two_factor";

const TRUST_DEVICE_COOKIE_SUFFIX: &str = "trust_device";

const DONT_REMEMBER_COOKIE_SUFFIX: &str = "dont_remember";

const METADATA_ENABLED: &str = "two_factor.enabled";

const METADATA_OTP_ENABLED: &str = "two_factor.otp_enabled";

const METADATA_TWO_FACTOR_COOKIE_MAX_AGE: &str = "two_factor.two_factor_cookie_max_age";

const METADATA_TRUST_DEVICE_MAX_AGE: &str = "two_factor.trust_device_max_age";

const METADATA_TOTP_DISABLED: &str = "two_factor.totp_disabled";

const DEFAULT_TWO_FACTOR_COOKIE_MAX_AGE_SECS: f64 = 600.0;

const DEFAULT_TRUST_DEVICE_MAX_AGE_SECS: f64 = 2_592_000.0;

#[derive(Clone, Copy)]
struct TwoFactorCookiePolicy {
    challenge_max_age: f64,
    trust_max_age: f64,
}

const DEFAULT_TOTP_PERIOD_SECS: f64 = 30.0;

const DEFAULT_TOTP_DIGITS: f64 = 6.0;

/// Two-factor authentication plugin providing TOTP, OTP, and backup code flows.
#[derive(Clone)]
pub struct TwoFactorPlugin {
    config: TwoFactorConfig,
}

/// Password authority and enrollment method for enabling the current user's factor.
#[derive(Debug, Default, Deserialize, Serialize, Validate)]
pub struct EnableTwoFactorInput {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub password: Option<String>,
    #[serde(default)]
    pub method: TwoFactorMethod,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub issuer: Option<String>,
}

/// The supported factor enrollment methods.
#[derive(Debug, Default, Deserialize, Serialize, PartialEq)]
#[serde(rename_all = "lowercase")]
pub enum TwoFactorMethod {
    Otp,
    #[default]
    Totp,
}

/// Password authority for disabling the current user's factor.
#[derive(Debug, Default, Deserialize, Serialize, Validate)]
pub struct DisableTwoFactorInput {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub password: Option<String>,
}

/// Password authority for retrieving the current user's stored authenticator URI.
#[derive(Debug, Default, Deserialize, Serialize, Validate)]
pub struct GetTotpUriInput {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub password: Option<String>,
}

/// Authenticator code and optional trusted-device preference.
#[derive(Debug, Deserialize, Serialize, Validate)]
pub struct VerifyTotpRequest {
    pub code: String,
    #[serde(rename = "trustDevice", skip_serializing_if = "Option::is_none")]
    pub trust_device: Option<bool>,
}

/// Public input for direct authenticator verification.
pub type VerifyTotpInput = VerifyTotpRequest;

#[derive(Deserialize)]
struct SendOtpRequest {
    #[serde(rename = "trustDevice")]
    _trust_device: Option<bool>,
}

impl crate::authentication_helpers::RequestBody for SendOtpRequest {
    const FIELDS: &'static [crate::authentication_helpers::JsonField] =
        &[crate::authentication_helpers::JsonField {
            name: "trustDevice",
            kind: crate::authentication_helpers::JsonFieldKind::Boolean,
            required: false,
        }];
}

/// Delivered code and optional trusted-device preference.
#[derive(Debug, Deserialize, Serialize, Validate)]
pub struct VerifyOtpRequest {
    pub code: String,
    #[serde(rename = "trustDevice", skip_serializing_if = "Option::is_none")]
    pub trust_device: Option<bool>,
}
/// Public input for direct delivered-code verification.
pub type VerifyOtpInput = VerifyOtpRequest;

/// Password admission for regenerating the current owner's recovery codes.
#[derive(Debug, Deserialize, Serialize, Validate)]
pub struct GenerateBackupCodesRequest {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub password: Option<String>,
}
/// Public input for direct backup-code regeneration.
pub type GenerateBackupCodesInput = GenerateBackupCodesRequest;

/// Recovery code and optional session/device publication preferences.
#[derive(Debug, Deserialize, Serialize, Validate)]
pub struct VerifyBackupCodeRequest {
    pub code: String,
    #[serde(rename = "disableSession", skip_serializing_if = "Option::is_none")]
    pub disable_session: Option<bool>,
    #[serde(rename = "trustDevice", skip_serializing_if = "Option::is_none")]
    pub trust_device: Option<bool>,
}
/// Public input for direct recovery-code verification.
pub type VerifyBackupCodeInput = VerifyBackupCodeRequest;

/// A generated authenticator enrollment or an immediately enabled OTP factor.
#[derive(Debug, Serialize, Deserialize)]
#[serde(tag = "method", rename_all = "lowercase")]
pub enum TwoFactorEnabled {
    Otp,
    Totp {
        #[serde(rename = "totpURI")]
        totp_uri: String,
        #[serde(rename = "backupCodes")]
        backup_codes: Vec<String>,
    },
}

/// An authenticator URI recovered from the caller's stored factor.
#[derive(Debug, Serialize, Deserialize)]
pub struct TotpUriOutput {
    #[serde(rename = "totpURI")]
    pub totp_uri: String,
}

#[derive(Debug, Serialize)]
pub(crate) struct SessionTokenResponse<U> {
    token: String,
    user: U,
}

#[derive(Debug, Serialize)]
struct BackupVerificationResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    token: Option<String>,
    user: UserView,
}

impl From<SessionTokenResponse<UserView>> for BackupVerificationResponse {
    fn from(response: SessionTokenResponse<UserView>) -> Self {
        Self {
            token: Some(response.token),
            user: response.user,
        }
    }
}

#[derive(Debug, Serialize)]
pub(crate) struct BackupCodesResponse {
    status: bool,
    #[serde(rename = "backupCodes")]
    backup_codes: Vec<String>,
}

#[derive(Debug, Serialize)]
pub(crate) struct TwoFactorRedirectResponse {
    #[serde(rename = "twoFactorRedirect")]
    two_factor_redirect: bool,
    /// Second factors this user can actually complete, so the client knows
    /// which challenge to present.
    #[serde(rename = "twoFactorMethods")]
    two_factor_methods: Vec<&'static str>,
}

struct PendingTwoFactorState<S: alibi_core::AuthSchema> {
    user: alibi_core::AdapterRecord<S::User>,
    verification: alibi_core::verification::VerificationSnapshot,
    key: String,
    dont_remember: bool,
}

enum ResolvedTwoFactorState<S: alibi_core::AuthSchema> {
    Session {
        user: alibi_core::AuthenticatedUser<S>,
        session: Box<alibi_core::wire::SessionView>,
        key: String,
    },
    Pending(PendingTwoFactorState<S>),
}

pub(crate) struct SignInTwoFactorRedirect {
    pub response: TwoFactorRedirectResponse,
    pub set_cookie_headers: Vec<String>,
}

pub(crate) struct TrustedDeviceCheck {
    pub trusted: bool,
    pub set_cookie_headers: Vec<String>,
}

impl TwoFactorPlugin {
    /// Install a custom OTP sender.
    #[must_use]
    pub fn custom_send_otp(mut self, sender: Arc<dyn SendTwoFactorOtp>) -> Self {
        self.config.send_otp = Some(sender);
        self
    }

    /// Generate a current TOTP from an application-owned UTF-8 secret.
    ///
    /// This corresponds to `auth.api.generateTOTP`; it has no public HTTP route.
    ///
    /// # Errors
    ///
    /// Returns an error if the secret cannot be decoded or the TOTP configuration is invalid.
    pub fn generate_totp(&self, secret: &str) -> AuthResult<String> {
        require_totp_enabled(&self.config)?;
        generate_totp_at(&self.config, secret, totp_counter(&self.config), 0.0)
    }

    /// Read the currently stored backup JSON for a user.
    ///
    /// Installed storage may contain any truthy JSON value. Dates are revived
    /// with the pinned grammar and numbers use JavaScript JSON serialization;
    /// generated code arrays retain their ordinary array-of-strings shape.
    ///
    /// This is the Rust server-side equivalent of the TypeScript
    /// `auth.api.viewBackupCodes` capability. It is intentionally not exposed
    /// as a public HTTP route.
    ///
    /// # Errors
    ///
    /// Returns an error if backup codes are unavailable, cannot be decrypted, or cannot be loaded.
    pub async fn view_backup_codes<S: alibi_core::AuthSchema>(
        &self,
        user_id: &str,
        ctx: &AuthContext<S>,
    ) -> AuthResult<serde_json::Value> {
        view_backup_codes_core(user_id, &self.config, ctx).await
    }
}

alibi_core::impl_auth_plugin! {
    TwoFactorPlugin, "two-factor";
    routes {
        post "/two-factor/enable" => handle_enable, "enable_two_factor";
        post "/two-factor/disable" => handle_disable, "disable_two_factor";
        post "/two-factor/get-totp-uri" => handle_get_totp_uri, "get_totp_uri";
        post "/two-factor/verify-totp" => handle_verify_totp, "verify_totp";
        post "/two-factor/send-otp" => handle_send_otp, "send_otp";
        post "/two-factor/verify-otp" => handle_verify_otp, "verify_otp";
        post "/two-factor/generate-backup-codes" => handle_generate_backup_codes, "generate_backup_codes";
        post "/two-factor/verify-backup-code" => handle_verify_backup_code, "verify_backup_code";
    }
    extra {
    route_openapi_metadata!(S);

        fn rate_limits(&self) -> Vec<alibi_core::PluginRateLimit> {
            vec![alibi_core::PluginRateLimit { matches: |path| path.starts_with("/two-factor/"), limit: alibi_core::EndpointRateLimit { window_seconds: 10.0, max_requests: 3.0 } }]
        }
        fn server_endpoints(&self) -> Vec<alibi_core::endpoint::EndpointDefinition> { endpoint::definitions() }

        fn validate_endpoint(&self, call: &alibi_core::endpoint::EndpointCall, _ctx: &AuthContext<S>) -> AuthResult<alibi_core::endpoint::EndpointInput> { endpoint::validate(call, &self.config) }

        async fn on_endpoint(&self, call: &alibi_core::endpoint::EndpointCall, ctx: &AuthContext<S>) -> AuthResult<alibi_core::endpoint::EndpointResponse> { self.call_endpoint(call, ctx).await }

        async fn on_init(
            &self,
            ctx: &mut alibi_core::AuthInitContext<S>,
        ) -> AuthResult<()> {
            let default = |mut input: alibi_core::CreateUser| {
                _ = input.two_factor_enabled.get_or_insert(false);
                Ok(input)
            };
            if ctx.config.user_validation.is_some() {ctx.register_user_creation_adapter_default(default);}
            else {ctx.register_user_create_transform(default);}
            ctx.set_metadata(METADATA_ENABLED, serde_json::Value::Bool(true));
            ctx.set_metadata(METADATA_TOTP_DISABLED, serde_json::Value::Bool(self.config.totp_disabled));
            ctx.set_metadata(
                METADATA_OTP_ENABLED,
                serde_json::Value::Bool(self.config.send_otp.is_some()),
            );
            ctx.set_metadata(
                METADATA_TWO_FACTOR_COOKIE_MAX_AGE,
                serde_json::Number::from_f64(self.config.two_factor_cookie_max_age)
                    .map_or(serde_json::Value::Null, serde_json::Value::Number),
            );
            ctx.set_metadata(
                METADATA_TRUST_DEVICE_MAX_AGE,
                serde_json::Number::from_f64(self.config.trust_device_max_age)
                    .map_or(serde_json::Value::Null, serde_json::Value::Number),
            );
            ctx.extensions.insert(TwoFactorCookiePolicy {
                challenge_max_age: self.config.two_factor_cookie_max_age,
                trust_max_age: self.config.trust_device_max_age,
            });
            Ok(())
        }
    }
}

impl TwoFactorPlugin {
    async fn handle_enable(
        &self,
        req: &AuthRequest,
        ctx: &AuthContext<impl alibi_core::AuthSchema>,
    ) -> AuthResult<AuthResponse> {
        let body: EnableTwoFactorInput =
            match parse_password_body(req, self.config.allow_passwordless, true) {
                Ok(body) => body,
                Err(response) => return Ok(response),
            };
        let (user, session) = super::helpers::ordinary_session(req, ctx).await?;

        let (response, set_cookie_headers) =
            match enable_core(&body, &user, &session, &self.config, ctx).await {
                Ok(result) => result,
                Err(
                    BackupOperationError::Auth(AuthError::SessionCreationCancelled)
                    | BackupOperationError::InvalidGeneration,
                ) => {
                    return Ok(AuthResponse::new(500));
                }
                Err(BackupOperationError::Auth(error)) => return Err(error),
            };
        let mut auth_response = AuthResponse::json(200, &response)?;
        for cookie in set_cookie_headers {
            auth_response = auth_response.with_appended_header("Set-Cookie", cookie);
        }
        Ok(auth_response)
    }

    async fn handle_disable(
        &self,
        req: &AuthRequest,
        ctx: &AuthContext<impl alibi_core::AuthSchema>,
    ) -> AuthResult<AuthResponse> {
        let body: DisableTwoFactorInput =
            match parse_password_body(req, self.config.allow_passwordless, false) {
                Ok(v) => v,
                Err(resp) => return Ok(resp),
            };
        self.disable_factor(&body, req, ctx).await
    }

    async fn disable_factor<S: alibi_core::AuthSchema>(
        &self,
        body: &DisableTwoFactorInput,
        req: &impl alibi_core::session::SessionRequest,
        ctx: &AuthContext<S>,
    ) -> AuthResult<AuthResponse> {
        let (user, session) = ctx
            .require_authoritative_session(req)
            .await
            .map_err(|error| {
                if matches!(
                    error,
                    AuthError::Unauthenticated | AuthError::SessionNotFound
                ) {
                    crate::helpers::unauthorized()
                } else {
                    error
                }
            })?;

        let (response, set_cookie_headers) =
            disable_core(body, &user, &session, req, &self.config, ctx).await?;
        let mut auth_response = AuthResponse::json(200, &response)?;
        for cookie in set_cookie_headers {
            auth_response = auth_response.with_appended_header("Set-Cookie", cookie);
        }
        Ok(auth_response)
    }

    async fn handle_get_totp_uri(
        &self,
        req: &AuthRequest,
        ctx: &AuthContext<impl alibi_core::AuthSchema>,
    ) -> AuthResult<AuthResponse> {
        let (user, _session) = super::helpers::ordinary_session(req, ctx).await?;
        let allow_passwordless = self
            .config
            .totp_allow_passwordless
            .unwrap_or(self.config.allow_passwordless);
        let body: GetTotpUriInput = match parse_password_body(req, allow_passwordless, false) {
            Ok(v) => v,
            Err(resp) => return Ok(resp),
        };

        let response = get_totp_uri_core(&body, &user, &self.config, ctx).await?;
        AuthResponse::json(200, &response).map_err(AuthError::from)
    }

    async fn handle_verify_totp(
        &self,
        req: &AuthRequest,
        ctx: &AuthContext<impl alibi_core::AuthSchema>,
    ) -> AuthResult<AuthResponse> {
        let body: VerifyTotpInput = match alibi_core::validate_request_body(req) {
            Ok(v) => v,
            Err(resp) => return Ok(resp),
        };

        self.verify_totp(&body, req, ctx).await
    }

    async fn verify_totp<S: alibi_core::AuthSchema>(
        &self,
        body: &VerifyTotpInput,
        req: &impl alibi_core::session::SessionRequest,
        ctx: &AuthContext<S>,
    ) -> AuthResult<AuthResponse> {
        let (response, set_cookie_headers) =
            match Box::pin(verify_totp_core(req, body, &self.config, ctx)).await {
                Ok(result) => result,
                Err(
                    TotpVerificationError::InvalidGeneration
                    | TotpVerificationError::SessionCreationCancelled,
                ) => return Ok(AuthResponse::new(500)),
                Err(TotpVerificationError::Auth(error)) => {
                    return verification_error_response(error, ctx);
                }
            };
        let mut auth_response = AuthResponse::json(200, &response)?;
        for cookie in set_cookie_headers {
            auth_response = auth_response.with_appended_header("Set-Cookie", cookie);
        }
        Ok(auth_response)
    }

    async fn handle_send_otp(
        &self,
        req: &AuthRequest,
        ctx: &AuthContext<impl alibi_core::AuthSchema>,
    ) -> AuthResult<AuthResponse> {
        if let Err(response) = crate::authentication_helpers::parse_body::<SendOtpRequest>(req) {
            return Ok(response);
        }
        self.send_otp(req, ctx).await
    }

    async fn send_otp<S: alibi_core::AuthSchema>(
        &self,
        req: &impl alibi_core::session::SessionRequest,
        ctx: &AuthContext<S>,
    ) -> AuthResult<AuthResponse> {
        let response = match Box::pin(send_otp_core(req, &self.config, ctx)).await {
            Ok(response) => response,
            Err(SendOtpError::InvalidGeneration) => return Ok(AuthResponse::new(500)),
            Err(SendOtpError::Auth(error)) => return Err(error),
        };
        AuthResponse::json(200, &response).map_err(AuthError::from)
    }

    async fn handle_verify_otp(
        &self,
        req: &AuthRequest,
        ctx: &AuthContext<impl alibi_core::AuthSchema>,
    ) -> AuthResult<AuthResponse> {
        let body: VerifyOtpRequest = match alibi_core::validate_request_body(req) {
            Ok(v) => v,
            Err(resp) => return Ok(resp),
        };

        self.verify_otp(&body, req, ctx).await
    }

    async fn verify_otp<S: alibi_core::AuthSchema>(
        &self,
        body: &VerifyOtpInput,
        req: &impl alibi_core::session::SessionRequest,
        ctx: &AuthContext<S>,
    ) -> AuthResult<AuthResponse> {
        let (response, set_cookie_headers) =
            match Box::pin(verify_otp_core(req, body, &self.config, ctx)).await {
                Ok(result) => result,
                Err(ExistingSessionFactorError::SessionCreationCancelled) => {
                    return Ok(AuthResponse::new(500));
                }
                // OTP's own code budget does not expire the pending-factor cookie.
                Err(ExistingSessionFactorError::Auth(
                    error @ AuthError::Upstream {
                        code: "TOO_MANY_ATTEMPTS_REQUEST_NEW_CODE",
                        ..
                    },
                )) => return Err(error),
                Err(ExistingSessionFactorError::Auth(error)) => {
                    return verification_error_response(error, ctx);
                }
            };
        let mut auth_response = AuthResponse::json(200, &response)?;
        for cookie in set_cookie_headers {
            auth_response = auth_response.with_appended_header("Set-Cookie", cookie);
        }
        Ok(auth_response)
    }

    async fn handle_generate_backup_codes(
        &self,
        req: &AuthRequest,
        ctx: &AuthContext<impl alibi_core::AuthSchema>,
    ) -> AuthResult<AuthResponse> {
        let (user, _session) = super::helpers::ordinary_session(req, ctx).await?;
        let allow_passwordless = self
            .config
            .backup_allow_passwordless
            .unwrap_or(self.config.allow_passwordless);
        let body: GenerateBackupCodesRequest =
            match parse_password_body(req, allow_passwordless, false) {
                Ok(v) => v,
                Err(resp) => return Ok(resp),
            };

        let response =
            match Box::pin(generate_backup_codes_core(&body, &user, &self.config, ctx)).await {
                Ok(response) => response,
                Err(BackupOperationError::InvalidGeneration) => return Ok(AuthResponse::new(500)),
                Err(BackupOperationError::Auth(error)) => return Err(error),
            };
        AuthResponse::json(200, &response).map_err(AuthError::from)
    }

    async fn handle_verify_backup_code(
        &self,
        req: &AuthRequest,
        ctx: &AuthContext<impl alibi_core::AuthSchema>,
    ) -> AuthResult<AuthResponse> {
        let body: VerifyBackupCodeRequest = match alibi_core::validate_request_body(req) {
            Ok(v) => v,
            Err(resp) => return Ok(resp),
        };

        self.verify_backup_code(&body, req, ctx).await
    }

    async fn verify_backup_code<S: alibi_core::AuthSchema>(
        &self,
        body: &VerifyBackupCodeInput,
        req: &impl alibi_core::session::SessionRequest,
        ctx: &AuthContext<S>,
    ) -> AuthResult<AuthResponse> {
        let (response, set_cookie_headers) =
            match Box::pin(verify_backup_code_core(req, body, &self.config, ctx)).await {
                Ok(result) => result,
                Err(error) => return verification_error_response(error, ctx),
            };
        let mut auth_response = AuthResponse::json(200, &response)?;
        for cookie in set_cookie_headers {
            auth_response = auth_response.with_appended_header("Set-Cookie", cookie);
        }
        Ok(auth_response)
    }
}

enum TotpVerificationError {
    Auth(AuthError),
    InvalidGeneration,
    SessionCreationCancelled,
}
impl From<AuthError> for TotpVerificationError {
    fn from(error: AuthError) -> Self {
        Self::Auth(error)
    }
}

enum SendOtpError {
    Auth(AuthError),
    // The pinned random-string generator throws before storage or delivery.
    InvalidGeneration,
}

impl From<AuthError> for SendOtpError {
    fn from(error: AuthError) -> Self {
        Self::Auth(error)
    }
}

struct FactorAttempt {
    identifier: String,
    count: f64,
    expires_at: chrono::DateTime<Utc>,
}

// Only an explicitly cancelled session creation is distinguished. The same
// AuthError from user updates or other operations retains its default response.
enum ExistingSessionFactorError {
    Auth(AuthError),
    SessionCreationCancelled,
}

impl From<AuthError> for ExistingSessionFactorError {
    fn from(error: AuthError) -> Self {
        Self::Auth(error)
    }
}

impl ExistingSessionFactorError {
    fn into_auth_error(self) -> AuthError {
        match self {
            Self::Auth(error) => error,
            Self::SessionCreationCancelled => AuthError::SessionCreationCancelled,
        }
    }
}

enum BackupOperationError {
    Auth(AuthError),
    InvalidGeneration,
}

impl From<AuthError> for BackupOperationError {
    fn from(error: AuthError) -> Self {
        Self::Auth(error)
    }
}

impl<S: alibi_core::AuthSchema> ResolvedTwoFactorState<S> {
    fn user(&self) -> alibi_core::AuthenticatedUser<S> {
        match self {
            Self::Session { user, .. } => user.clone(),
            Self::Pending(pending) => alibi_core::AuthenticatedUser::Stored(pending.user.clone()),
        }
    }

    fn key(&self) -> &str {
        match self {
            Self::Session { key, .. } => key,
            Self::Pending(pending) => &pending.key,
        }
    }
}

impl std::fmt::Debug for TwoFactorPlugin {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("TwoFactorPlugin").finish_non_exhaustive()
    }
}

pub(crate) fn is_enabled(ctx: &AuthContext<impl alibi_core::AuthSchema>) -> bool {
    ctx.get_metadata(METADATA_ENABLED)
        .and_then(serde_json::Value::as_bool)
        .unwrap_or(false)
}

const fn require_totp_enabled(config: &TwoFactorConfig) -> AuthResult<()> {
    if config.totp_disabled {
        return Err(AuthError::Upstream {
            status: 400,
            code: "TOTP_NOT_CONFIGURED",
            message: "totp isn't configured",
        });
    }
    Ok(())
}

// LCOV_EXCL_START
#[cfg(test)]
mod tests {
    use super::cookies::sign_cookie_value;
    use super::*;
    use crate::test_helpers;
    use alibi_core::wire::{SessionView, UserView};
    use alibi_core::{AuthPlugin, AuthVerification};
    use alibi_core::{CreateAccount, CreateUser, HttpMethod};
    use chrono::Duration;
    use cookie::Cookie;

    type TestSchema = alibi_seaorm::store::__private_test_support::bundled_schema::BundledSchema;

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn skip_enrollment_hooks_retain_factor_generation_and_current_token_on_rejection() {
        use alibi_core::{AuthConfig, CreateSession};
        use alibi_seaorm::{
            Database, DatabaseHooks, HookControl, SeaOrmBackend, SeaOrmHookContext, SeaOrmStore,
        };
        struct PolicyHook {
            cancel_session: bool,
            session_forbidden: bool,
            observed: Arc<std::sync::Mutex<Vec<(String, String)>>>,
        }
        #[async_trait]
        impl DatabaseHooks<TestSchema, SeaOrmBackend> for PolicyHook {
            async fn before_update_user(
                &self,
                id: &str,
                update: &mut UpdateUser,
                context: &SeaOrmHookContext<'_>,
            ) -> AuthResult<HookControl> {
                if context
                    .request
                    .as_ref()
                    .is_some_and(|request| request.path.ends_with("/two-factor/enable"))
                {
                    self.observed
                        .lock()
                        .unwrap()
                        .push(("user".into(), id.into()));
                    assert_eq!(update.two_factor_enabled, Some(true));
                    if !self.cancel_session {
                        return Err(AuthError::Upstream {
                            status: 400,
                            code: "USER_UPDATE_DENIED",
                            message: "Configured user update denied",
                        });
                    }
                }
                Ok(HookControl::Continue)
            }
            async fn before_create_session(
                &self,
                session: &mut CreateSession,
                context: &SeaOrmHookContext<'_>,
            ) -> AuthResult<HookControl> {
                if context
                    .request
                    .as_ref()
                    .is_some_and(|request| request.path.ends_with("/two-factor/enable"))
                {
                    self.observed
                        .lock()
                        .unwrap()
                        .push(("session".into(), session.user_id.clone()));
                    assert_eq!(
                        session.active_organization_id.as_deref(),
                        Some("retained-org")
                    );
                    assert_eq!(session.active_team_id.as_deref(), Some("retained-team"));
                    assert_eq!(session.impersonated_by.as_deref(), Some("retained-admin"));
                    assert_eq!(session.ip_address.as_deref(), Some("127.0.0.9"));
                    assert_eq!(session.user_agent.as_deref(), Some("retained-agent"));
                    return if self.session_forbidden {
                        Err(AuthError::forbidden(
                            "session creation cancelled by database hook",
                        ))
                    } else {
                        Ok(HookControl::Cancel)
                    };
                }
                Ok(HookControl::Continue)
            }
        }
        let password_hash = alibi_core::hash_password(None, "password123")
            .await
            .unwrap();
        for (cancel_session, session_forbidden) in [(false, false), (true, false), (true, true)] {
            for existing in [false, true] {
                let db = Database::connect("sqlite::memory:").await.unwrap();
                alibi_seaorm::store::__private_test_support::migrator::run_migrations(&db)
                    .await
                    .unwrap();
                let config = Arc::new(AuthConfig::new("skip-hook-secret-at-least-32-characters"));
                let mut ctx = AuthContext::new(
                    Arc::clone(&config),
                    Arc::new(SeaOrmStore::<TestSchema>::new(
                        Arc::clone(&config),
                        db.clone(),
                    )),
                );
                let user = ctx
                    .database
                    .create_user(
                        CreateUser::new()
                            .with_name("Hook Owner")
                            .with_email("hook-owner@fixture.test"),
                    )
                    .await
                    .unwrap();
                _ = ctx
                    .database
                    .create_account(CreateAccount {
                        additional_fields: alibi_core::field_policy::FieldValues::default(),
                        user_id: user.id.clone(),
                        account_id: user.id.clone(),
                        provider_id: "credential".into(),
                        password: Some(password_hash.clone()),
                        access_token: None,
                        refresh_token: None,
                        id_token: None,
                        access_token_expires_at: None,
                        refresh_token_expires_at: None,
                        scope: None,
                    })
                    .await
                    .unwrap();
                let session = ctx
                    .database
                    .create_session(CreateSession {
                        token: None,
                        user_id: user.id.clone(),
                        expires_at: Utc::now() + ctx.config.session.expires_in,
                        active_organization_id: Some("retained-org".into()),
                        active_team_id: Some("retained-team".into()),
                        impersonated_by: Some("retained-admin".into()),
                        ip_address: Some("127.0.0.9".into()),
                        user_agent: Some("retained-agent".into()),
                        additional_fields: alibi_core::field_policy::FieldValues::default(),
                    })
                    .await
                    .unwrap();
                let factor = if existing {
                    Some(
                        ctx.database
                            .create_two_factor(CreateTwoFactor {
                                user_id: user.id.clone(),
                                secret: encrypt_value(&ctx.config, "historical-secret").unwrap(),
                                backup_codes: encrypt_value(&ctx.config, "[\"historical-backup\"]")
                                    .unwrap(),
                                verified: Some(false),
                                failed_verification_count: Some(0.5),
                                locked_until: Some(Utc::now() + Duration::minutes(1)),
                            })
                            .await
                            .unwrap(),
                    )
                } else {
                    None
                };
                let observed = Arc::new(std::sync::Mutex::new(Vec::new()));
                ctx.database = Arc::new(
                    SeaOrmStore::<TestSchema>::new(config, db.clone()).with_hooks(vec![Arc::new(
                        PolicyHook {
                            cancel_session,
                            session_forbidden,
                            observed: Arc::clone(&observed),
                        },
                    )]),
                );
                let plugin = TwoFactorPlugin::with_config(TwoFactorConfig {
                    skip_verification_on_enable: true,
                    ..Default::default()
                });
                let mut init = alibi_core::AuthInitContext::new(
                    Arc::clone(&ctx.config),
                    Arc::clone(&ctx.database),
                );
                plugin.on_init(&mut init).await.unwrap();
                crate::OrganizationPlugin::with_config(crate::organization::OrganizationConfig {
                    teams: crate::organization::TeamsConfig {
                        enabled: true,
                        ..Default::default()
                    },
                    ..Default::default()
                })
                .on_init(&mut init)
                .await
                .unwrap();
                crate::AdminPlugin::new().on_init(&mut init).await.unwrap();
                ctx.database = init.database_with_registered_transforms();
                let parts = init.into_parts();
                ctx.metadata = parts.metadata;
                ctx.extensions = parts.extensions;
                let cookie = create_session_cookie(&session.token, &ctx.config).unwrap();
                let mut request = AuthRequest::new(HttpMethod::Post, "/two-factor/enable");
                _ = request
                    .headers
                    .insert("cookie".into(), cookie.split(';').next().unwrap().into());
                request.body = Some(br#"{"password":"password123"}"#.to_vec());
                let result = alibi_core::with_request_hook_context(
                    &request,
                    plugin.on_request(&request, &ctx),
                )
                .await;
                if cancel_session && !session_forbidden {
                    let response = result.unwrap().unwrap();
                    assert_eq!(response.status, 500);
                    assert_eq!(response.body.len(), 0);
                } else {
                    let error = result.unwrap_err();
                    assert_eq!(error.status_code(), if cancel_session { 403 } else { 400 });
                    if session_forbidden {
                        assert!(matches!(error, AuthError::Forbidden(_)));
                    }
                }
                let expected = if cancel_session {
                    vec![
                        ("user".into(), user.id.clone()),
                        ("session".into(), user.id.clone()),
                    ]
                } else {
                    vec![("user".into(), user.id.clone())]
                };
                assert_eq!(*observed.lock().unwrap(), expected);
                assert_eq!(
                    serde_json::to_value(
                        ctx.database
                            .get_two_factor_by_user_id(&user.id)
                            .await
                            .unwrap()
                    )
                    .unwrap(),
                    serde_json::to_value(factor).unwrap()
                );
                let stored_user = ctx
                    .database
                    .get_user_by_id(&user.id)
                    .await
                    .unwrap()
                    .unwrap();
                assert_eq!(stored_user.two_factor_enabled(), cancel_session);
                let sessions = ctx.database.get_user_sessions(&user.id).await.unwrap();
                assert_eq!(sessions.len(), 1);
                assert_eq!(
                    serde_json::to_value(
                        (sessions)
                            .first()
                            .expect("fixture contains the requested index")
                    )
                    .unwrap(),
                    serde_json::to_value(&session).unwrap()
                );
                let (current_user, current_session) = ctx.require_session(&request).await.unwrap();
                assert_eq!(current_user.id, user.id);
                assert_eq!(current_session.token, session.token);
                db.close().await.unwrap();
            }
        }
    }

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn two_factor_password_checks_use_configured_native_hasher_and_utf16_maximum() {
        use alibi_core::{PasswordHasher, ScryptHasher, UpdateAccount};
        struct PrefixedHasher {
            received: std::sync::Mutex<Vec<(String, String)>>,
        }
        #[async_trait]
        impl PasswordHasher for PrefixedHasher {
            async fn hash(&self, password: &str) -> AuthResult<String> {
                ScryptHasher
                    .hash(&format!("provider-prefix:{password}"))
                    .await
            }
            async fn verify(&self, hash: &str, password: &str) -> AuthResult<bool> {
                self.received
                    .lock()
                    .unwrap()
                    .push((hash.into(), password.into()));
                ScryptHasher
                    .verify(hash, &format!("provider-prefix:{password}"))
                    .await
            }
        }
        let (mut ctx, user, session) =
            create_test_context_with_credential_user("native-provider@fixture.test", false).await;
        let provider = Arc::new(PrefixedHasher {
            received: std::sync::Mutex::new(Vec::new()),
        });
        let hash = provider.hash("password123").await.unwrap();
        let account = ctx
            .database
            .get_user_accounts(&user.id)
            .await
            .unwrap()
            .into_iter()
            .find(|account| account.provider_id == "credential")
            .unwrap();
        _ = ctx
            .database
            .update_account(
                &account.id,
                UpdateAccount {
                    password: Some(hash.clone()),
                    ..Default::default()
                },
            )
            .await
            .unwrap();
        let plugin = TwoFactorPlugin::new();
        let mut init =
            alibi_core::AuthInitContext::new(Arc::clone(&ctx.config), Arc::clone(&ctx.database));
        crate::EmailPasswordPlugin::new()
            .password_max_length(13)
            .password_hasher(Arc::<PrefixedHasher>::clone(&provider))
            .on_init(&mut init)
            .await
            .unwrap();
        plugin.on_init(&mut init).await.unwrap();
        let parts = init.into_parts();
        ctx.extensions = parts.extensions;
        ctx.metadata = parts.metadata;
        let cookie = create_session_cookie(&session.token, &ctx.config).unwrap();
        let mut request = AuthRequest::new(HttpMethod::Post, "/two-factor/enable");
        _ = request
            .headers
            .insert("cookie".into(), cookie.split(';').next().unwrap().into());
        request.body = Some(br#"{"password":"password123"}"#.to_vec());
        let response = plugin.on_request(&request, &ctx).await.unwrap().unwrap();
        assert_eq!(response.status, 200);
        let factor = ctx
            .database
            .get_two_factor_by_user_id(&user.id)
            .await
            .unwrap()
            .unwrap();
        request.path = "/two-factor/get-totp-uri".into();
        request.body = Some(br#"{"password":"wrong-password"}"#.to_vec());
        let wrong = plugin.on_request(&request, &ctx).await.unwrap_err();
        assert_eq!(
            wrong.error_payload().1.as_deref(),
            Some("PASSWORD_TOO_LONG")
        );
        // A different valid-length password must reach the configured real verifier.
        request.body = Some(br#"{"password":"wrong-pass"}"#.to_vec());
        let wrong_2 = plugin.on_request(&request, &ctx).await.unwrap_err();
        assert_eq!(
            wrong_2.error_payload().1.as_deref(),
            Some("INVALID_PASSWORD")
        );
        request.body =
            Some(serde_json::to_vec(&serde_json::json!({"password":"🍵".repeat(7)})).unwrap());
        let long = plugin.on_request(&request, &ctx).await.unwrap_err();
        assert_eq!(long.error_payload().1.as_deref(), Some("PASSWORD_TOO_LONG"));
        assert_eq!(
            *provider.received.lock().unwrap(),
            vec![
                (hash.clone(), "password123".into()),
                (hash, "wrong-pass".into())
            ]
        );
        assert_eq!(
            serde_json::to_value(
                ctx.database
                    .get_two_factor_by_user_id(&user.id)
                    .await
                    .unwrap()
                    .unwrap()
            )
            .unwrap(),
            serde_json::to_value(factor).unwrap()
        );
        assert!(
            ctx.database
                .get_session(&session.token)
                .await
                .unwrap()
                .is_some()
        );
    }

    #[tokio::test]
    async fn signed_empty_factor_challenge_cannot_read_a_seeded_empty_identifier() {
        let (ctx, user, _) =
            create_test_context_with_credential_user("empty-challenge@fixture.test", true).await;
        let seeded = ctx
            .database
            .create_verification(CreateVerification {
                identifier: String::new(),
                value: user.id.clone(),
                expires_at: Utc::now() + Duration::minutes(5),
            })
            .await
            .unwrap();
        let signed = alibi_core::utils::cookie_utils::sign_cookie_value("", &ctx.config.secret);
        assert_eq!(
            alibi_core::utils::cookie_utils::verify_cookie_value(&signed, &ctx.config.secret),
            Some(String::new())
        );
        let mut req = AuthRequest::new(HttpMethod::Post, "/two-factor/verify-otp");
        _ = req.headers.insert(
            "cookie".into(),
            format!(
                "{}={signed}",
                related_cookie_name(&ctx.config, TWO_FACTOR_COOKIE_SUFFIX)
            ),
        );
        let error = resolve_two_factor_state(&req, &ctx)
            .await
            .err()
            .expect("An empty signed challenge must be rejected");
        assert_eq!(error.status_code(), 401);
        let untouched = ctx
            .database
            .get_verification_by_identifier("")
            .await
            .unwrap()
            .unwrap();
        assert_eq!(untouched.id(), seeded.id());
        assert_eq!(untouched.value(), seeded.value());
        assert_eq!(untouched.expires_at(), seeded.expires_at());
        assert_eq!(
            ctx.database
                .get_user_sessions(&user.id)
                .await
                .unwrap()
                .len(),
            1
        );
    }

    #[tokio::test]
    async fn pending_factor_preferences_authenticate_the_first_cookie_and_preserve_issued_expiry() {
        let (ctx, user, _) =
            create_test_context_with_credential_user("preferences@fixture.test", true).await;
        let preference_name = related_cookie_name(&ctx.config, DONT_REMEMBER_COOKIE_SUFFIX);
        let empty = alibi_core::utils::cookie_utils::sign_cookie_value("", &ctx.config.secret);
        let signed = alibi_core::utils::cookie_utils::sign_cookie_value("true", &ctx.config.secret);
        let foreign = alibi_core::utils::cookie_utils::sign_cookie_value("true", "foreign-secret");
        for (preference, temporary) in [
            (empty.clone(), false),
            (signed.clone(), true),
            (foreign, false),
            ("true".to_owned(), false),
            (format!("{empty}; {preference_name}={signed}"), false),
            (format!("{signed}; {preference_name}={empty}"), true),
        ] {
            let challenge = begin_sign_in_challenge(&user, None, &ctx).await.unwrap();
            let challenge_cookie = challenge
                .set_cookie_headers
                .iter()
                .find(|header| {
                    header.starts_with(&format!(
                        "{}=",
                        related_cookie_name(&ctx.config, TWO_FACTOR_COOKIE_SUFFIX)
                    ))
                })
                .unwrap()
                .split(';')
                .next()
                .unwrap();
            let mut req = AuthRequest::new(HttpMethod::Post, "/two-factor/verify-otp");
            _ = req.headers.insert(
                "cookie".into(),
                format!("{challenge_cookie}; {preference_name}={preference}"),
            );
            let ResolvedTwoFactorState::Pending(pending) =
                resolve_two_factor_state(&req, &ctx).await.unwrap()
            else {
                panic!("A signed pending challenge must resolve without a session cookie");
            };
            assert_eq!(pending.dont_remember, temporary);
            let (completed, headers) =
                finalize_pending_two_factor(pending, &req, false, true, &ctx)
                    .await
                    .unwrap();
            let before = ctx
                .database
                .get_session(&completed.token)
                .await
                .unwrap()
                .unwrap();
            let lifetime = before.expires_at() - before.created_at();
            assert!(
                (lifetime - Duration::days(if temporary { 1 } else { 7 }))
                    .num_milliseconds()
                    .abs()
                    < 1000
            );
            let mut read = AuthRequest::new(HttpMethod::Get, "/get-session");
            let cookies = headers
                .iter()
                .filter(|header| !header.contains("Max-Age=0"))
                .map(|header| header.split(';').next().unwrap())
                .collect::<Vec<_>>()
                .join("; ");
            _ = read.headers.insert("cookie".into(), cookies);
            let (authenticated_user, authenticated_session) =
                ctx.require_session(&read).await.unwrap();
            assert_eq!(authenticated_user.id(), user.id);
            assert_eq!(authenticated_session.token, completed.token);
            assert_eq!(authenticated_session.expires_at, before.expires_at());
            let after = ctx
                .database
                .get_session(&completed.token)
                .await
                .unwrap()
                .unwrap();
            assert_eq!(after.expires_at(), before.expires_at());
            assert_eq!(after.updated_at(), before.updated_at());
        }
    }

    async fn create_test_context_with_credential_user(
        email: &str,
        two_factor_enabled: bool,
    ) -> (AuthContext<TestSchema>, UserView, SessionView) {
        let mut ctx = test_helpers::create_test_context().await;
        let mut init =
            alibi_core::AuthInitContext::new(Arc::clone(&ctx.config), Arc::clone(&ctx.database));
        TwoFactorPlugin::new().on_init(&mut init).await.unwrap();
        ctx.database = init.database_with_registered_transforms();
        let parts = init.into_parts();
        ctx.metadata = parts.metadata;
        ctx.extensions = parts.extensions;
        let user = test_helpers::create_user(
            &ctx,
            CreateUser::new()
                .with_email(email)
                .with_name("Two Factor Tester"),
        )
        .await;

        let password_hash = alibi_core::hash_password(None, "password123")
            .await
            .unwrap();
        _ = ctx
            .database
            .create_account(CreateAccount {
                additional_fields: alibi_core::field_policy::FieldValues::default(),
                user_id: user.id.clone(),
                account_id: user.id.clone(),
                provider_id: "credential".to_owned(),
                access_token: None,
                refresh_token: None,
                id_token: None,
                access_token_expires_at: None,
                refresh_token_expires_at: None,
                scope: None,
                password: Some(password_hash),
            })
            .await
            .unwrap();

        let user = if two_factor_enabled {
            UserView::from(
                &ctx.database
                    .update_user(
                        &user.id,
                        UpdateUser {
                            two_factor_enabled: Some(true),
                            ..Default::default()
                        },
                    )
                    .await
                    .unwrap(),
            )
        } else {
            user
        };

        let session = test_helpers::create_session(&ctx, user.id.clone(), Duration::hours(1)).await;
        (ctx, user, session)
    }

    fn cookie_value(header: &str) -> String {
        Cookie::parse(header)
            .expect("Set-Cookie header should parse")
            .value()
            .to_owned()
    }

    #[test]
    fn test_signed_cookie_round_trip_and_tamper_rejection() {
        let signed = sign_cookie_value("secret-value", "payload-value");
        let verified = verify_factor_cookie_value("secret-value", &signed);
        assert_eq!(verified.as_deref(), Some("payload-value"));

        let tampered = signed.replacen("payload-value", "other-value", 1);
        let tampered_verified = verify_factor_cookie_value("secret-value", &tampered);
        assert!(tampered_verified.is_none());
    }

    #[tokio::test]
    async fn test_begin_sign_in_challenge_sets_pending_cookie_and_remember_choice() {
        let (ctx, user, _session) =
            create_test_context_with_credential_user("challenge@example.com", true).await;

        let challenge = begin_sign_in_challenge(&user, Some(false), &ctx)
            .await
            .unwrap();
        assert!(challenge.response.two_factor_redirect);

        let two_factor_cookie = challenge
            .set_cookie_headers
            .iter()
            .find(|header| header.starts_with("better-auth.two_factor="))
            .cloned()
            .expect("challenge should set the two-factor cookie");
        let dont_remember_cookie = challenge
            .set_cookie_headers
            .iter()
            .find(|header| header.starts_with("better-auth.dont_remember="))
            .cloned()
            .expect("challenge should set the remember-choice cookie");

        let two_factor_req = test_helpers::create_auth_request_no_query(
            HttpMethod::Post,
            "/two-factor/verify-otp",
            None,
            None,
        );
        let mut req = two_factor_req;
        req.headers.insert(
            "cookie".to_owned(),
            format!(
                "better-auth.two_factor={}; better-auth.dont_remember={}",
                cookie_value(&two_factor_cookie),
                cookie_value(&dont_remember_cookie)
            ),
        );

        let identifier = read_signed_cookie(&req, TWO_FACTOR_COOKIE_SUFFIX, &ctx)
            .expect("signed cookie should verify");
        let verification = ctx
            .database
            .get_verification_by_identifier(&identifier)
            .await
            .unwrap()
            .expect("challenge should persist a pending verification");
        assert_eq!(verification.value(), user.id);
    }

    #[tokio::test]
    async fn test_inspect_trusted_device_rotates_server_state() {
        let (ctx, user, _session) =
            create_test_context_with_credential_user("trusted@example.com", true).await;

        let trust_cookie = create_trust_device_cookie_header(&user, &ctx)
            .await
            .unwrap();
        let mut req = test_helpers::create_auth_request_no_query(
            HttpMethod::Post,
            "/sign-in/email",
            None,
            None,
        );
        req.headers.insert(
            "cookie".to_owned(),
            format!("better-auth.trust_device={}", cookie_value(&trust_cookie)),
        );

        let original_cookie = read_signed_cookie(&req, TRUST_DEVICE_COOKIE_SUFFIX, &ctx)
            .expect("trust cookie should verify");
        let original_identifier = original_cookie
            .split_once('!')
            .expect("trust cookie should include the identifier")
            .1
            .to_owned();

        let result = inspect_trusted_device(&req, &user, &ctx).await.unwrap();
        assert!(result.trusted);
        assert_eq!(result.set_cookie_headers.len(), 1);

        let rotated_cookie = (*(result.set_cookie_headers)
            .first()
            .expect("fixture contains the requested index"))
        .clone();
        let mut rotated_req = test_helpers::create_auth_request_no_query(
            HttpMethod::Post,
            "/sign-in/email",
            None,
            None,
        );
        rotated_req.headers.insert(
            "cookie".to_owned(),
            format!("better-auth.trust_device={}", cookie_value(&rotated_cookie)),
        );
        let rotated_value = read_signed_cookie(&rotated_req, TRUST_DEVICE_COOKIE_SUFFIX, &ctx)
            .expect("rotated trust cookie should verify");
        let rotated_identifier = rotated_value
            .split_once('!')
            .expect("rotated cookie should include the identifier")
            .1
            .to_owned();

        assert_ne!(original_identifier, rotated_identifier);
        assert!(
            ctx.database
                .get_verification_by_identifier(&original_identifier)
                .await
                .unwrap()
                .is_none(),
            "the previous trust-device record should be deleted during rotation",
        );
        assert!(
            ctx.database
                .get_verification_by_identifier(&rotated_identifier)
                .await
                .unwrap()
                .is_some(),
            "the rotated trust-device record should be persisted",
        );
    }

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn totp_enrollment_and_real_verification_apply_user_hooks_and_preserve_rotation_owner() {
        let (mut ctx, user, session) =
            create_test_context_with_credential_user("reissue@example.com", false).await;
        ctx.database.delete_session(&session.token).await.unwrap();
        let session = ctx
            .database
            .create_session(alibi_core::CreateSession {
                token: None,
                user_id: user.id.clone(),
                expires_at: session.expires_at,
                ip_address: Some("127.0.0.7".into()),
                user_agent: Some("configured-agent".into()),
                active_organization_id: Some("configured-organization".into()),
                active_team_id: Some("configured-team".into()),
                impersonated_by: Some("configured-admin".into()),
                additional_fields: alibi_core::field_policy::FieldValues::default(),
            })
            .await
            .unwrap();
        let plugin = TwoFactorPlugin::new();
        let mut configured =
            alibi_core::AuthInitContext::new(Arc::clone(&ctx.config), Arc::clone(&ctx.database));
        plugin.on_init(&mut configured).await.unwrap();
        crate::OrganizationPlugin::with_config(crate::organization::OrganizationConfig {
            teams: crate::organization::TeamsConfig {
                enabled: true,
                ..Default::default()
            },
            ..Default::default()
        })
        .on_init(&mut configured)
        .await
        .unwrap();
        crate::AdminPlugin::new()
            .on_init(&mut configured)
            .await
            .unwrap();
        let configured = configured.into_parts();
        ctx.metadata = configured.metadata;
        ctx.extensions = configured.extensions;
        let cookie = create_session_cookie(&session.token, &ctx.config).unwrap();
        let mut enrollment = AuthRequest::new(HttpMethod::Post, "/two-factor/enable");
        _ = enrollment
            .headers
            .insert("cookie".into(), cookie.split(';').next().unwrap().into());
        enrollment.body = Some(br#"{"password":"password123"}"#.to_vec());
        let enabled = plugin.on_request(&enrollment, &ctx).await.unwrap().unwrap();
        assert_eq!(enabled.status, 200);
        let factor = ctx
            .database
            .get_two_factor_by_user_id(&user.id)
            .await
            .unwrap()
            .unwrap();
        assert_eq!(factor.verified, Some(false));

        let captured = Arc::new(std::sync::Mutex::new(Vec::new()));
        let observed = Arc::clone(&captured);
        let mut init =
            alibi_core::AuthInitContext::new(Arc::clone(&ctx.config), Arc::clone(&ctx.database));
        // Keep the initialized plugin policy when adding the observer. The
        // wrapped adapter projects the old snapshot using this same context.
        init.metadata = ctx.metadata.clone();
        init.extensions = ctx.extensions.clone();
        init.register_user_update_transform(move |id, mut update| {
            observed
                .lock()
                .unwrap()
                .push((id.to_owned(), update.two_factor_enabled));
            if update.two_factor_enabled == Some(true) {
                update.name = Some("Hook Updated Owner".into());
            }
            Ok(update)
        });
        ctx.database = init.database_with_registered_transforms();
        let plaintext = decrypt_value(&ctx.config, &factor.secret).unwrap();
        let code = plugin.generate_totp(&plaintext).unwrap();
        let mut verification = AuthRequest::new(HttpMethod::Post, "/two-factor/verify-totp");
        verification.headers = enrollment.headers;
        verification.body = Some(serde_json::to_vec(&serde_json::json!({"code":code})).unwrap());
        let response = plugin
            .on_request(&verification, &ctx)
            .await
            .unwrap()
            .unwrap();
        assert_eq!(response.status, 200);
        let payload: serde_json::Value = serde_json::from_slice(&response.body).unwrap();
        assert_eq!(
            (*(*(payload).get("user").unwrap_or(&serde_json::Value::Null))
                .get("twoFactorEnabled")
                .unwrap_or(&serde_json::Value::Null)),
            false
        );
        // Upstream returns the old snapshot while rotating the browser cookie.
        assert_eq!(
            (*(payload).get("token").unwrap_or(&serde_json::Value::Null)),
            session.token
        );
        let set_cookie_headers = response.headers.get_all("Set-Cookie").collect::<Vec<_>>();
        let rotated = alibi_core::utils::cookie_utils::verify_cookie_value(
            &cookie_value(
                (set_cookie_headers)
                    .first()
                    .expect("fixture contains the requested index"),
            ),
            &ctx.config.secret,
        )
        .expect("the session cookie must authenticate its token");
        assert_ne!(rotated, session.token);
        assert_eq!(set_cookie_headers.len(), 1);
        assert!(
            ctx.database
                .get_session(&session.token)
                .await
                .unwrap()
                .is_none(),
            "the original session should be deleted after re-issuing",
        );
        assert!(
            ctx.database.get_session(&rotated).await.unwrap().is_some(),
            "the new session token should be persisted",
        );
        assert_eq!(
            *captured.lock().unwrap(),
            vec![(user.id.clone(), Some(true))]
        );
        let stored_user = ctx
            .database
            .get_user_by_id(&user.id)
            .await
            .unwrap()
            .unwrap();
        assert_eq!(stored_user.two_factor_enabled, Some(true));
        assert_eq!(stored_user.name(), Some("Hook Updated Owner"));
        let verified_factor = ctx
            .database
            .get_two_factor_by_user_id(&user.id)
            .await
            .unwrap()
            .unwrap();
        assert_eq!(verified_factor.id, factor.id);
        assert_eq!(verified_factor.secret, factor.secret);
        assert_eq!(verified_factor.backup_codes, factor.backup_codes);
        assert_eq!(verified_factor.verified, Some(true));
        let mut browser = AuthRequest::new(HttpMethod::Get, "/get-session");
        _ = browser.headers.insert(
            "cookie".into(),
            (*(set_cookie_headers)
                .first()
                .expect("fixture contains the requested index"))
            .split(';')
            .next()
            .unwrap()
            .into(),
        );
        let (current_user, current_session) = ctx.require_session(&browser).await.unwrap();
        assert_eq!(current_user.id, user.id);
        assert_eq!(current_user.name(), Some("Hook Updated Owner"));
        assert_eq!(current_session.token, rotated);
        assert_eq!(current_session.ip_address, session.ip_address);
        assert_eq!(current_session.user_agent, session.user_agent);
        assert_eq!(
            current_session.active_organization_id,
            session.active_organization_id
        );
        assert_eq!(current_session.active_team_id, session.active_team_id);
        assert_eq!(current_session.impersonated_by, session.impersonated_by);
    }

    #[tokio::test]
    async fn test_view_backup_codes_returns_decrypted_codes() {
        let plugin = TwoFactorPlugin::new();
        let (ctx, user, _session) =
            create_test_context_with_credential_user("view-codes@example.com", true).await;

        let expected_codes = vec!["ABCDE-12345".to_owned(), "FGHIJ-67890".to_owned()];
        let encrypted = encrypt_value(
            &ctx.config,
            &serde_json::to_string(&expected_codes).unwrap(),
        )
        .unwrap();
        _ = ctx
            .database
            .create_two_factor(CreateTwoFactor {
                user_id: user.id.clone(),
                secret: encrypt_value(&ctx.config, "totp-secret").unwrap(),
                backup_codes: encrypted,
                ..Default::default()
            })
            .await
            .unwrap();

        let backup_codes = plugin.view_backup_codes(&user.id, &ctx).await.unwrap();
        assert_eq!(backup_codes, serde_json::json!(expected_codes));
    }

    #[tokio::test]
    async fn test_view_backup_codes_rejects_invalid_stored_json() {
        let plugin = TwoFactorPlugin::new();
        let (ctx, user, _session) =
            create_test_context_with_credential_user("invalid-view-codes@example.com", true).await;

        _ = ctx
            .database
            .create_two_factor(CreateTwoFactor {
                user_id: user.id.clone(),
                secret: encrypt_value(&ctx.config, "totp-secret").unwrap(),
                backup_codes: encrypt_value(&ctx.config, "[").unwrap(),
                ..Default::default()
            })
            .await
            .unwrap();

        let err = plugin.view_backup_codes(&user.id, &ctx).await.unwrap_err();
        assert_eq!(err.to_string(), "Invalid backup code");
    }

    #[test]
    fn test_routes_do_not_expose_view_backup_codes() {
        let plugin = TwoFactorPlugin::new();
        assert!(
            <TwoFactorPlugin as AuthPlugin<TestSchema>>::routes(&plugin)
                .iter()
                .all(|route| route.path != "/two-factor/view-backup-codes"),
            "view-backup-codes must stay server-only",
        );
    }

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn disable_preserves_persisted_extensions_and_removes_all_matching_trust_records() {
        let (mut ctx, user, first_session) =
            create_test_context_with_credential_user("disable-extensions@fixture.test", true).await;
        let mut init =
            alibi_core::AuthInitContext::new(Arc::clone(&ctx.config), Arc::clone(&ctx.database));
        crate::admin::AdminPlugin::new()
            .on_init(&mut init)
            .await
            .unwrap();
        crate::organization::OrganizationPlugin::with_config(
            crate::organization::OrganizationConfig {
                teams: crate::organization::TeamsConfig {
                    enabled: true,
                    ..Default::default()
                },
                ..Default::default()
            },
        )
        .on_init(&mut init)
        .await
        .unwrap();
        ctx.metadata.extend(init.into_parts().metadata);
        ctx.database
            .delete_session(&first_session.token)
            .await
            .unwrap();
        let current = ctx
            .database
            .create_session(alibi_core::CreateSession {
                token: None,
                user_id: user.id.clone(),
                expires_at: Utc::now() + Duration::hours(1),
                ip_address: Some("192.0.2.45".to_owned()),
                user_agent: Some("fixture-agent".to_owned()),
                impersonated_by: Some("trusted-impersonator".to_owned()),
                active_organization_id: Some("trusted-organization".to_owned()),
                active_team_id: Some("trusted-team".to_owned()),
                additional_fields: alibi_core::field_policy::FieldValues::default(),
            })
            .await
            .unwrap();
        ctx.database
            .create_two_factor(CreateTwoFactor {
                user_id: user.id.clone(),
                secret: "stored-secret".to_owned(),
                backup_codes: "stored-codes".to_owned(),
                ..Default::default()
            })
            .await
            .unwrap();
        for _ in 0..2 {
            ctx.database
                .create_verification(CreateVerification {
                    identifier: "trusted-device-record".to_owned(),
                    value: user.id.clone(),
                    expires_at: Utc::now() + Duration::days(30),
                })
                .await
                .unwrap();
        }
        let session_cookie =
            alibi_core::utils::cookie_utils::sign_cookie_value(current.token(), &ctx.config.secret);
        let trust_cookie = alibi_core::utils::cookie_utils::sign_cookie_value(
            "trust-token!trusted-device-record",
            &ctx.config.secret,
        );
        let mut req = AuthRequest::new(HttpMethod::Post, "/two-factor/disable");
        req.body =
            Some(serde_json::to_vec(&serde_json::json!({"password":"password123"})).unwrap());
        _ = req.headers.insert(
            "cookie".to_owned(),
            format!(
                "{}={session_cookie}; {}={trust_cookie}",
                ctx.config.session.cookie_name,
                related_cookie_name(&ctx.config, TRUST_DEVICE_COOKIE_SUFFIX)
            ),
        );
        let response = TwoFactorPlugin::new()
            .on_request(&req, &ctx)
            .await
            .unwrap()
            .unwrap();
        assert_eq!(response.status, 200);
        let stored = ctx.database.get_user_sessions(&user.id).await.unwrap();
        assert_eq!(stored.len(), 1);
        let replacement = (stored)
            .first()
            .expect("fixture contains the requested index");
        assert_ne!(replacement.token(), current.token());
        assert_eq!(
            replacement.active_organization_id(),
            Some("trusted-organization")
        );
        assert_eq!(replacement.active_team_id(), Some("trusted-team"));
        assert_eq!(replacement.impersonated_by(), Some("trusted-impersonator"));
        assert_eq!(replacement.ip_address(), current.ip_address());
        assert_eq!(replacement.user_agent(), current.user_agent());
        assert!(
            ctx.database
                .get_session(current.token())
                .await
                .unwrap()
                .is_none()
        );
        assert!(
            !ctx.database
                .get_user_by_id(&user.id)
                .await
                .unwrap()
                .unwrap()
                .two_factor_enabled()
        );
        assert!(
            ctx.database
                .get_two_factor_by_user_id(&user.id)
                .await
                .unwrap()
                .is_none()
        );
        assert!(
            ctx.database
                .get_verification_by_identifier("trusted-device-record")
                .await
                .unwrap()
                .is_none()
        );
    }

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn otp_async_codec_failures_consume_only_their_stage_and_delivery_rejection_retains_rotation_state()
     {
        struct Callback {
            fail_store: bool,
            fail_compare: bool,
            observations: Arc<std::sync::Mutex<Vec<String>>>,
            delivered: Arc<std::sync::Mutex<Option<String>>>,
        }
        #[async_trait]
        impl TwoFactorOtpHasher for Callback {
            async fn hash(&self, otp: &str) -> AuthResult<String> {
                let mut observations = self.observations.lock().unwrap();
                observations.push(format!("hash:{otp}"));
                if self.fail_store || (self.fail_compare && observations.len() > 2) {
                    return Err(AuthError::Upstream {
                        status: 400,
                        code: "CODEC_REJECTED",
                        message: "Configured codec rejected",
                    });
                }
                drop(observations);

                Ok(format!("stored-{otp}"))
            }
        }
        #[async_trait]
        impl SendTwoFactorOtp for Callback {
            async fn send(&self, _user: &UserView, otp: &str) -> AuthResult<()> {
                self.observations
                    .lock()
                    .unwrap()
                    .push(format!("send:{otp}"));
                *self.delivered.lock().unwrap() = Some(otp.to_owned());
                Err(AuthError::bad_request("Configured async delivery rejected"))
            }
        }
        for (fail_store, fail_compare) in [(true, false), (false, true), (false, false)] {
            let (mut ctx, user, original) =
                create_test_context_with_credential_user("codec@fixture.test", false).await;
            ctx.database.delete_session(&original.token).await.unwrap();
            let session = ctx
                .database
                .create_session(alibi_core::CreateSession {
                    additional_fields: alibi_core::field_policy::FieldValues::default(),
                    token: None,
                    user_id: user.id.clone(),
                    expires_at: original.expires_at,
                    ip_address: Some("127.0.0.8".into()),
                    user_agent: Some("otp-agent".into()),
                    active_organization_id: Some("otp-org".into()),
                    active_team_id: Some("otp-team".into()),
                    impersonated_by: Some("otp-admin".into()),
                })
                .await
                .unwrap();
            let observations = Arc::new(std::sync::Mutex::new(Vec::new()));
            let delivered = Arc::new(std::sync::Mutex::new(None));
            let callback = Arc::new(Callback {
                fail_store,
                fail_compare,
                observations: Arc::clone(&observations),
                delivered: Arc::clone(&delivered),
            });
            let plugin = TwoFactorPlugin::with_config(TwoFactorConfig {
                send_otp: Some(Arc::<Callback>::clone(&callback)),
                otp_storage: TwoFactorOtpStorage::CustomHash(callback),
                ..Default::default()
            });
            let mut init = alibi_core::AuthInitContext::new(
                Arc::clone(&ctx.config),
                Arc::clone(&ctx.database),
            );
            plugin.on_init(&mut init).await.unwrap();
            crate::OrganizationPlugin::with_config(crate::organization::OrganizationConfig {
                teams: crate::organization::TeamsConfig {
                    enabled: true,
                    ..Default::default()
                },
                ..Default::default()
            })
            .on_init(&mut init)
            .await
            .unwrap();
            crate::AdminPlugin::new().on_init(&mut init).await.unwrap();

            init.register_user_update_transform(|_, mut update| {
                if update.two_factor_enabled == Some(true) {
                    update.name = Some("OTP Hook Owner".into());
                }
                Ok(update)
            });
            ctx.database = init.database_with_registered_transforms();
            let parts = init.into_parts();
            ctx.metadata = parts.metadata;
            ctx.extensions = parts.extensions;

            let identifier = format!("2fa-otp-{}!{}", user.id, session.id);
            let mut request = AuthRequest::new(HttpMethod::Post, "/two-factor/send-otp");
            request.headers.insert(
                "cookie".into(),
                create_session_cookie(&session.token, &ctx.config)
                    .unwrap()
                    .split(';')
                    .next()
                    .unwrap()
                    .into(),
            );
            request.body = Some(b"{}".to_vec());
            let result = plugin.on_request(&request, &ctx).await;
            if fail_store {
                assert!(matches!(
                    result,
                    Err(AuthError::Upstream {
                        code: "CODEC_REJECTED",
                        ..
                    })
                ));
                assert!(delivered.lock().unwrap().is_none());
                assert!(
                    ctx.database
                        .get_latest_verification_by_identifier(&identifier)
                        .await
                        .unwrap()
                        .is_none()
                );
            } else {
                assert_eq!(
                    result.unwrap().unwrap().status,
                    200,
                    "async delivery failure must retain issued state"
                );
                let otp = delivered
                    .lock()
                    .unwrap()
                    .clone()
                    .expect("actual callback delivery");
                let stored = ctx
                    .database
                    .get_latest_verification_by_identifier(&identifier)
                    .await
                    .unwrap()
                    .unwrap();
                assert_eq!(stored.value, format!("stored-{otp}:0"));
                request.path = "/two-factor/verify-otp".into();
                request.body = Some(serde_json::to_vec(&serde_json::json!({"code":otp})).unwrap());
                let result_2 = plugin.on_request(&request, &ctx).await;
                assert!(
                    ctx.database
                        .get_latest_verification_by_identifier(&identifier)
                        .await
                        .unwrap()
                        .is_none()
                );
                if fail_compare {
                    assert!(matches!(
                        result_2,
                        Err(AuthError::Upstream {
                            code: "CODEC_REJECTED",
                            ..
                        })
                    ));
                    assert!(
                        !ctx.database
                            .get_user_by_id(&user.id)
                            .await
                            .unwrap()
                            .unwrap()
                            .two_factor_enabled()
                    );
                    assert!(
                        ctx.database
                            .get_session(&session.token)
                            .await
                            .unwrap()
                            .is_some()
                    );
                } else {
                    let response = result_2.unwrap().unwrap();
                    assert_eq!(response.status, 200);
                    let payload: serde_json::Value =
                        serde_json::from_slice(&response.body).unwrap();
                    assert_eq!(
                        (*(*(payload).get("user").unwrap_or(&serde_json::Value::Null))
                            .get("name")
                            .unwrap_or(&serde_json::Value::Null)),
                        "OTP Hook Owner"
                    );
                    assert_eq!(
                        (*(*(payload).get("user").unwrap_or(&serde_json::Value::Null))
                            .get("twoFactorEnabled")
                            .unwrap_or(&serde_json::Value::Null)),
                        true
                    );
                    let token = (*(payload).get("token").unwrap_or(&serde_json::Value::Null))
                        .as_str()
                        .unwrap();
                    assert_ne!(token, session.token);
                    let new = ctx.database.get_session(token).await.unwrap().unwrap();
                    assert_eq!(new.user_id, session.user_id);
                    assert_eq!(new.ip_address, session.ip_address);
                    assert_eq!(new.user_agent, session.user_agent);
                    assert_eq!(new.active_organization_id, session.active_organization_id);
                    assert_eq!(new.active_team_id, session.active_team_id);
                    assert_eq!(new.impersonated_by, session.impersonated_by);
                    assert!(
                        ctx.database
                            .get_session(&session.token)
                            .await
                            .unwrap()
                            .is_none()
                    );
                    let header = response.headers.get_all("Set-Cookie").next().unwrap();
                    assert_eq!(
                        alibi_core::utils::cookie_utils::verify_cookie_value(
                            &cookie_value(header),
                            &ctx.config.secret
                        )
                        .as_deref(),
                        Some(token)
                    );
                }
                assert_eq!(
                    observations.lock().unwrap().as_slice(),
                    &[
                        format!("hash:{otp}"),
                        format!("send:{otp}"),
                        format!("hash:{otp}")
                    ]
                );
            }
            assert!(
                ctx.database
                    .get_two_factor_by_user_id(&user.id)
                    .await
                    .unwrap()
                    .is_none()
            );
        }
    }

    #[tokio::test]
    async fn otp_enable_without_delivery_checks_password_then_rejects_without_mutating_the_owner() {
        let (ctx, user, session) =
            create_test_context_with_credential_user("disabled-otp@fixture.test", false).await;
        let plugin = TwoFactorPlugin::new();
        let mut request = AuthRequest::new(HttpMethod::Post, "/two-factor/enable");
        request.headers.insert(
            "cookie".into(),
            create_session_cookie(&session.token, &ctx.config)
                .unwrap()
                .split(';')
                .next()
                .unwrap()
                .into(),
        );
        request.body = Some(br#"{"password":"wrong","method":"otp"}"#.to_vec());
        let wrong = plugin.on_request(&request, &ctx).await.unwrap_err();
        assert_eq!(
            serde_json::from_slice::<serde_json::Value>(&wrong.to_auth_response().body).unwrap(),
            serde_json::json!({"message":"Invalid password","code":"INVALID_PASSWORD"})
        );
        request.body = Some(br#"{"password":"password123","method":"otp"}"#.to_vec());
        assert!(matches!(
            plugin.on_request(&request, &ctx).await,
            Err(AuthError::Upstream {
                status: 400,
                code: "OTP_NOT_CONFIGURED",
                message: "OTP is not available"
            })
        ));
        assert!(
            !ctx.database
                .get_user_by_id(&user.id)
                .await
                .unwrap()
                .unwrap()
                .two_factor_enabled()
        );
        assert!(
            ctx.database
                .get_two_factor_by_user_id(&user.id)
                .await
                .unwrap()
                .is_none()
        );
        let sessions = ctx.database.get_user_sessions(&user.id).await.unwrap();
        assert_eq!(sessions.len(), 1);
        assert_eq!(
            (sessions)
                .first()
                .expect("fixture contains the requested index")
                .token,
            session.token
        );
    }

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn configured_backup_callback_errors_preserve_factor_user_and_current_session() {
        // This guards Rust callback error identity, which the cross-runtime happy-path
        // cipher fixture cannot exercise. Pinned runtime independently preserves both
        // 400 and 403 callback errors with this cancellation-like message.
        type Observed = Arc<std::sync::Mutex<Vec<(String, String)>>>;
        fn denied(status: u16) -> AuthError {
            AuthError::Upstream {
                status,
                code: "BACKUP_CALLBACK_DENIED",
                message: "session creation cancelled by database hook",
            }
        }
        struct RejectingCipher {
            status: u16,
            observed: Observed,
        }
        #[async_trait]
        impl TwoFactorBackupCipher for RejectingCipher {
            async fn encrypt(&self, json: &str) -> AuthResult<String> {
                self.observed
                    .lock()
                    .unwrap()
                    .push(("encrypt".into(), json.into()));
                Err(denied(self.status))
            }
            async fn decrypt(&self, _stored: &str) -> AuthResult<String> {
                panic!("enrollment and regeneration must not decode an existing factor")
            }
        }
        for phase in ["generate", "encrypt"] {
            for status in [400, 403] {
                let (ctx, user, session) =
                    create_test_context_with_credential_user("backup-callback@fixture.test", false)
                        .await;
                let observed: Observed = Arc::default();
                let generator_observed = Arc::clone(&observed);
                let plugin = TwoFactorPlugin::with_config(TwoFactorConfig {
                    skip_verification_on_enable: true,
                    custom_backup_codes_generate: Some(Arc::new(move || {
                        generator_observed
                            .lock()
                            .unwrap()
                            .push(("generate".into(), String::new()));
                        if phase == "generate" {
                            Err(denied(status))
                        } else {
                            Ok(vec!["callback-code".into(), "callback-code".into()])
                        }
                    })),
                    backup_storage: TwoFactorBackupStorage::CustomCipher(Arc::new(
                        RejectingCipher {
                            status,
                            observed: Arc::clone(&observed),
                        },
                    )),
                    ..Default::default()
                });
                let cookie = create_session_cookie(&session.token, &ctx.config).unwrap();
                let mut request = AuthRequest::new(HttpMethod::Post, "/two-factor/enable");
                _ = request
                    .headers
                    .insert("cookie".into(), cookie.split(';').next().unwrap().into());
                request.body = Some(br#"{"password":"password123"}"#.to_vec());
                // Resolve the short-lived native fixture session before observing
                // callback effects; this performs the ordinary expiry refresh.
                _ = ctx.require_session(&request).await.unwrap();
                let before_sessions =
                    serde_json::to_value(ctx.database.get_user_sessions(&user.id).await.unwrap())
                        .unwrap();
                let error = plugin.on_request(&request, &ctx).await.unwrap_err();
                assert!(
                    matches!(error, AuthError::Upstream { status: actual, code: "BACKUP_CALLBACK_DENIED", .. } if actual == status)
                );
                assert!(
                    ctx.database
                        .get_two_factor_by_user_id(&user.id)
                        .await
                        .unwrap()
                        .is_none()
                );
                assert!(
                    !ctx.database
                        .get_user_by_id(&user.id)
                        .await
                        .unwrap()
                        .unwrap()
                        .two_factor_enabled()
                );
                assert_eq!(
                    serde_json::to_value(ctx.database.get_user_sessions(&user.id).await.unwrap())
                        .unwrap(),
                    before_sessions
                );
                let expected = if phase == "generate" {
                    vec![("generate".into(), String::new())]
                } else {
                    vec![
                        ("generate".into(), String::new()),
                        (
                            "encrypt".into(),
                            "[\"callback-code\",\"callback-code\"]".into(),
                        ),
                    ]
                };
                assert_eq!(*observed.lock().unwrap(), expected);

                _ = ctx
                    .database
                    .update_user(
                        &user.id,
                        UpdateUser {
                            two_factor_enabled: Some(true),
                            ..Default::default()
                        },
                    )
                    .await
                    .unwrap();
                let factor = ctx
                    .database
                    .create_two_factor(CreateTwoFactor {
                        user_id: user.id.clone(),
                        secret: "installed-factor-secret".into(),
                        backup_codes: "installed-factor-codes".into(),
                        verified: Some(true),
                        failed_verification_count: Some(2.5),
                        locked_until: Some(Utc::now() + Duration::minutes(1)),
                    })
                    .await
                    .unwrap();
                let before_user =
                    serde_json::to_value(ctx.database.get_user_by_id(&user.id).await.unwrap())
                        .unwrap();
                observed.lock().unwrap().clear();
                let mut regeneration =
                    AuthRequest::new(HttpMethod::Post, "/two-factor/generate-backup-codes");
                regeneration.headers.clone_from(&request.headers);
                regeneration.body.clone_from(&request.body);
                let error_2 = plugin.on_request(&regeneration, &ctx).await.unwrap_err();
                assert!(
                    matches!(error_2, AuthError::Upstream { status: actual, code: "BACKUP_CALLBACK_DENIED", .. } if actual == status)
                );
                assert_eq!(*observed.lock().unwrap(), expected);
                assert_eq!(
                    serde_json::to_value(
                        ctx.database
                            .get_two_factor_by_user_id(&user.id)
                            .await
                            .unwrap()
                    )
                    .unwrap(),
                    serde_json::json!(factor)
                );
                assert_eq!(
                    serde_json::to_value(ctx.database.get_user_by_id(&user.id).await.unwrap())
                        .unwrap(),
                    before_user
                );
                assert_eq!(
                    serde_json::to_value(ctx.database.get_user_sessions(&user.id).await.unwrap())
                        .unwrap(),
                    before_sessions
                );
                assert_eq!(
                    ctx.require_session(&regeneration).await.unwrap().1.token,
                    session.token
                );
            }
        }
    }

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn pending_backup_cipher_errors_restore_only_decode_stage_attempts() {
        type Observed = Arc<std::sync::Mutex<Vec<(String, String)>>>;
        struct Cipher {
            phase: &'static str,
            rejected: Arc<std::sync::atomic::AtomicBool>,
            observed: Observed,
        }
        #[async_trait]
        impl TwoFactorBackupCipher for Cipher {
            async fn encrypt(&self, json: &str) -> AuthResult<String> {
                self.observed
                    .lock()
                    .unwrap()
                    .push(("encrypt".into(), json.into()));
                if self.phase == "encrypt"
                    && self.rejected.load(std::sync::atomic::Ordering::SeqCst)
                {
                    return Err(AuthError::Upstream {
                        status: 400,
                        code: "BACKUP_CALLBACK_DENIED",
                        message: "Callback denied",
                    });
                }
                Ok(format!("backup-{json}"))
            }
            async fn decrypt(&self, stored: &str) -> AuthResult<String> {
                self.observed
                    .lock()
                    .unwrap()
                    .push(("decrypt".into(), stored.into()));
                if self.phase == "decrypt"
                    && self.rejected.load(std::sync::atomic::Ordering::SeqCst)
                {
                    return Err(AuthError::Upstream {
                        status: 400,
                        code: "BACKUP_CALLBACK_DENIED",
                        message: "Callback denied",
                    });
                }
                Ok(stored.strip_prefix("backup-").unwrap().into())
            }
        }
        for phase in ["decrypt", "encrypt"] {
            let (ctx, user, session) = create_test_context_with_credential_user(
                "backup-pending-callback@fixture.test",
                true,
            )
            .await;
            ctx.database.delete_session(&session.token).await.unwrap();
            let factor = ctx
                .database
                .create_two_factor(CreateTwoFactor {
                    user_id: user.id.clone(),
                    secret: "installed-factor-secret".into(),
                    backup_codes: "backup-[\"same\",\"same\",\"remaining\"]".into(),
                    verified: Some(true),
                    failed_verification_count: Some(2.5),
                    locked_until: None,
                })
                .await
                .unwrap();
            let challenge = begin_sign_in_challenge(&user, None, &ctx).await.unwrap();
            let cookie_name = related_cookie_name(&ctx.config, TWO_FACTOR_COOKIE_SUFFIX);
            let cookie = challenge
                .set_cookie_headers
                .iter()
                .find(|header| header.starts_with(&format!("{cookie_name}=")))
                .unwrap();
            let mut request = AuthRequest::new(HttpMethod::Post, "/two-factor/verify-backup-code");
            request
                .headers
                .insert("cookie".into(), cookie.split(';').next().unwrap().into());
            request.body =
                Some(br#"{"code":"same","disableSession":true,"trustDevice":true}"#.to_vec());
            let key = read_signed_cookie(&request, TWO_FACTOR_COOKIE_SUFFIX, &ctx).unwrap();
            let attempt_id = format!("2fa-attempts-{key}");
            let before_challenge = serde_json::to_value(
                ctx.database
                    .get_verification_by_identifier(&key)
                    .await
                    .unwrap(),
            )
            .unwrap();
            let before_user =
                serde_json::to_value(ctx.database.get_user_by_id(&user.id).await.unwrap()).unwrap();
            let rejected = Arc::new(std::sync::atomic::AtomicBool::new(true));
            let observed: Observed = Arc::default();
            let plugin = TwoFactorPlugin::with_config(TwoFactorConfig {
                backup_storage: TwoFactorBackupStorage::CustomCipher(Arc::new(Cipher {
                    phase,
                    rejected: Arc::clone(&rejected),
                    observed: Arc::clone(&observed),
                })),
                ..Default::default()
            });
            let error = plugin.on_request(&request, &ctx).await.unwrap_err();
            assert!(matches!(
                error,
                AuthError::Upstream {
                    status: 400,
                    code: "BACKUP_CALLBACK_DENIED",
                    ..
                }
            ));
            assert_eq!(
                serde_json::to_value(
                    ctx.database
                        .get_two_factor_by_user_id(&user.id)
                        .await
                        .unwrap()
                )
                .unwrap(),
                serde_json::json!(factor)
            );
            assert_eq!(
                serde_json::to_value(
                    ctx.database
                        .get_verification_by_identifier(&key)
                        .await
                        .unwrap()
                )
                .unwrap(),
                before_challenge
            );
            assert_eq!(
                serde_json::to_value(ctx.database.get_user_by_id(&user.id).await.unwrap()).unwrap(),
                before_user
            );
            assert_eq!(
                ctx.database
                    .get_user_sessions(&user.id)
                    .await
                    .unwrap()
                    .len(),
                0
            );
            let attempts = ctx
                .database
                .get_verification_by_identifier(&attempt_id)
                .await
                .unwrap();
            if phase == "decrypt" {
                assert_eq!(attempts.unwrap().value(), "0");
                assert_eq!(
                    *observed.lock().unwrap(),
                    vec![("decrypt".into(), factor.backup_codes.clone())]
                );
            } else {
                assert!(attempts.is_none());
                assert_eq!(
                    *observed.lock().unwrap(),
                    vec![
                        ("decrypt".into(), factor.backup_codes.clone()),
                        ("encrypt".into(), "[\"remaining\"]".into())
                    ]
                );
            }
            rejected.store(false, std::sync::atomic::Ordering::SeqCst);
            let retry = plugin.on_request(&request, &ctx).await;
            if phase == "decrypt" {
                let response = retry.unwrap().unwrap();
                assert_eq!(response.status, 200);
                let body: serde_json::Value = serde_json::from_slice(&response.body).unwrap();
                assert!(body.get("token").is_none());
                assert_eq!(
                    (*(*(body).get("user").unwrap_or(&serde_json::Value::Null))
                        .get("id")
                        .unwrap_or(&serde_json::Value::Null)),
                    user.id
                );
                assert!(response.headers.get("set-cookie").is_none());
                let updated = ctx
                    .database
                    .get_two_factor_by_user_id(&user.id)
                    .await
                    .unwrap()
                    .unwrap();
                assert_eq!(updated.id, factor.id);
                assert_eq!(updated.secret, factor.secret);
                assert_eq!(updated.backup_codes, "backup-[\"remaining\"]");
                assert_eq!(updated.failed_verification_count, Some(0.0));
            } else {
                let error_2 = retry.unwrap_err();
                assert_eq!(error_2.status_code(), 401);
                assert_eq!(error_2.to_string(), "Invalid two factor cookie");
                assert_eq!(
                    serde_json::to_value(
                        ctx.database
                            .get_two_factor_by_user_id(&user.id)
                            .await
                            .unwrap()
                    )
                    .unwrap(),
                    serde_json::json!(factor)
                );
            }
            assert!(
                ctx.database
                    .get_verification_by_identifier(&attempt_id)
                    .await
                    .unwrap()
                    .is_none()
            );
            assert_eq!(
                serde_json::to_value(
                    ctx.database
                        .get_verification_by_identifier(&key)
                        .await
                        .unwrap()
                )
                .unwrap(),
                before_challenge
            );
            assert_eq!(
                ctx.database
                    .get_user_sessions(&user.id)
                    .await
                    .unwrap()
                    .len(),
                0
            );
        }
    }

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn authenticated_otp_maps_only_session_creation_cancellation_and_preserves_hook_inputs() {
        use alibi_core::{AuthConfig, CreateSession};
        use alibi_seaorm::{
            Database, DatabaseHooks, HookControl, SeaOrmBackend, SeaOrmHookContext, SeaOrmStore,
        };

        struct Hook {
            mode: u8,
            observed: Arc<std::sync::Mutex<Vec<String>>>,
        }
        #[async_trait]
        impl DatabaseHooks<TestSchema, SeaOrmBackend> for Hook {
            async fn before_update_user(
                &self,
                _id: &str,
                update: &mut UpdateUser,
                _context: &SeaOrmHookContext<'_>,
            ) -> AuthResult<HookControl> {
                if update.two_factor_enabled == Some(true) {
                    self.observed.lock().unwrap().push("user".into());
                    if self.mode == 3 {
                        return Err(AuthError::SessionCreationCancelled);
                    }
                }
                Ok(HookControl::Continue)
            }
            async fn before_create_session(
                &self,
                session: &mut CreateSession,
                context: &SeaOrmHookContext<'_>,
            ) -> AuthResult<HookControl> {
                if context
                    .request
                    .as_ref()
                    .is_some_and(|request| request.path.ends_with("/two-factor/verify-otp"))
                {
                    self.observed.lock().unwrap().push("session".into());
                    assert_eq!(session.ip_address.as_deref(), Some("127.0.0.7"));
                    assert_eq!(session.user_agent.as_deref(), Some("original-otp-agent"));
                    return match self.mode {
                        0 => Ok(HookControl::Cancel),
                        1 => Err(AuthError::forbidden(
                            "session creation cancelled by database hook",
                        )),
                        _ => Err(AuthError::bad_request(
                            "session creation cancelled by database hook",
                        )),
                    };
                }
                Ok(HookControl::Continue)
            }
        }
        struct Sender(Arc<std::sync::Mutex<Option<String>>>);
        #[async_trait]
        impl SendTwoFactorOtp for Sender {
            async fn send(&self, _user: &UserView, otp: &str) -> AuthResult<()> {
                *self.0.lock().unwrap() = Some(otp.into());
                Ok(())
            }
        }
        for mode in 0..4 {
            let database = Database::connect("sqlite::memory:").await.unwrap();
            alibi_seaorm::store::__private_test_support::migrator::run_migrations(&database)
                .await
                .unwrap();
            let config = Arc::new(AuthConfig::new("authenticated-otp-hook-secret-at-least-32"));
            let mut ctx = AuthContext::new(
                Arc::clone(&config),
                Arc::new(SeaOrmStore::<TestSchema>::new(
                    Arc::clone(&config),
                    database.clone(),
                )),
            );
            let user = ctx
                .database
                .create_user(
                    CreateUser::new()
                        .with_name("Native OTP Owner")
                        .with_email("native-otp@fixture.test"),
                )
                .await
                .unwrap();
            let session = ctx
                .database
                .create_session(CreateSession {
                    token: None,
                    user_id: user.id.clone(),
                    expires_at: Utc::now() + Duration::hours(1),
                    ip_address: Some("127.0.0.7".into()),
                    user_agent: Some("original-otp-agent".into()),
                    active_organization_id: None,
                    active_team_id: None,
                    impersonated_by: None,
                    additional_fields: alibi_core::field_policy::FieldValues::default(),
                })
                .await
                .unwrap();
            let observed = Arc::new(std::sync::Mutex::new(Vec::new()));
            ctx.database = Arc::new(SeaOrmStore::<TestSchema>::new(config, database).with_hooks(
                vec![Arc::new(Hook {
                    mode,
                    observed: Arc::clone(&observed),
                })],
            ));
            let delivered = Arc::new(std::sync::Mutex::new(None));
            let plugin = TwoFactorPlugin::with_config(TwoFactorConfig {
                send_otp: Some(Arc::new(Sender(Arc::clone(&delivered)))),
                ..Default::default()
            });
            let mut init = alibi_core::AuthInitContext::new(
                Arc::clone(&ctx.config),
                Arc::clone(&ctx.database),
            );
            plugin.on_init(&mut init).await.unwrap();
            ctx.database = init.database_with_registered_transforms();
            let parts = init.into_parts();
            ctx.metadata = parts.metadata;
            ctx.extensions = parts.extensions;
            let mut request = AuthRequest::new(HttpMethod::Post, "/two-factor/send-otp");
            request.headers.insert(
                "cookie".into(),
                create_session_cookie(&session.token, &ctx.config)
                    .unwrap()
                    .split(';')
                    .next()
                    .unwrap()
                    .into(),
            );
            request.body = Some(b"{}".to_vec());
            assert_eq!(
                plugin
                    .on_request(&request, &ctx)
                    .await
                    .unwrap()
                    .unwrap()
                    .status,
                200
            );
            let before = ctx.database.get_user_sessions(&user.id).await.unwrap();
            assert_eq!(before.len(), 1);
            assert_eq!(
                (before)
                    .first()
                    .expect("fixture contains the requested session")
                    .id,
                session.id
            );
            assert_eq!(
                (before)
                    .first()
                    .expect("fixture contains the requested session")
                    .token,
                session.token
            );
            let code = delivered
                .lock()
                .unwrap()
                .clone()
                .expect("real delivery required");
            request.path = "/two-factor/verify-otp".into();
            request.body = Some(
                serde_json::to_vec(&serde_json::json!({"code":code,"trustDevice":true})).unwrap(),
            );
            let result =
                alibi_core::with_request_hook_context(&request, plugin.on_request(&request, &ctx))
                    .await;
            match mode {
                0 => {
                    let response = result.unwrap().unwrap();
                    assert_eq!(response.status, 500);
                    assert_eq!(response.body.len(), 0);
                    assert_eq!(response.headers.get_all("Set-Cookie").count(), 0);
                }
                1 => assert!(matches!(result, Err(AuthError::Forbidden(_)))),
                2 => assert!(matches!(result, Err(AuthError::BadRequest(_)))),
                _ => assert!(matches!(result, Err(AuthError::SessionCreationCancelled))),
            }
            assert_eq!(
                *observed.lock().unwrap(),
                if mode == 3 {
                    vec!["user"]
                } else {
                    vec!["user", "session"]
                }
            );
            let stored_user = ctx
                .database
                .get_user_by_id(&user.id)
                .await
                .unwrap()
                .unwrap();
            assert_eq!(stored_user.two_factor_enabled(), mode != 3);
            let sessions = ctx.database.get_user_sessions(&user.id).await.unwrap();
            assert_eq!(
                serde_json::to_value(sessions).unwrap(),
                serde_json::to_value(&before).unwrap()
            );
            assert!(
                ctx.database
                    .get_two_factor_by_user_id(&user.id)
                    .await
                    .unwrap()
                    .is_none()
            );
            let key = format!("2fa-otp-{}!{}", user.id, session.id);
            assert!(
                ctx.database
                    .get_verification_by_identifier(&key)
                    .await
                    .unwrap()
                    .is_none()
            );
            let retry = plugin.on_request(&request, &ctx).await.unwrap_err();
            assert!(
                matches!(retry, AuthError::BadRequest(ref message) if message == "OTP has expired")
            );
            assert_eq!(
                ctx.database
                    .get_user_by_id(&user.id)
                    .await
                    .unwrap()
                    .unwrap()
                    .two_factor_enabled(),
                mode != 3
            );
        }
    }

    #[tokio::test]
    #[expect(
        clippy::too_many_lines,
        reason = "Keep this ordered integration scenario and its assertions together; Result propagates setup failures"
    )]
    async fn otp_delivery_keeps_owned_request_context_after_dropped_rejecting_observer_and_rotation()
     {
        use alibi_core::{BackgroundTaskCompletion, BackgroundTaskHandler};
        use tokio::sync::oneshot;
        type Observations = Arc<std::sync::Mutex<Vec<(String, String, String, bool)>>>;
        struct Sender {
            entered: std::sync::Mutex<Option<oneshot::Sender<String>>>,
            release: std::sync::Mutex<Option<oneshot::Receiver<()>>>,
            finished: std::sync::Mutex<Option<oneshot::Sender<()>>>,
            observed: Observations,
        }
        impl Sender {
            fn record(&self, user: &UserView) {
                let request = alibi_core::hooks::current_request_hook_context()
                    .expect("real delivery must retain initiating request context");
                self.observed.lock().unwrap().push((
                    request.path,
                    (*(request.headers)
                        .get("x-delivery-origin")
                        .expect("fixture contains the requested index"))
                    .clone(),
                    (*(request.query)
                        .get("delivery")
                        .expect("fixture contains the requested index"))
                    .clone(),
                    user.two_factor_enabled.unwrap_or(false),
                ));
            }
        }
        #[async_trait]
        impl SendTwoFactorOtp for Sender {
            async fn send(&self, user: &UserView, otp: &str) -> AuthResult<()> {
                self.record(user);
                _ = self
                    .entered
                    .lock()
                    .unwrap()
                    .take()
                    .unwrap()
                    .send(otp.into());
                let release = self.release.lock().unwrap().take().unwrap();
                release.await.unwrap();
                self.record(user);
                _ = self.finished.lock().unwrap().take().unwrap().send(());
                Err(AuthError::forbidden(
                    "actual asynchronous delivery rejected",
                ))
            }
        }
        struct Observer;
        impl BackgroundTaskHandler for Observer {
            fn handle(&self, completion: BackgroundTaskCompletion) -> AuthResult<()> {
                drop(completion);
                Err(AuthError::forbidden("actual application observer rejected"))
            }
        }
        let mut ctx = test_helpers::create_test_context().await;
        ctx.config = Arc::new((*ctx.config).clone().background_tasks(Arc::new(Observer)));
        let (entered, entry) = oneshot::channel();
        let (release, gate) = oneshot::channel();
        let (finished, done) = oneshot::channel();
        let observed = Arc::new(std::sync::Mutex::new(Vec::new()));
        let plugin = Arc::new(TwoFactorPlugin::with_config(TwoFactorConfig {
            send_otp: Some(Arc::new(Sender {
                entered: std::sync::Mutex::new(Some(entered)),
                release: std::sync::Mutex::new(Some(gate)),
                finished: std::sync::Mutex::new(Some(finished)),
                observed: Arc::clone(&observed),
            })),
            ..Default::default()
        }));
        let mut init =
            alibi_core::AuthInitContext::new(Arc::clone(&ctx.config), Arc::clone(&ctx.database));
        plugin.on_init(&mut init).await.unwrap();
        ctx.database = init.database_with_registered_transforms();
        let parts = init.into_parts();
        ctx.metadata = parts.metadata;
        ctx.extensions = parts.extensions;
        let user = test_helpers::create_user(
            &ctx,
            CreateUser::new()
                .with_name("Owned Delivery")
                .with_email("delivery-owner@fixture.test"),
        )
        .await;
        let session = test_helpers::create_session(&ctx, user.id.clone(), Duration::hours(1)).await;
        let ctx = Arc::new(ctx);
        let mut request = AuthRequest::new(HttpMethod::Post, "/two-factor/send-otp");
        _ = request.headers.insert(
            "cookie".into(),
            create_session_cookie(&session.token, &ctx.config)
                .unwrap()
                .split(';')
                .next()
                .unwrap()
                .into(),
        );
        _ = request
            .headers
            .insert("x-delivery-origin".into(), "original-sender".into());
        _ = request
            .query
            .insert("delivery".into(), "original-marker".into());
        request.body = Some(b"{}".to_vec());
        let running_ctx = Arc::clone(&ctx);
        let running_plugin = Arc::clone(&plugin);
        let running_request = request.clone();
        let mut running = tokio::spawn(async move {
            alibi_core::with_request_hook_context(
                &running_request,
                running_plugin.on_request(&running_request, &running_ctx),
            )
            .await
        });
        let code = entry.await.unwrap();
        let response = tokio::time::timeout(std::time::Duration::from_secs(1), &mut running)
            .await
            .expect("configured response must not await held application delivery")
            .unwrap()
            .unwrap()
            .unwrap();
        assert_eq!(response.status, 200);
        let identifier = format!("2fa-otp-{}!{}", user.id, session.id);
        let row = ctx
            .database
            .get_verification_by_identifier(&identifier)
            .await
            .unwrap()
            .unwrap();
        assert_eq!(row.value(), format!("{code}:0"));
        assert!(
            ctx.database
                .get_session(&session.token)
                .await
                .unwrap()
                .is_some()
        );
        request.path = "/two-factor/verify-otp".into();
        _ = request
            .headers
            .insert("x-delivery-origin".into(), "different-verifier".into());
        _ = request
            .query
            .insert("delivery".into(), "different-marker".into());
        request.body = Some(serde_json::to_vec(&serde_json::json!({"code":code})).unwrap());
        let verified =
            alibi_core::with_request_hook_context(&request, plugin.on_request(&request, &ctx))
                .await
                .unwrap()
                .unwrap();
        assert_eq!(verified.status, 200);
        assert!(
            ctx.database
                .get_user_by_id(&user.id)
                .await
                .unwrap()
                .unwrap()
                .two_factor_enabled()
        );
        assert!(
            ctx.database
                .get_session(&session.token)
                .await
                .unwrap()
                .is_none()
        );
        assert!(
            ctx.database
                .get_verification_by_identifier(&identifier)
                .await
                .unwrap()
                .is_none()
        );
        assert!(
            ctx.database
                .get_two_factor_by_user_id(&user.id)
                .await
                .unwrap()
                .is_none()
        );
        _ = release.send(());
        tokio::time::timeout(std::time::Duration::from_secs(1), done)
            .await
            .expect("owned delivery must finish after its actual release")
            .unwrap();
        assert_eq!(
            *observed.lock().unwrap(),
            vec![
                (
                    "/two-factor/send-otp".into(),
                    "original-sender".into(),
                    "original-marker".into(),
                    false
                );
                2
            ]
        );
    }
}
// LCOV_EXCL_STOP