salsa/tests/manual_registration.rs

#![cfg(not(feature = "inventory"))]

mod ingredients {
    #[salsa::input]
    pub(super) struct MyInput {
        field: u32,
    }

    #[salsa::tracked]
    pub(super) struct MyTracked<'db> {
        pub(super) field: u32,
    }

    #[salsa::interned]
    pub(super) struct MyInterned<'db> {
        pub(super) field: u32,
    }

    #[salsa::tracked]
    pub(super) fn intern<'db>(db: &'db dyn salsa::Database, input: MyInput) -> MyInterned<'db> {
        MyInterned::new(db, input.field(db))
    }

    #[salsa::tracked]
    pub(super) fn track<'db>(db: &'db dyn salsa::Database, input: MyInput) -> MyTracked<'db> {
        MyTracked::new(db, input.field(db))
    }
}

#[salsa::db]
#[derive(Clone, Default)]
pub struct DatabaseImpl {
    storage: salsa::Storage<Self>,
}

#[salsa::db]
impl salsa::Database for DatabaseImpl {}

#[test]
fn single_database() {
    let db = DatabaseImpl {
        storage: salsa::Storage::builder()
            .ingredient::<ingredients::track>()
            .ingredient::<ingredients::intern>()
            .ingredient::<ingredients::MyInput>()
            .ingredient::<ingredients::MyTracked<'_>>()
            .ingredient::<ingredients::MyInterned<'_>>()
            .build(),
    };

    let input = ingredients::MyInput::new(&db, 1);

    let tracked = ingredients::track(&db, input);
    let interned = ingredients::intern(&db, input);

    assert_eq!(tracked.field(&db), 1);
    assert_eq!(interned.field(&db), 1);
}

#[test]
fn multiple_databases() {
    let db1 = DatabaseImpl {
        storage: salsa::Storage::builder()
            .ingredient::<ingredients::intern>()
            .ingredient::<ingredients::MyInput>()
            .ingredient::<ingredients::MyInterned<'_>>()
            .build(),
    };

    let input = ingredients::MyInput::new(&db1, 1);
    let interned = ingredients::intern(&db1, input);
    assert_eq!(interned.field(&db1), 1);

    // Create a second database with different ingredient indices.
    let db2 = DatabaseImpl {
        storage: salsa::Storage::builder()
            .ingredient::<ingredients::track>()
            .ingredient::<ingredients::intern>()
            .ingredient::<ingredients::MyInput>()
            .ingredient::<ingredients::MyTracked<'_>>()
            .ingredient::<ingredients::MyInterned<'_>>()
            .build(),
    };

    let input = ingredients::MyInput::new(&db2, 2);
    let interned = ingredients::intern(&db2, input);
    assert_eq!(interned.field(&db2), 2);

    let input = ingredients::MyInput::new(&db2, 3);
    let tracked = ingredients::track(&db2, input);
    assert_eq!(tracked.field(&db2), 3);
}