rustauth-plugins 0.2.0

Official RustAuth plugin modules.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use rustauth_core::context::create_auth_context;
use rustauth_core::db::MemoryAdapter;
use rustauth_core::options::RustAuthOptions;
use rustauth_plugins::siwe::{siwe, SiweSchemaOptions};
use std::sync::Arc;

use super::{nonce, options, router, verify, WALLET};

#[test]
fn siwe_plugin_adds_wallet_address_schema() -> Result<(), Box<dyn std::error::Error>> {
    let context = create_auth_context(RustAuthOptions {
        plugins: vec![siwe(options())?],
        ..RustAuthOptions::default()
    })?;

    let table = context
        .db_schema
        .table("wallet_address")
        .ok_or("wallet_address schema missing")?;
    assert_eq!(table.name, "wallet_addresses");
    assert_eq!(
        table.field("user_id").map(|field| field.name.as_str()),
        Some("user_id")
    );
    assert_eq!(
        table.field("address").map(|field| field.name.as_str()),
        Some("address")
    );
    assert_eq!(
        table.field("chain_id").map(|field| field.name.as_str()),
        Some("chain_id")
    );
    assert_eq!(
        table.field("is_primary").map(|field| field.name.as_str()),
        Some("is_primary")
    );
    assert_eq!(
        table.field("created_at").map(|field| field.name.as_str()),
        Some("created_at")
    );
    Ok(())
}

#[test]
fn siwe_schema_options_override_table_and_field_names() -> Result<(), Box<dyn std::error::Error>> {
    let mut opts = options();
    opts = opts.schema(
        SiweSchemaOptions::new()
            .table_name("wallet_address")
            .field_name("user_id", "user_id")
            .field_name("chain_id", "chain_id")
            .field_name("is_primary", "is_primary")
            .field_name("created_at", "created_at"),
    );
    let context = create_auth_context(RustAuthOptions {
        plugins: vec![siwe(opts)?],
        ..RustAuthOptions::default()
    })?;

    let table = context
        .db_schema
        .table("wallet_address")
        .ok_or("wallet_address schema missing")?;
    assert_eq!(table.name, "wallet_address");
    assert_eq!(
        table.field("user_id").map(|field| field.name.as_str()),
        Some("user_id")
    );
    assert_eq!(
        table.field("chain_id").map(|field| field.name.as_str()),
        Some("chain_id")
    );
    assert_eq!(
        table.field("is_primary").map(|field| field.name.as_str()),
        Some("is_primary")
    );
    assert_eq!(
        table.field("created_at").map(|field| field.name.as_str()),
        Some("created_at")
    );
    Ok(())
}

#[tokio::test]
async fn custom_schema_configuration_still_allows_endpoint_flow(
) -> Result<(), Box<dyn std::error::Error>> {
    let adapter = Arc::new(MemoryAdapter::new());
    let router = router(
        adapter.clone(),
        options().schema(
            SiweSchemaOptions::new()
                .table_name("wallet_address")
                .field_name("user_id", "user_id")
                .field_name("address", "wallet_address")
                .field_name("chain_id", "chain_id")
                .field_name("is_primary", "is_primary")
                .field_name("created_at", "created_at"),
        ),
    )?;

    nonce(&router, WALLET, Some(1)).await?;
    let response = verify(
        &router,
        WALLET,
        Some(1),
        "valid_message",
        "valid_signature",
        None,
    )
    .await?;

    assert_eq!(response.status(), http::StatusCode::OK);
    assert_eq!(adapter.len("wallet_address").await, 1);
    Ok(())
}