src/lib/test-var-expand.c

	test-var-expand.c revision 1bc529e36d36f9a525c6af023ae014ebc6cdb440
/* Copyright (c) 2009-2014 Dovecot authors, see the included COPYING file */

#include "test-lib.h"
#include "str.h"
#include "env-util.h"
#include "hostpid.h"
#include "var-expand.h"

struct var_expand_test {
    const char *in;
    const char *out;
};

struct var_get_key_range_test {
    const char *in;
    unsigned int idx, size;
};

static void test_var_expand_ranges(void)
{
    static struct var_expand_test tests[] = {
        { "%v", "value1234" },
        { "%3v", "val" },
        { "%3.2v", "ue" },
        { "%3.-2v", "ue12" },
        { "%-3.2v", "23" },
        { "%0.-1v", "value123" },
        { "%-4.-1v", "123" }
    };
    static struct var_expand_table table[] = {
        { 'v', "value1234", NULL },
        { '\0', NULL, NULL }
    };
    string_t *str = t_str_new(128);
    unsigned int i;

    test_begin("var_expand");
    for (i = 0; i < N_ELEMENTS(tests); i++) {
        str_truncate(str, 0);
        var_expand(str, tests[i].in, table);
        test_assert(strcmp(tests[i].out, str_c(str)) == 0);
    }
    test_end();
}

static void test_var_expand_builtin(void)
{
    static struct var_expand_test tests[] = {
        { "%{hostname}", NULL },
        { "%{pid}", NULL },
        { "a%{env:FOO}b", "abaRb" },
        { "%50Hv", "1f" },
        { "%50Hw", "2e" },
        { "%50Nv", "25" },
        { "%50Nw", "e" }
    };
    static struct var_expand_table table[] = {
        { 'v', "value", NULL },
        { 'w', "value2", NULL },
        { '\0', NULL, NULL }
    };
    string_t *str = t_str_new(128);
    unsigned int i;

    tests[0].out = my_hostname;
    tests[1].out = my_pid;
    env_put("FOO=baR");

    test_begin("var_expand");
    for (i = 0; i < N_ELEMENTS(tests); i++) {
        str_truncate(str, 0);
        var_expand(str, tests[i].in, table);
        test_assert(strcmp(tests[i].out, str_c(str)) == 0);
    }
    test_end();
}

static void test_var_get_key_range(void)
{
    static struct var_get_key_range_test tests[] = {
        { "", 0, 0 },
        { "{", 1, 0 },
        { "k", 0, 1 },
        { "{key}", 1, 3 },
        { "5.5Rk", 4, 1 },
        { "5.5R{key}", 5, 3 },
        { "{key", 1, 3 }
    };
    unsigned int i, idx, size;

    test_begin("var_get_key_range");
    for (i = 0; i < N_ELEMENTS(tests); i++) {
        var_get_key_range(tests[i].in, &idx, &size);
        test_assert(tests[i].idx == idx);
        test_assert(tests[i].size == size);

        if (tests[i].size == 1)
            test_assert(tests[i].in[idx] == var_get_key(tests[i].in));
    }
    test_end();
}

void test_var_expand(void)
{
    test_var_expand_ranges();
    test_var_expand_builtin();
    test_var_get_key_range();
}