nidus-core 1.0.4

use std::{
    any::type_name,
    sync::{
        Arc, Barrier,
        atomic::{AtomicUsize, Ordering},
    },
    thread,
    time::Duration,
};

use nidus_core::{Container, Inject, NidusError, ProviderLifetime};

#[derive(Debug, PartialEq, Eq)]
struct Database(&'static str);

#[derive(Debug)]
struct UsersRepository {
    database: Inject<Database>,
}

#[derive(Debug)]
struct SelfReferential;

#[derive(Debug)]
struct CircularA {
    _b: Inject<CircularB>,
}

#[derive(Debug)]
struct CircularB {
    _a: Inject<CircularA>,
}

#[test]
fn request_factories_reuse_within_scope_but_not_across_scopes() {
    let calls = Arc::new(AtomicUsize::new(0));
    let mut container = Container::new();
    container
        .register_request::<Database, _>({
            let calls = Arc::clone(&calls);
            move |_container| {
                let call = calls.fetch_add(1, Ordering::SeqCst);
                Ok(Database(if call == 0 { "first" } else { "second" }))
            }
        })
        .unwrap();

    let first_scope = container.request_scope();
    let first = first_scope.resolve::<Database>().unwrap();
    let first_again = first_scope.resolve::<Database>().unwrap();
    let second_scope = container.request_scope();
    let second = second_scope.resolve::<Database>().unwrap();

    assert!(Arc::ptr_eq(&first, &first_again));
    assert!(!Arc::ptr_eq(&first, &second));
    assert_eq!(first.0, "first");
    assert_eq!(second.0, "second");
}

#[test]
fn request_factory_runs_once_under_concurrent_first_resolution_in_scope() {
    let calls = Arc::new(AtomicUsize::new(0));
    let mut container = Container::new();
    container
        .register_request::<Database, _>({
            let calls = Arc::clone(&calls);
            move |_container| {
                calls.fetch_add(1, Ordering::SeqCst);
                thread::sleep(Duration::from_millis(25));
                Ok(Database("request"))
            }
        })
        .unwrap();
    let scope = Arc::new(nidus_core::RequestScope::from_shared_container(Arc::new(
        container,
    )));
    let ready = Arc::new(Barrier::new(8));

    let handles = (0..8)
        .map(|_| {
            let scope = Arc::clone(&scope);
            let ready = Arc::clone(&ready);
            thread::spawn(move || {
                ready.wait();
                scope.resolve::<Database>().unwrap()
            })
        })
        .collect::<Vec<_>>();

    let instances = handles
        .into_iter()
        .map(|handle| handle.join().unwrap())
        .collect::<Vec<_>>();

    for instance in &instances[1..] {
        assert!(Arc::ptr_eq(&instances[0], instance));
    }
    assert_eq!(calls.load(Ordering::SeqCst), 1);
}

#[test]
fn request_scope_reports_self_circular_resolution() {
    let mut container = Container::new();
    container
        .register_request_scoped::<SelfReferential, _>(|scope| {
            let _self_reference = scope.inject::<SelfReferential>()?;
            Ok(SelfReferential)
        })
        .unwrap();
    let scope = container.request_scope();

    let error = scope.resolve::<SelfReferential>().unwrap_err();

    assert_circular_provider_error(error, type_name::<SelfReferential>());
}

#[test]
fn request_scope_reports_indirect_circular_resolution() {
    let mut container = Container::new();
    container
        .register_request_scoped::<CircularA, _>(|scope| {
            Ok(CircularA {
                _b: scope.inject::<CircularB>()?,
            })
        })
        .unwrap();
    container
        .register_request_scoped::<CircularB, _>(|scope| {
            Ok(CircularB {
                _a: scope.inject::<CircularA>()?,
            })
        })
        .unwrap();
    let scope = container.request_scope();

    let error = scope.resolve::<CircularA>().unwrap_err();

    assert_circular_provider_error(error, type_name::<CircularA>());
}

#[test]
fn request_scope_resolves_scoped_wrapper() {
    let mut container = Container::new();
    container
        .register_request::<Database, _>(|_container| Ok(Database("request")))
        .unwrap();
    let scope = container.request_scope();

    let database = scope.scoped::<Database>().unwrap();

    assert_eq!(database.0, "request");
}

#[test]
fn request_scope_optional_reuses_request_scoped_instances() {
    let calls = Arc::new(AtomicUsize::new(0));
    let mut container = Container::new();
    container
        .register_factory::<Database, _>(ProviderLifetime::Request, {
            let calls = Arc::clone(&calls);
            move |_container| {
                calls.fetch_add(1, Ordering::SeqCst);
                Ok(Database("request"))
            }
        })
        .unwrap();
    let scope = container.request_scope();

    let first = scope.optional::<Database>().unwrap().into_option().unwrap();
    let second = scope.optional::<Database>().unwrap().into_option().unwrap();

    assert_eq!(calls.load(Ordering::SeqCst), 1);
    assert!(Arc::ptr_eq(&first.into_inner(), &second.into_inner()));
}

#[test]
fn request_scoped_factories_resolve_dependencies_through_same_scope() {
    let calls = Arc::new(AtomicUsize::new(0));
    let mut container = Container::new();
    container
        .register_request::<Database, _>({
            let calls = Arc::clone(&calls);
            move |_container| {
                calls.fetch_add(1, Ordering::SeqCst);
                Ok(Database("request"))
            }
        })
        .unwrap();
    container
        .register_request_scoped::<UsersRepository, _>(|scope| {
            Ok(UsersRepository {
                database: scope.inject::<Database>()?,
            })
        })
        .unwrap();

    let scope = container.request_scope();
    let repository = scope.resolve::<UsersRepository>().unwrap();
    let database = scope.resolve::<Database>().unwrap();

    assert_eq!(calls.load(Ordering::SeqCst), 1);
    assert!(Arc::ptr_eq(
        &repository.database.clone().into_inner(),
        &database
    ));
}

#[test]
fn request_factories_require_explicit_request_scope() {
    let mut container = Container::new();
    container
        .register_factory::<Database, _>(ProviderLifetime::Request, |_container| {
            Ok(Database("request"))
        })
        .unwrap();

    let error = container.resolve::<Database>().unwrap_err();

    assert!(matches!(error, NidusError::RequestScopeRequired { .. }));
    assert!(error.to_string().contains("Database"));
}

fn assert_circular_provider_error(error: NidusError, expected_type_name: &'static str) {
    match error {
        NidusError::ProviderFactory { source, .. } => {
            assert_circular_provider_error(*source, expected_type_name);
        }
        NidusError::CircularProviderResolution { type_name } => {
            assert_eq!(type_name, expected_type_name);
        }
        error => panic!("expected circular provider resolution error, got {error:?}"),
    }
}