qubit-atomic 0.10.2

/*******************************************************************************
 *
 *    Copyright (c) 2025 - 2026.
 *    Haixing Hu, Qubit Co. Ltd.
 *
 *    All rights reserved.
 *
 ******************************************************************************/

use qubit_atomic::Atomic;
use std::sync::Arc;
use std::thread;

#[test]
fn test_new() {
    let atomic = Atomic::<bool>::new(true);
    assert!(atomic.load());
    let atomic = Atomic::<bool>::new(false);
    assert!(!atomic.load());
}

#[test]
fn test_default() {
    let atomic = Atomic::<bool>::default();
    assert!(!atomic.load());
}

#[test]
fn test_from() {
    let atomic = Atomic::<bool>::from(true);
    assert!(atomic.load());
}

#[test]
fn test_get_set() {
    let atomic = Atomic::<bool>::new(false);
    atomic.store(true);
    assert!(atomic.load());
    atomic.store(false);
    assert!(!atomic.load());
}

#[test]
fn test_swap() {
    let atomic = Atomic::<bool>::new(false);
    let old = atomic.swap(true);
    assert!(!old);
    assert!(atomic.load());
}

#[test]
fn test_compare_and_set_success() {
    let atomic = Atomic::<bool>::new(false);
    assert!(atomic.compare_set(false, true).is_ok());
    assert!(atomic.load());
}

#[test]
fn test_compare_and_set_failure() {
    let atomic = Atomic::<bool>::new(false);
    match atomic.compare_set(true, false) {
        Ok(_) => panic!("Should fail"),
        Err(actual) => assert!(!actual),
    }
    assert!(!atomic.load());
}

#[test]
fn test_compare_and_exchange() {
    let atomic = Atomic::<bool>::new(false);
    let prev = atomic.compare_and_exchange(false, true);
    assert!(!prev);
    assert!(atomic.load());

    let prev = atomic.compare_and_exchange(false, false);
    assert!(prev);
    assert!(atomic.load());
}

#[test]
fn test_get_and_set() {
    let atomic = Atomic::<bool>::new(false);
    let old = atomic.fetch_set();
    assert!(!old);
    assert!(atomic.load());
}

#[test]
fn test_get_and_clear() {
    let atomic = Atomic::<bool>::new(true);
    let old = atomic.fetch_clear();
    assert!(old);
    assert!(!atomic.load());
}

#[test]
fn test_get_and_negate() {
    let atomic = Atomic::<bool>::new(false);
    assert!(!atomic.fetch_not());
    assert!(atomic.load());
    assert!(atomic.fetch_not());
    assert!(!atomic.load());
}

#[test]
fn test_get_and_logical_and() {
    let atomic = Atomic::<bool>::new(true);
    assert!(atomic.fetch_and(false));
    assert!(!atomic.load());

    atomic.store(true);
    assert!(atomic.fetch_and(true));
    assert!(atomic.load());
}

#[test]
fn test_get_and_logical_or() {
    let atomic = Atomic::<bool>::new(false);
    assert!(!atomic.fetch_or(true));
    assert!(atomic.load());

    atomic.store(false);
    assert!(!atomic.fetch_or(false));
    assert!(!atomic.load());
}

#[test]
fn test_get_and_logical_xor() {
    let atomic = Atomic::<bool>::new(false);
    assert!(!atomic.fetch_xor(true));
    assert!(atomic.load());

    assert!(atomic.fetch_xor(true));
    assert!(!atomic.load());
}

#[test]
fn test_compare_and_set_if_false() {
    let atomic = Atomic::<bool>::new(false);
    assert!(atomic.set_if_false(true).is_ok());
    assert!(atomic.load());

    assert!(atomic.set_if_false(false).is_err());
    assert!(atomic.load());
}

#[test]
fn test_compare_and_set_if_true() {
    let atomic = Atomic::<bool>::new(true);
    assert!(atomic.set_if_true(false).is_ok());
    assert!(!atomic.load());

    assert!(atomic.set_if_true(true).is_err());
    assert!(!atomic.load());
}

