src/test-zebra.c


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

#include "zebra.h"
#include <stdio.h>
#include "Record_Info.h"

typedef int testFunction(char *err);

int test_zdab(char *err)
{
    int rv;
    bank b;
    int nzdabs = 0;

    zebraFile *z = zebra_open("Muons.zdab");

    while (1) {
        rv = next_bank(z, &b);

        if (rv == -1) {
            sprintf(err, "error getting bank: %s\n", zebra_err);
            goto err;
        } else if (rv == 1) {
            break;
        }

        if (b.name == ZDAB_RECORD) nzdabs += 1;
    }

    if (nzdabs != 5) {
        sprintf(err, "expected %i zdab records but got %i", 5, nzdabs);
        goto err;
    }

    zebra_close(z);

    return 0;

err:
    zebra_close(z);

    return -1;
}

struct tests {
    testFunction *test;
    char *name;
} tests[] = {
    {test_zdab, "test_zdab"}
};

int main(int argc, char **argv)
{
    int i;
    char err[256];
    int retval = 0;
    struct tests test;

    for (i = 0; i < sizeof(tests)/sizeof(struct tests); i++) {
        test = tests[i];

        if (!test.test(err)) {
            printf("[\033[92mok\033[0m] %s\n", test.name);
        } else {
            printf("[\033[91mfail\033[0m] %s: %s\n", test.name, err);
            retval = 1;
        }
    }

    return retval;
}