import context
import time
import x.async as xasync

fn test_every_runs_at_least_one_iteration() {
	parent_ctx, cancel := xasync.with_cancel()
	ran := chan bool{cap: 1}
	result := chan string{cap: 1}
	worker := spawn fn [parent_ctx, ran, result] () {
		xasync.every(parent_ctx, 5 * time.millisecond, fn [ran] (mut ctx context.Context) ! {
			_ = ctx
			ran <- true
		}) or {
			result <- err.msg()
			return
		}
		result <- 'ok'
	}()

	select {
		did_run := <-ran {
			assert did_run
		}
		1 * time.second {
			assert false, 'periodic job did not run'
		}
	}
	cancel()
	select {
		msg := <-result {
			assert msg == 'context canceled'
		}
		1 * time.second {
			assert false, 'every did not stop after cancellation'
		}
	}
	worker.wait()
}

fn test_every_stops_on_cancellation() {
	parent_ctx, cancel := xasync.with_cancel()
	entered := chan bool{cap: 1}
	result := chan string{cap: 1}
	worker := spawn fn [parent_ctx, entered, result] () {
		entered <- true
		xasync.every(parent_ctx, 1 * time.second, fn (mut ctx context.Context) ! {
			_ = ctx
		}) or {
			result <- err.msg()
			return
		}
		result <- 'ok'
	}()

	select {
		did_enter := <-entered {
			assert did_enter
		}
		1 * time.second {
			assert false, 'every worker did not start'
		}
	}
	cancel()
	select {
		msg := <-result {
			assert msg == 'context canceled'
		}
		1 * time.second {
			assert false, 'every did not stop on context cancellation'
		}
	}
	worker.wait()
}

fn test_every_returns_immediately_when_parent_is_already_canceled() {
	parent_ctx, cancel := xasync.with_cancel()
	cancel()
	xasync.every(parent_ctx, 1 * time.second, fn (mut ctx context.Context) ! {
		_ = ctx
	}) or {
		assert err.msg() == 'context canceled'
		return
	}
	assert false
}

fn test_every_returns_iteration_error() {
	parent_ctx, cancel := xasync.with_cancel()
	result := chan string{cap: 1}
	worker := spawn fn [parent_ctx, result] () {
		xasync.every(parent_ctx, 5 * time.millisecond, fn (mut ctx context.Context) ! {
			_ = ctx
			return error('periodic failed')
		}) or {
			result <- err.msg()
			return
		}
		result <- 'ok'
	}()

	select {
		msg := <-result {
			assert msg == 'periodic failed'
		}
		1 * time.second {
			assert false, 'every did not return the periodic job error'
		}
	}
	cancel()
	worker.wait()
}

fn test_every_rejects_zero_interval() {
	xasync.every(context.background(), 0 * time.millisecond, fn (mut ctx context.Context) ! {
		_ = ctx
	}) or {
		assert err.msg() == 'async: interval must be positive'
		return
	}
	assert false
}

fn test_every_rejects_negative_interval() {
	xasync.every(context.background(), -1 * time.millisecond, fn (mut ctx context.Context) ! {
		_ = ctx
	}) or {
		assert err.msg() == 'async: interval must be positive'
		return
	}
	assert false
}

fn test_every_rejects_nil_job() {
	nil_job := unsafe { xasync.JobFn(nil) }
	xasync.every(context.background(), 1 * time.second, nil_job) or {
		assert err.msg() == 'async: job function is nil'
		return
	}
	assert false
}

fn test_every_does_not_overlap_iterations() {
	parent_ctx, cancel := xasync.with_cancel()
	active := chan bool{cap: 1}
	active <- true
	entered := chan bool{cap: 2}
	release := chan bool{cap: 2}
	overlap := chan bool{cap: 1}
	result := chan string{cap: 1}
	worker := spawn fn [parent_ctx, active, entered, release, overlap, result] () {
		xasync.every(parent_ctx, 5 * time.millisecond, fn [active, entered, release, overlap] (mut ctx context.Context) ! {
			select {
				_ := <-active {}
				else {
					overlap <- true
					return error('periodic overlap')
				}
			}
			entered <- true
			done := ctx.done()
			select {
				_ := <-release {}
				_ := <-done {
					active <- true
					return ctx.err()
				}
			}
			active <- true
		}) or {
			result <- err.msg()
			return
		}
		result <- 'ok'
	}()

	wait_for_periodic_entry(entered)
	select {
		_ := <-entered {
			assert false, 'periodic iterations overlapped while first job was still running'
		}
		50 * time.millisecond {}
	}
	select {
		did_overlap := <-overlap {
			assert !did_overlap
		}
		else {}
	}

	release <- true
	wait_for_periodic_entry(entered)
	cancel()
	release <- true
	select {
		msg := <-result {
			assert msg == 'context canceled'
		}
		1 * time.second {
			assert false, 'every did not stop after non-overlap test cancellation'
		}
	}
	worker.wait()
}

fn wait_for_periodic_entry(entered chan bool) {
	select {
		did_enter := <-entered {
			assert did_enter
		}
		1 * time.second {
			assert false, 'periodic job did not enter'
		}
	}
}