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
|
#define UNIT_TEST
#include <math.h>
#include <stdio.h>
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-function"
#pragma GCC diagnostic ignored "-Wunused-variable"
#include "../psi-notify.c" /* put it in the same translation unit */
#pragma GCC diagnostic pop
#define COL_NORMAL "\x1B[0m"
#define COL_GREEN "\x1B[32m"
#define COL_RED "\x1B[31m"
#define t_assert(test) \
do { \
if (!(test)) { \
printf(" %s[FAIL:%d] %s%s\n", COL_RED, __LINE__, #test, \
COL_NORMAL); \
return false; \
} \
printf(" %s[PASS:%d] %s%s\n", COL_GREEN, __LINE__, #test, \
COL_NORMAL); \
} while (0)
#define t_run(test) \
do { \
printf("%s:\n", #test); \
bool ret = test(); \
printf("\n"); \
if (!ret) { \
return ret; \
} \
} while (0)
static bool test_config_parse_basic(void) {
const char *raw_config = "# Comment at beginning of line\n"
" # Comment with some spaces\n\n \n"
"update 3 # c\n"
"threshold cpu some avg10 50.00 #c\n"
"threshold memory full avg60 10.00 #c\n"
"threshold io full avg300 100.00 #c\n"
"log_pressures yes #c";
FILE *f = fmemopen((void *)raw_config, strlen(raw_config), "r");
memset(&cfg, 0, sizeof(Config));
config_update_from_file(&f);
t_assert(cfg.update_interval == 3);
t_assert(cfg.log_pressures);
t_assert(cfg.cpu.thresholds.avg10.some == 50.00);
t_assert(cfg.memory.thresholds.avg60.full == 10.00);
t_assert(cfg.io.thresholds.avg300.full == 100.00);
t_assert(isnan(cfg.cpu.thresholds.avg60.some));
return true;
}
static bool test_config_parse_init_no_file_uses_defaults(void) {
FILE *f = NULL;
memset(&cfg, 0, sizeof(Config));
config_update_from_file(&f);
t_assert(cfg.cpu.thresholds.avg10.some == 50.00);
t_assert(cfg.memory.thresholds.avg10.some == 10.00);
t_assert(cfg.io.thresholds.avg10.full == 15.00);
return true;
}
static bool test_pressure_check(void) {
FILE *psi_f;
const char *raw_psi =
"some avg10=5.00 avg60=10.02 avg300=100.00 total=453225698\n"
"full avg10=5.00 avg60=20.02 avg300=90.00 total=416296780\n";
FILE *config_f = fmemopen((void *)"", strlen(""), "r");
config_init(&config_f);
/* Threshold are too low, but we're within 5 pct boundary. */
psi_f = fmemopen((void *)raw_psi, strlen(raw_psi), "r");
cfg.memory.thresholds.avg300.full = 90.00;
t_assert(pressure_check(&cfg.memory, psi_f) == A_STABILISING);
/* Add another threshold which trips. */
psi_f = fmemopen((void *)raw_psi, strlen(raw_psi), "r");
cfg.memory.thresholds.avg300.full = 9.99;
t_assert(pressure_check(&cfg.memory, psi_f) == A_ACTIVE);
/* Thresholds are too low. */
psi_f = fmemopen((void *)raw_psi, strlen(raw_psi), "r");
cfg.memory.thresholds.avg300.full = 95.00;
t_assert(pressure_check(&cfg.memory, psi_f) == A_INACTIVE);
return true;
}
static bool run_tests(void) {
t_run(test_config_parse_basic);
t_run(test_config_parse_init_no_file_uses_defaults);
t_run(test_pressure_check);
return true;
}
int main(void) { return run_tests() ? EXIT_SUCCESS : EXIT_FAILURE; }
|
