module async

import context
import time

fn test_async_context_cancel_closes_done_and_sets_error() {
    mut ctx, cancel := new_cancel_context(context.background())
    cancel()
    done := ctx.done()
    select {
        _ := <-done {
            assert ctx.err().msg() == context_canceled
        }
        1 * time.second {
            assert false, 'cancel did not close AsyncContext.done()'
        }
    }
    cancel()
    assert ctx.err().msg() == context_canceled
}

fn test_async_context_timeout_closes_done_and_sets_deadline_error() {
    mut ctx, cancel := new_timeout_context(context.background(), 20 * time.millisecond)
    defer {
        cancel()
    }
    done := ctx.done()
    select {
        _ := <-done {
            assert ctx.err().msg() == context_deadline_exceeded
        }
        1 * time.second {
            assert false, 'timeout did not close AsyncContext.done()'
        }
    }
}

fn test_async_context_parent_cancel_propagates_to_child() {
    parent, parent_cancel := new_cancel_context(context.background())
    mut child, child_cancel := new_cancel_context(context.Context(parent))
    defer {
        child_cancel()
    }
    parent_cancel()
    done := child.done()
    select {
        _ := <-done {
            assert child.err().msg() == context_canceled
        }
        1 * time.second {
            assert false, 'parent cancellation did not reach child AsyncContext'
        }
    }
}