#[test]
fn test_concurrent_toggle() {
    let flag = Arc::new(Atomic::<bool>::new(false));
    let mut handles = vec![];

    for _ in 0..10 {
        let flag = flag.clone();
        let handle = thread::spawn(move || {
            for _ in 0..100 {
                flag.fetch_not();
            }
        });
        handles.push(handle);
    }

    for handle in handles {
        handle.join().unwrap();
    }

    // After 1000 toggles, should be false (even number)
    assert!(!flag.load());
}

#[test]
fn test_concurrent_set_once() {
    let flag = Arc::new(Atomic::<bool>::new(false));
    let mut handles = vec![];
    let success_count = Arc::new(std::sync::atomic::AtomicUsize::new(0));

    for _ in 0..10 {
        let flag = flag.clone();
        let success_count = success_count.clone();
        let handle = thread::spawn(move || {
            if flag.set_if_false(true).is_ok() {
                success_count.fetch_add(1, std::sync::atomic::Ordering::Relaxed);
            }
        });
        handles.push(handle);
    }

    for handle in handles {
        handle.join().unwrap();
    }

    // Only one thread should succeed
    assert!(flag.load());
    assert_eq!(success_count.load(std::sync::atomic::Ordering::Relaxed), 1);
}

#[test]
fn test_trait_atomic() {
    fn test_atomic(atomic: &Atomic<bool>) {
        atomic.store(true);
        assert!(atomic.load());
        let old = atomic.swap(false);
        assert!(old);
    }

    let atomic = Atomic::<bool>::new(false);
    test_atomic(&atomic);
}

#[test]
fn test_debug_display() {
    let atomic = Atomic::<bool>::new(true);
    let debug_str = format!("{:?}", atomic);
    assert!(debug_str.contains("true"));
    let display_str = format!("{}", atomic);
    assert_eq!(display_str, "true");
}

#[test]
fn test_compare_and_set_weak_success() {
    let atomic = Atomic::<bool>::new(false);
    assert!(atomic.compare_set_weak(false, true).is_ok());
    assert!(atomic.load());
}

#[test]
fn test_compare_and_set_weak_failure() {
    let atomic = Atomic::<bool>::new(false);
    match atomic.compare_set_weak(true, false) {
        Ok(_) => panic!("Should fail"),
        Err(actual) => assert!(!actual),
    }
    assert!(!atomic.load());
}

#[test]
fn test_compare_and_exchange_weak() {
    let atomic = Atomic::<bool>::new(false);
    let prev = atomic.compare_and_exchange_weak(false, true);
    assert!(!prev);
    assert!(atomic.load());

    let prev = atomic.compare_and_exchange_weak(false, false);
    assert!(prev);
    assert!(atomic.load());
}

#[test]
fn test_inner() {
    use std::sync::atomic::Ordering;

    let atomic = Atomic::<bool>::new(false);
    atomic.inner().store(true, Ordering::Relaxed);
    assert!(atomic.inner().load(Ordering::Relaxed));

    atomic.inner().store(false, Ordering::Release);
    assert!(!atomic.inner().load(Ordering::Acquire));
}

#[test]
fn test_get_and_set_already_true() {
    let atomic = Atomic::<bool>::new(true);
    let old = atomic.fetch_set();
    assert!(old);
    assert!(atomic.load());
}

#[test]
fn test_get_and_clear_already_false() {
    let atomic = Atomic::<bool>::new(false);
    let old = atomic.fetch_clear();
    assert!(!old);
    assert!(!atomic.load());
}

#[test]
fn test_get_and_logical_and_both_false() {
    let atomic = Atomic::<bool>::new(false);
    assert!(!atomic.fetch_and(false));
    assert!(!atomic.load());
}

#[test]
fn test_get_and_logical_and_false_true() {
    let atomic = Atomic::<bool>::new(false);
    assert!(!atomic.fetch_and(true));
    assert!(!atomic.load());
}

#[test]
fn test_get_and_logical_or_both_true() {
    let atomic = Atomic::<bool>::new(true);
    assert!(atomic.fetch_or(true));
    assert!(atomic.load());
}

