#include "test.h"
#include "template.h"

START_TESTS 

TEST(text_only) {
    char *input = "Hello world.";
    char *output = template(input, NULL);
    assert_str(output, "Hello world.");
    free(output);
}

TEST(expr_number) {
    char *input = "Hello {{ 5 }}.";
    char *output = template(input, NULL);
    assert_str(output, "Hello 5.");
    free(output);
}

TEST(expr_string) {
    char *input = "Hello {{ \"world\" }}.";
    char *output = template(input, NULL);
    assert_str(output, "Hello world.");
    free(output);
}

TEST(expr_symbol) {
    char *input = "Hello {{name}}.";
    struct hashmap *ctx = hashmap_new();
    hashmap_insert(ctx, "name", "world");
    char *output = template(input, ctx);
    assert_str(output, "Hello world.");
    hashmap_free(ctx);
    free(output);
}

TEST(var_whitespace) {
    char *input = "Hello \n{{-name -}}\n.";
    struct hashmap *ctx = hashmap_new();
    hashmap_insert(ctx, "name", "world");
    char *output = template(input, ctx);
    assert_str(output, "Helloworld.");
    hashmap_free(ctx);
    free(output);
}

TEST(multiline) {
    char *input = "Hello {{name}}.\nL2";
    struct hashmap *ctx = hashmap_new();
    hashmap_insert(ctx, "name", "world");
    char *output = template(input, ctx);
    assert_str(output, "Hello world.\nL2");
    hashmap_free(ctx);
    free(output);
}

TEST(for_block) {
    char *input = "{% for n in numbers %}{{ n }}, {% endfor %}";
    struct hashmap *ctx = hashmap_new();

    struct vector *numbers = vector_new(3);
    vector_push(numbers, "1");
    vector_push(numbers, "2");
    vector_push(numbers, "3");
    hashmap_insert(ctx, "numbers", numbers);

    char *output = template(input, ctx);
    assert_str(output, "1, 2, 3, ");
    vector_free(numbers);
    hashmap_free(ctx);
    free(output);
}

TEST(var_dot_notation) {
    char *input = "Hello {{user.name}}!";
     struct hashmap *user = hashmap_new();
    hashmap_insert(user, "name", "Danny");

    struct hashmap *ctx = hashmap_new();
    hashmap_insert(ctx, "user", user);
    
    char *output = template(input, ctx);
    assert_str(output, "Hello Danny!");
    hashmap_free(ctx);
    free(output);
}

TEST(comments) {
    char *input = "Hello {# comment here #} world.";
    char *output = template(input, NULL);
    assert_str(output, "Hello  world.");
    free(output);
}

END_TESTS