githubmirror/libblkio

tests/get-queue.c

61 строка1 KB

Stefan Hajnoczi

LICENSE: relicense to MIT OR APACHE-2.0

31 авг 2022, 18:43

31 авг 2022, 18:43501476a

// SPDX-License-Identifier: (MIT OR Apache-2.0)
#include "util.h"

enum {
    TEST_FILE_SIZE = 2 * 1024 * 1024,
};

static char filename[] = "started-XXXXXX";

static void cleanup(void)
{
    unlink(filename);
}

int main(int argc, char **argv)
{
    struct test_opts opts;
    struct blkio *b;
    int fd;

    parse_generic_opts(&opts, argc, argv);

    ok(blkio_create(opts.driver, &b));
    assert(b);

    assert(blkio_get_queue(b, -1) == NULL);
    assert(blkio_get_queue(b, 0) == NULL);
    assert(blkio_get_queue(b, 1) == NULL);

    register_cleanup(cleanup);
    fd = create_file(filename, TEST_FILE_SIZE);

    ok(blkio_set_int(b, "fd", fd));
    ok(blkio_connect(b));

    fd = -1; /* ownership passed to libblkio */

    assert(blkio_get_queue(b, -1) == NULL);
    assert(blkio_get_queue(b, 0) == NULL);
    assert(blkio_get_queue(b, 1) == NULL);

    ok(blkio_set_int(b, "num-queues", 2));

    assert(blkio_get_queue(b, -1) == NULL);
    assert(blkio_get_queue(b, 0) == NULL);
    assert(blkio_get_queue(b, 1) == NULL);

    ok(blkio_start(b));

    assert(blkio_get_queue(b, 0) != NULL);
    assert(blkio_get_queue(b, 1) != NULL);
    assert(blkio_get_queue(b, 3) == NULL);
    assert(blkio_get_queue(b, 2) == NULL);
    assert(blkio_get_queue(b, 10) == NULL);
    assert(blkio_get_queue(b, -1) == NULL);

    blkio_destroy(&b);
    assert(!b);

    return 0;
}