apatheia/tests/exFailure.nim

68 lines
1.4 KiB
Nim
Raw Normal View History

2024-02-21 19:51:43 -07:00
import std/os
import std/sequtils
import chronos
import chronos/threadsync
import chronos/unittest2/asynctests
import taskpools
import apatheia/queues
2024-02-21 20:07:24 -07:00
type
2024-02-21 22:12:55 -07:00
Seq*[T] = object
2024-02-21 22:11:50 -07:00
data*: ptr UncheckedArray[T]
size*: int
2024-02-21 22:12:55 -07:00
2024-02-21 20:07:24 -07:00
DataObj = ref object
2024-02-21 22:12:55 -07:00
holder: Seq[char]
2024-02-21 20:07:24 -07:00
2024-02-21 22:12:55 -07:00
template toOpenArray*[T](arr: Seq[T]): auto =
2024-02-21 22:11:50 -07:00
system.toOpenArray(arr.data, 0, arr.size)
2024-02-21 22:12:55 -07:00
proc worker(data: ptr Seq[char], queue: SignalQueue[int]) =
2024-02-21 20:26:08 -07:00
os.sleep(1_000)
2024-02-21 22:13:45 -07:00
echo "running worker: "
2024-02-21 22:09:34 -07:00
assert data[].data != nil
2024-02-21 22:04:43 -07:00
echo "worker: ", data[].toOpenArray()
discard queue.send(data[].toOpenArray().len())
2024-02-21 19:51:43 -07:00
2024-02-21 20:07:24 -07:00
proc finalizer(obj: DataObj) =
echo "FINALIZE!!"
2024-02-21 22:08:12 -07:00
obj.holder.data.dealloc()
2024-02-21 22:09:34 -07:00
obj.holder.data = nil
2024-02-21 20:07:24 -07:00
proc runTest(tp: TaskPool, queue: SignalQueue[int]) {.async.} =
## init
var obj: DataObj
new(obj, finalizer)
2024-02-21 22:08:12 -07:00
obj.holder.data = cast[ptr UncheckedArray[char]](alloc0(13))
for i, c in "hello world!":
obj.holder.data[i] = c
obj.holder.size = 12
2024-02-21 20:07:24 -07:00
echo "spawn worker"
2024-02-21 22:04:43 -07:00
tp.spawn worker(addr obj.holder, queue)
2024-02-21 20:07:24 -07:00
2024-02-21 20:26:08 -07:00
let res =
await wait(queue).wait(100.milliseconds)
2024-02-21 20:07:24 -07:00
check res.get() == 12
2024-02-21 19:51:43 -07:00
suite "async tests":
var tp = Taskpool.new(num_threads = 2) # Default to the number of hardware threads.
var queue = newSignalQueue[int]()
asyncTest "test":
2024-02-21 20:26:08 -07:00
try:
await runTest(tp, queue)
2024-02-21 22:09:34 -07:00
except AsyncTimeoutError:
2024-02-21 20:26:08 -07:00
echo "Run GC"
GC_fullCollect()
os.sleep(2_000)
echo "Done"
2024-02-21 19:51:43 -07:00