/*
 * blogc: A blog compiler.
 * Copyright (C) 2015 Rafael G. Martins <rafael@rafaelmartins.eng.br>
 *
 * This program can be distributed under the terms of the BSD License.
 * See the file COPYING.
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif /* HAVE_CONFIG_H */

#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
#include <string.h>
#include <stdio.h>
#include "../src/template-parser.h"
#include "../src/loader.h"
#include "../src/utils/utils.h"


static void
test_get_filename(void **state)
{
    char *f = blogc_get_filename("/home/foo/asd/bola.txt");
    assert_string_equal(f, "bola");
    free(f);
    f = blogc_get_filename("/home/foo/asd/bola.guda.txt");
    assert_string_equal(f, "bola.guda");
    free(f);
    f = blogc_get_filename("bola.txt");
    assert_string_equal(f, "bola");
    free(f);
    f = blogc_get_filename("bola.guda.txt");
    assert_string_equal(f, "bola.guda");
    free(f);
    f = blogc_get_filename("/home/foo/asd/bola");
    assert_string_equal(f, "bola");
    free(f);
    f = blogc_get_filename("bola");
    assert_string_equal(f, "bola");
    free(f);
    f = blogc_get_filename("");
    assert_null(f);
    free(f);
    f = blogc_get_filename(NULL);
    assert_null(f);
    free(f);
}


char*
__wrap_blogc_file_get_contents(const char *path, size_t *len, blogc_error_t **err)
{
    assert_null(*err);
    const char *_path = mock_type(const char*);
    if (_path != NULL)
        assert_string_equal(path, _path);
    char *rv = mock_type(char*);
    *len = 0;
    if (rv != NULL)
        *len = strlen(rv);
    return rv;
}


int
__wrap_blogc_fprintf(FILE *stream, const char *format, ...)
{
    assert_true(stream == mock_type(FILE*));
    assert_string_equal(format, mock_type(const char*));
    return strlen(format);
}


static void
test_template_parse_from_file(void **state)
{
    blogc_error_t *err = NULL;
    will_return(__wrap_blogc_file_get_contents, "bola");
    will_return(__wrap_blogc_file_get_contents, b_strdup("{{ BOLA }}\n"));
    b_slist_t *l = blogc_template_parse_from_file("bola", &err);
    assert_null(err);
    assert_non_null(l);
    assert_int_equal(b_slist_length(l), 2);
    blogc_template_free_stmts(l);
}


static void
test_template_parse_from_file_null(void **state)
{
    blogc_error_t *err = NULL;
    will_return(__wrap_blogc_file_get_contents, "bola");
    will_return(__wrap_blogc_file_get_contents, NULL);
    b_slist_t *l = blogc_template_parse_from_file("bola", &err);
    assert_null(err);
    assert_null(l);
}


static void
test_source_parse_from_file(void **state)
{
    blogc_error_t *err = NULL;
    will_return(__wrap_blogc_file_get_contents, "bola.txt");
    will_return(__wrap_blogc_file_get_contents, b_strdup(
        "ASD: 123\n"
        "--------\n"
        "bola"));
    b_trie_t *t = blogc_source_parse_from_file("bola.txt", &err);
    assert_null(err);
    assert_non_null(t);
    assert_int_equal(b_trie_size(t), 4);
    assert_string_equal(b_trie_lookup(t, "ASD"), "123");
    assert_string_equal(b_trie_lookup(t, "FILENAME"), "bola");
    assert_string_equal(b_trie_lookup(t, "CONTENT"), "<p>bola</p>\n");
    assert_string_equal(b_trie_lookup(t, "RAW_CONTENT"), "bola");
    b_trie_free(t);
}


static void
test_source_parse_from_file_null(void **state)
{
    blogc_error_t *err = NULL;
    will_return(__wrap_blogc_file_get_contents, "bola.txt");
    will_return(__wrap_blogc_file_get_contents, NULL);
    b_trie_t *t = blogc_source_parse_from_file("bola.txt", &err);
    assert_null(err);
    assert_null(t);
}