#[test]
fn test_get_and_logical_or_true_false() {
    let atomic = Atomic::<bool>::new(true);
    assert!(atomic.fetch_or(false));
    assert!(atomic.load());
}

#[test]
fn test_get_and_logical_xor_both_false() {
    let atomic = Atomic::<bool>::new(false);
    assert!(!atomic.fetch_xor(false));
    assert!(!atomic.load());
}

#[test]
fn test_get_and_logical_xor_both_true() {
    let atomic = Atomic::<bool>::new(true);
    assert!(atomic.fetch_xor(true));
    assert!(!atomic.load());
}

#[test]
fn test_get_and_logical_xor_true_false() {
    let atomic = Atomic::<bool>::new(true);
    assert!(atomic.fetch_xor(false));
    assert!(atomic.load());
}

#[test]
fn test_compare_and_set_if_false_already_true() {
    let atomic = Atomic::<bool>::new(true);
    match atomic.set_if_false(false) {
        Ok(_) => panic!("Should fail"),
        Err(actual) => assert!(actual),
    }
    assert!(atomic.load());
}

#[test]
fn test_compare_and_set_if_true_already_false() {
    let atomic = Atomic::<bool>::new(false);
    match atomic.set_if_true(true) {
        Ok(_) => panic!("Should fail"),
        Err(actual) => assert!(!actual),
    }
    assert!(!atomic.load());
}

#[test]
fn test_trait_atomic_compare_and_set() {
    fn test_atomic(atomic: &Atomic<bool>) {
        assert!(atomic.compare_set(false, true).is_ok());
        assert!(atomic.load());
        assert!(atomic.compare_set(false, false).is_err());
    }

    let atomic = Atomic::<bool>::new(false);
    test_atomic(&atomic);
}

#[test]
fn test_trait_atomic_compare_and_exchange() {
    fn test_atomic(atomic: &Atomic<bool>) {
        let prev = atomic.compare_and_exchange(false, true);
        assert!(!prev);
        assert!(atomic.load());
    }

    let atomic = Atomic::<bool>::new(false);
    test_atomic(&atomic);
}

#[test]
fn test_trait_atomic_compare_set_weak() {
    fn test_atomic(atomic: &Atomic<bool>) {
        atomic.store(false);
        assert!(atomic.compare_set_weak(false, true).is_ok());
        assert!(atomic.load());
    }

    let atomic = Atomic::<bool>::new(false);
    test_atomic(&atomic);
}

#[test]
fn test_trait_atomic_compare_exchange_weak() {
    fn test_atomic(atomic: &Atomic<bool>) {
        atomic.store(false);
        let prev = atomic.compare_and_exchange_weak(false, true);
        assert!(!prev);
        assert!(atomic.load());
    }

    let atomic = Atomic::<bool>::new(false);
    test_atomic(&atomic);
}

#[test]
fn test_trait_atomic_fetch_update() {
    fn test_atomic(atomic: &Atomic<bool>) {
        atomic.store(false);
        let old = atomic.fetch_update(|x| !x);
        assert!(!old);
        assert!(atomic.load());
    }

    let atomic = Atomic::<bool>::new(false);
    test_atomic(&atomic);
}

#[test]
fn test_concurrent_compare_and_set_weak() {
    let flag = Arc::new(Atomic::<bool>::new(false));
    let mut handles = vec![];
    let success_count = Arc::new(std::sync::atomic::AtomicUsize::new(0));

    for _ in 0..10 {
        let flag = flag.clone();
        let success_count = success_count.clone();
        let handle = thread::spawn(move || {
            let mut current = flag.load();
            loop {
                match flag.compare_set_weak(current, true) {
                    Ok(_) => {
                        success_count.fetch_add(1, std::sync::atomic::Ordering::Relaxed);
                        break;
                    }
                    Err(actual) => current = actual,
                }
            }
        });
        handles.push(handle);
    }

    for handle in handles {
        handle.join().unwrap();
    }

    assert!(flag.load());
    // At least one thread should succeed
    assert!(success_count.load(std::sync::atomic::Ordering::Relaxed) >= 1);
}