varch/test/test_set.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#if defined(TEST_TARGET_set)
#include <varch/command.h>
#include <varch/set.h>
#else  
#include "init.h"
#include "command.h"
#include "set.h"
#endif

static void test_it(void)
{
    set_t set = set(int);
    int i, index;
    void *data;

    i = -100; set_insert(set, i, &i);
    i = 1024; set_insert(set, i, &i);
    i = 0; set_insert(set, i, &i);
    i = 7; set_insert(set, i, &i);
    i = -2; set_insert(set, i, &i);
    i = -2; set_insert(set, i, &i);

    set_at(set, int, 3) = 100;

    set_it_init(set, SET_HEAD);
    i = set_size(set);
    while (i--)
    {
        data = set_it_get(set, &index);
        printf("set[%d] = %d\r\n", index, *(int *)data);
    }

    _set(set);
}

static void test_base(void) 
{
    set_t set = set(int);
    int i;

    for (i = 0; i < 100; i++)
    {
        set_insert(set, i, &i);
    }
    printf("size = %d, data size = %d\r\n", set_size(set), set_dsize(set));

    i = -100; set_insert(set, i, &i);
    i = 1024; set_insert(set, i, &i);

    printf("set[6] = %d\r\n", set_at(set, int, 6));
    printf("set[-100] = %d\r\n", set_at(set, int, -100));
    printf("set[1024] = %d\r\n", set_at(set, int, 1024));

    set_at(set, int, 6) = 11111;
    printf("set[6] = %d\r\n", set_at(set, int, 6));

    _set(set);
}

static void usage(void)
{
    printf(
"Usage: set [opt] [arg]\n"
"\n"
"options:\n"
"    -h                  Print help\n"
"    -v                  Print version\n"
"\n"
"argument:\n"
"    <read>              Test default read function\n"
"    <write>             Test default write function\n"
    );
}

static int test(int argc, char *argv[])
{
    /* reset getopt */
    command_opt_init();

    while (1)
    {
        int opt = command_getopt(argc, argv, "hv");
        if (opt == -1) break;

        switch (opt) 
        {
        case 'v' :
            printf("set version %d.%d.%d\r\n", SET_V_MAJOR, SET_V_MINOR, SET_V_PATCH);
            return 0;
        case '?':
            printf("Unknown option `%c`\r\n", command_optopt);
            return -1;
        case 'h' : 
        default:
            usage();
            return 0;
        }
    }

    // test_base();
    test_it();
    
    return 0;
}

#if defined(TEST_TARGET_set)
int main(int argc, char *argv[])
{
    return test(argc, argv);
}
#else 
void test_set(void)
{
    command_export("set", test);
}
init_export_app(test_set);
#endif