static void
test_source_parse_from_files(void **state)
{
    will_return(__wrap_blogc_file_get_contents, "bola1.txt");
    will_return(__wrap_blogc_file_get_contents, b_strdup(
        "ASD: 123\n"
        "DATE: 2001-02-03 04:05:06\n"
        "--------\n"
        "bola"));
    will_return(__wrap_blogc_file_get_contents, "bola2.txt");
    will_return(__wrap_blogc_file_get_contents, b_strdup(
        "ASD: 456\n"
        "DATE: 2002-02-03 04:05:06\n"
        "--------\n"
        "bola"));
    will_return(__wrap_blogc_file_get_contents, "bola3.txt");
    will_return(__wrap_blogc_file_get_contents, b_strdup(
        "ASD: 789\n"
        "DATE: 2003-02-03 04:05:06\n"
        "--------\n"
        "bola"));
    blogc_error_t *err = NULL;
    b_slist_t *s = NULL;
    s = b_slist_append(s, b_strdup("bola1.txt"));
    s = b_slist_append(s, b_strdup("bola2.txt"));
    s = b_slist_append(s, b_strdup("bola3.txt"));
    b_trie_t *c = b_trie_new(free);
    b_slist_t *t = blogc_source_parse_from_files(c, s, &err);
    assert_null(err);
    assert_non_null(t);
    assert_int_equal(b_slist_length(t), 3);  // it is enough, no need to look at the items
    assert_int_equal(b_trie_size(c), 4);
    assert_string_equal(b_trie_lookup(c, "FILENAME_FIRST"), "bola1");
    assert_string_equal(b_trie_lookup(c, "FILENAME_LAST"), "bola3");
    assert_string_equal(b_trie_lookup(c, "DATE_FIRST"), "2001-02-03 04:05:06");
    assert_string_equal(b_trie_lookup(c, "DATE_LAST"), "2003-02-03 04:05:06");
    b_trie_free(c);
    b_slist_free_full(s, free);
    b_slist_free_full(t, (b_free_func_t) b_trie_free);
}


static void
test_source_parse_from_files_without_all_dates(void **state)
{
    will_return(__wrap_blogc_fprintf, stderr);
    will_return(__wrap_blogc_fprintf,
        "blogc: warning: 'DATE' variable provided for at least one source "
        "file, but not for all source files. This means that you may get wrong "
        "values for 'DATE_FIRST' and 'DATE_LAST' variables.\n");
    will_return(__wrap_blogc_file_get_contents, "bola1.txt");
    will_return(__wrap_blogc_file_get_contents, b_strdup(
        "ASD: 123\n"
        "--------\n"
        "bola"));
    will_return(__wrap_blogc_file_get_contents, "bola2.txt");
    will_return(__wrap_blogc_file_get_contents, b_strdup(
        "ASD: 456\n"
        "DATE: 2002-02-03 04:05:06\n"
        "--------\n"
        "bola"));
    will_return(__wrap_blogc_file_get_contents, "bola3.txt");
    will_return(__wrap_blogc_file_get_contents, b_strdup(
        "ASD: 789\n"
        "DATE: 2003-02-03 04:05:06\n"
        "--------\n"
        "bola"));
    blogc_error_t *err = NULL;
    b_slist_t *s = NULL;
    s = b_slist_append(s, b_strdup("bola1.txt"));
    s = b_slist_append(s, b_strdup("bola2.txt"));
    s = b_slist_append(s, b_strdup("bola3.txt"));
    b_trie_t *c = b_trie_new(free);
    b_slist_t *t = blogc_source_parse_from_files(c, s, &err);
    assert_null(err);
    assert_non_null(t);
    assert_int_equal(b_slist_length(t), 3);  // it is enough, no need to look at the items
    assert_int_equal(b_trie_size(c), 3);
    assert_string_equal(b_trie_lookup(c, "FILENAME_FIRST"), "bola1");
    assert_string_equal(b_trie_lookup(c, "FILENAME_LAST"), "bola3");
    assert_string_equal(b_trie_lookup(c, "DATE_LAST"), "2003-02-03 04:05:06");
    b_trie_free(c);
    b_slist_free_full(s, free);
    b_slist_free_full(t, (b_free_func_t) b_trie_free);
}


static void
test_source_parse_from_files_null(void **state)
{
    blogc_error_t *err = NULL;
    b_slist_t *s = NULL;
    b_trie_t *c = b_trie_new(free);
    b_slist_t *t = blogc_source_parse_from_files(c, s, &err);
    assert_null(err);
    assert_null(t);
    assert_int_equal(b_slist_length(t), 0);
    assert_int_equal(b_trie_size(c), 0);
    b_trie_free(c);
    b_slist_free_full(s, free);
    b_slist_free_full(t, (b_free_func_t) b_trie_free);
}


int
main(void)
{
    const UnitTest tests[] = {
        unit_test(test_get_filename),
        unit_test(test_template_parse_from_file),
        unit_test(test_template_parse_from_file_null),
        unit_test(test_source_parse_from_file),
        unit_test(test_source_parse_from_file_null),
        unit_test(test_source_parse_from_files),
        unit_test(test_source_parse_from_files_without_all_dates),
        unit_test(test_source_parse_from_files_null),
    };
    return run_tests(tests);
}