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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* blogc: A blog compiler.
* Copyright (C) 2014-2019 Rafael G. Martins <rafael@rafaelmartins.eng.br>
*
* This program can be distributed under the terms of the BSD License.
* See the file LICENSE.
*/
#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <squareball.h>
#include "../../src/blogc/rusage.h"
int
__wrap_getrusage(int who, struct rusage *usage)
{
assert_int_equal(who, RUSAGE_SELF);
int rv = mock_type(int);
if (rv == 0) {
usage->ru_utime.tv_sec = 1;
usage->ru_utime.tv_usec = 2;
usage->ru_stime.tv_sec = 3;
usage->ru_stime.tv_usec = 4;
usage->ru_maxrss = 10250;
}
return rv;
}
static void
test_rusage_get(void **state)
{
will_return(__wrap_getrusage, -1);
assert_null(blogc_rusage_get());
will_return(__wrap_getrusage, 0);
blogc_rusage_t *r = blogc_rusage_get();
assert_non_null(r);
assert_int_equal(r->cpu_time, 4000006);
assert_int_equal(r->memory, 10250);
free(r);
}
static void
test_rusage_format_cpu_time(void **state)
{
char *f = blogc_rusage_format_cpu_time(0);
assert_string_equal(f, "0us");
free(f);
f = blogc_rusage_format_cpu_time(123);
assert_string_equal(f, "123us");
free(f);
f = blogc_rusage_format_cpu_time(1234);
assert_string_equal(f, "1.234ms");
free(f);
f = blogc_rusage_format_cpu_time(3000);
assert_string_equal(f, "3ms");
free(f);
f = blogc_rusage_format_cpu_time(12345678);
assert_string_equal(f, "12.346s");
free(f);
}
static void
test_rusage_format_memory(void **state)
{
char *f = blogc_rusage_format_memory(0);
assert_string_equal(f, "0KB");
free(f);
f = blogc_rusage_format_memory(123);
assert_string_equal(f, "123KB");
free(f);
f = blogc_rusage_format_memory(1234);
assert_string_equal(f, "1.205MB");
free(f);
f = blogc_rusage_format_memory(12345678);
assert_string_equal(f, "11.774GB");
free(f);
}
static void
test_rusage_inject(void **state)
{
sb_trie_t *t = sb_trie_new(free);
will_return(__wrap_getrusage, -1);
blogc_rusage_inject(t);
assert_int_equal(sb_trie_size(t), 0);
will_return(__wrap_getrusage, 0);
blogc_rusage_inject(t);
assert_int_equal(sb_trie_size(t), 2);
assert_string_equal(sb_trie_lookup(t, "BLOGC_RUSAGE_CPU_TIME"), "4.000s");
assert_string_equal(sb_trie_lookup(t, "BLOGC_RUSAGE_MEMORY"), "10.010MB");
sb_trie_free(t);
}
int
main(void)
{
const UnitTest tests[] = {
unit_test(test_rusage_get),
unit_test(test_rusage_format_cpu_time),
unit_test(test_rusage_format_memory),
unit_test(test_rusage_inject),
};
return run_tests(tests);
}
|
