nodeinfotest.c 2.7 KB
Newer Older
1
#include <config.h>
2

3 4 5
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
6
#include <unistd.h>
7 8 9 10

#include "testutils.h"
#include "internal.h"
#include "nodeinfo.h"
J
Jim Meyering 已提交
11
#include "util.h"
12

13 14 15
#ifndef __linux__

static int
16
mymain(int argc ATTRIBUTE_UNUSED, char **argv ATTRIBUTE_UNUSED)
17 18 19 20 21 22
{
    exit (77); /* means 'test skipped' for automake */
}

#else

23
static char *progname;
24
static char *abs_srcdir;
25 26 27 28 29

#define MAX_FILE 4096

extern int linuxNodeInfoCPUPopulate(virConnectPtr conn, FILE *cpuinfo, virNodeInfoPtr nodeinfo);

30
static int linuxTestCompareFiles(const char *cpuinfofile, const char *outputfile) {
31 32 33 34
    char actualData[MAX_FILE];
    char expectData[MAX_FILE];
    char *expect = &expectData[0];
    virNodeInfo nodeinfo;
35
    FILE *cpuinfo;
36 37 38 39 40 41 42 43 44 45 46 47 48 49

    if (virtTestLoadFile(outputfile, &expect, MAX_FILE) < 0)
        return -1;

    cpuinfo = fopen(cpuinfofile, "r");
    if (!cpuinfo)
        return -1;
    if (linuxNodeInfoCPUPopulate(NULL, cpuinfo, &nodeinfo) < 0) {
        fclose(cpuinfo);
        return -1;
    }
    fclose(cpuinfo);

    snprintf(actualData, MAX_FILE,
50
             "CPUs: %u, MHz: %u, Nodes: %u, Sockets: %u, Cores: %u, Threads: %u\n",
51
             nodeinfo.cpus, nodeinfo.mhz, nodeinfo.nodes, nodeinfo.sockets,
52
             nodeinfo.cores, nodeinfo.threads);
53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68

    if (STRNEQ(actualData, expectData)) {
        if (getenv("DEBUG_TESTS")) {
            printf("Expect %d '%s'\n", (int)strlen(expectData), expectData);
            printf("Actual %d '%s'\n", (int)strlen(actualData), actualData);
        }
        return -1;
    }

    return 0;
}


static int linuxTestNodeInfo(const void *data) {
    char cpuinfo[PATH_MAX];
    char output[PATH_MAX];
69 70 71 72
    snprintf(cpuinfo, PATH_MAX, "%s/nodeinfodata/linux-%s.cpuinfo",
             abs_srcdir, (const char*)data);
    snprintf(output, PATH_MAX, "%s/nodeinfodata/linux-%s.txt",
             abs_srcdir, (const char*)data);
73
    return linuxTestCompareFiles(cpuinfo, output);
74 75 76
}


77 78
static int
mymain(int argc, char **argv)
79 80 81 82 83 84 85 86 87 88 89
{
    int ret = 0;
    int i;
    const char *nodeData[] = {
        "nodeinfo-1",
        "nodeinfo-2",
        "nodeinfo-3",
        "nodeinfo-4",
        "nodeinfo-5",
        "nodeinfo-6",
    };
90
    char cwd[PATH_MAX];
91

92 93 94
    abs_srcdir = getenv("abs_srcdir");
    if (!abs_srcdir)
        abs_srcdir = getcwd(cwd, sizeof(cwd));
95

96 97 98 99
    progname = argv[0];

    if (argc > 1) {
        fprintf(stderr, "Usage: %s\n", progname);
100
        return(EXIT_FAILURE);
101 102 103 104
    }

    virInitialize();

J
Jim Meyering 已提交
105
    for (i = 0 ; i < ARRAY_CARDINALITY(nodeData); i++)
106
      if (virtTestRun(nodeData[i], 1, linuxTestNodeInfo, nodeData[i]) != 0)
107
        ret = -1;
108

109
    return(ret==0 ? EXIT_SUCCESS : EXIT_FAILURE);
110
}
111

112 113
#endif /* __linux__ */

114
VIRT_TEST_MAIN(mymain)