forked from edubart/minicoro
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtestsuite.c
More file actions
109 lines (88 loc) · 2.94 KB
/
testsuite.c
File metadata and controls
109 lines (88 loc) · 2.94 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#define MCO_ZERO_MEMORY
#define MINICORO_IMPL
#include "minicoro.h"
#include <assert.h>
#include <stdio.h>
void coro_entry2(mco_coro* co2) {
mco_coro* co = NULL;
assert(mco_running() == co2);
assert(mco_status(co2) == MCO_RUNNING);
assert(mco_pop(co2, &co, sizeof(co)) == MCO_SUCCESS);
assert(mco_pop(co2, NULL, mco_get_bytes_stored(co2)) == MCO_SUCCESS);
assert(mco_status(co) == MCO_NORMAL);
assert(mco_get_bytes_stored(co2) == 0);
printf("hello 2\n");
assert(mco_yield(mco_running()) == MCO_SUCCESS);
printf("world! 2\n");
}
int dummy_user_data = 0;
void coro_entry(mco_coro* co) {
char buffer[128] = {0};
int ret;
mco_coro* co2;
/* Startup checks */
assert(mco_get_user_data(co) == &dummy_user_data);
assert(mco_running() == co);
assert(mco_status(co) == MCO_RUNNING);
/* Get storage 1 */
assert(mco_get_bytes_stored(co) == 6);
assert(mco_peek(co, buffer, mco_get_bytes_stored(co)) == MCO_SUCCESS);
assert(strcmp(buffer, "hello") == 0);
assert(mco_pop(co, NULL, mco_get_bytes_stored(co)) == MCO_SUCCESS);
puts(buffer);
/* Set storage 1 */
ret = 1;
assert(mco_push(co, &ret, sizeof(ret)) == MCO_SUCCESS);
/* Yield 1 */
assert(mco_yield(co) == MCO_SUCCESS);
/* Get storage 2 */
assert(mco_get_bytes_stored(co) == 7);
assert(mco_pop(co, buffer, mco_get_bytes_stored(co)) == MCO_SUCCESS);
assert(strcmp(buffer, "world!") == 0);
puts(buffer);
/* Set storage 2 */
ret = 2;
assert(mco_push(co, &ret, sizeof(ret)) == MCO_SUCCESS);
/* Nested coroutine test */
mco_desc desc = mco_desc_init(coro_entry2, 0);
assert(mco_create(&co2, &desc) == MCO_SUCCESS);
assert(mco_push(co2, &co, sizeof(co)) == MCO_SUCCESS);
assert(mco_resume(co2) == MCO_SUCCESS);
assert(mco_resume(co2) == MCO_SUCCESS);
assert(mco_get_bytes_stored(co2) == 0);
assert(mco_status(co2) == MCO_DEAD);
assert(mco_status(co) == MCO_RUNNING);
assert(mco_running() == co);
assert(mco_destroy(co2) == MCO_SUCCESS);
}
int main(void) {
mco_coro* co;
int ret = 0;
/* Create coroutine */
mco_desc desc = mco_desc_init(coro_entry, 0);
desc.user_data = &dummy_user_data;
assert(mco_create(&co, &desc) == MCO_SUCCESS);
assert(mco_status(co) == MCO_SUSPENDED);
/* Set storage 1 */
const char first_word[] = "hello";
assert(mco_push(co, first_word, sizeof(first_word)) == MCO_SUCCESS);
/* Resume 1 */
assert(mco_resume(co) == MCO_SUCCESS);
assert(mco_status(co) == MCO_SUSPENDED);
/* Get storage 1 */
assert(mco_pop(co, &ret, sizeof(ret)) == MCO_SUCCESS);
assert(ret == 1);
/* Set storage 2 */
const char second_word[] = "world!";
assert(mco_push(co, second_word, sizeof(second_word)) == MCO_SUCCESS);
/* Resume 2 */
assert(mco_resume(co) == MCO_SUCCESS);
assert(mco_status(co) == MCO_DEAD);
/* Get storage 2 */
assert(mco_pop(co, &ret, sizeof(ret)) == MCO_SUCCESS);
assert(ret == 2);
/* Destroy */
assert(mco_destroy(co) == MCO_SUCCESS);
printf("Test suite succeeded!\n");
return 0;
}