#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <head-unit.h>
#include "contact_list.h"

static void beforetest(void)
{
    contact_list_clear();
}

static void aftertest(void)
{
    contact_list_clear();
}

static void empty_list_when_none_added(void)
{
    struct contact_node_t *list = get_contact_list();
    assert_is_null(list);
    destroy_list(list);
}

static void contains_one_element(void)
{
    contact_list_add("James", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    assert_int_equals(1, get_size(list));
    destroy_list(list);
}

static void first_element_correct(void)
{
    contact_list_add("James", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *james = list->contact;

    assert_string_equals("James", james->name);
    destroy_list(list);
}

static void contains_two_elements(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    struct contact_node_t *list = get_contact_list();

    assert_int_equals(2, get_size(list));
    destroy_list(list);
}

static void first_and_second_elements_correct(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    
    struct contact_t *dave = list->contact;
    struct contact_t *james = (list->next)->contact;
    
    assert_string_equals("James", james->name);
    assert_string_equals("Dave", dave->name);
    destroy_list(list);
}

static void contains_three_elements(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    
    assert_int_equals(3, get_size(list));
    destroy_list(list);
}

static void first_three_elements_correct(void)
{
    contact_list_add("Bob", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("James", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *bob = list->contact;
    struct contact_t *dave = (list->next)->contact;
    struct contact_t *james = ((list->next)->next)->contact;
    
    assert_string_equals("James", james->name);
    assert_string_equals("Dave", dave->name);
    assert_string_equals("Bob", bob->name);
    destroy_list(list);
}

static void add_twice_at_beginning_adds_once(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *bob = list->contact;
    struct contact_t *dave = (list->next)->contact;
    struct contact_t *james = ((list->next)->next)->contact;
    
    assert_int_equals(3, get_size(list));    
    assert_string_equals("James", james->name);
    assert_string_equals("Dave", dave->name);
    assert_string_equals("Bob", bob->name);
    destroy_list(list);
}

static void add_twice_in_middle_adds_once(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("James", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *bob = list->contact;
    struct contact_t *dave = (list->next)->contact;
    struct contact_t *james = ((list->next)->next)->contact;
    
    assert_int_equals(3, get_size(list));    
    assert_string_equals("James", james->name);
    assert_string_equals("Dave", dave->name);
    assert_string_equals("Bob", bob->name);
    destroy_list(list);
}

static void add_twice_at_end_adds_once(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);
    contact_list_add("James", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *bob = list->contact;
    struct contact_t *dave = (list->next)->contact;
    struct contact_t *james = ((list->next)->next)->contact;
    
    assert_int_equals(3, get_size(list));    
    assert_string_equals("James", james->name);
    assert_string_equals("Dave", dave->name);
    assert_string_equals("Bob", bob->name);
    destroy_list(list);
}

static void remove_when_none_does_nothing(void)
{
    contact_list_remove("James");
    struct contact_node_t *list = get_contact_list();

    assert_int_equals(0, get_size(list));
    destroy_list(list);
}

static void remove_when_one_removes(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_remove("James");
    struct contact_node_t *list = get_contact_list();
    
    assert_int_equals(0, get_size(list));
    destroy_list(list);
}

static void remove_first_when_two(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);

    contact_list_remove("James");
    struct contact_node_t *list = get_contact_list();
    
    assert_int_equals(1, get_size(list));
    struct contact_t *dave = list->contact;
    assert_string_equals("Dave", dave->name);
    destroy_list(list);
}

static void remove_second_when_two(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);

    contact_list_remove("Dave");
    struct contact_node_t *list = get_contact_list();
    
    assert_int_equals(1, get_size(list));
    struct contact_t *james = list->contact;
    assert_string_equals("James", james->name);
    destroy_list(list);
}

static void remove_first_when_three(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    contact_list_remove("James");
    struct contact_node_t *list = get_contact_list();
    
    assert_int_equals(2, get_size(list));
    struct contact_t *bob = list->contact;
    struct contact_t *dave = (list->next)->contact;
    
    assert_string_equals("Dave", dave->name);
    assert_string_equals("Bob", bob->name);
    destroy_list(list);
}

static void remove_second_when_three(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    contact_list_remove("Dave");
    struct contact_node_t *list = get_contact_list();
    
    assert_int_equals(2, get_size(list));
    struct contact_t *bob = list->contact;
    struct contact_t *james = (list->next)->contact;
    
    assert_string_equals("James", james->name);
    assert_string_equals("Bob", bob->name);
    destroy_list(list);
}

static void remove_third_when_three(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    contact_list_remove("Bob");
    struct contact_node_t *list = get_contact_list();
    
    assert_int_equals(2, get_size(list));
    struct contact_t *dave = list->contact;
    struct contact_t *james = (list->next)->contact;
    
    assert_string_equals("James", james->name);
    assert_string_equals("Dave", dave->name);
    destroy_list(list);
}

static void test_show_when_value(void)
{
    contact_list_add("James", "away", NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *james = list->contact;
    
    assert_string_equals("away", james->show);
    destroy_list(list);
}

static void test_show_online_when_no_value(void)
{
    contact_list_add("James", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *james = list->contact;
    
    assert_string_equals("online", james->show);
    destroy_list(list);
}

static void test_show_online_when_empty_string(void)
{
    contact_list_add("James", "", NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *james = list->contact;
    
    assert_string_equals("online", james->show);
    destroy_list(list);
}

static void test_status_when_value(void)
{
    contact_list_add("James", NULL, "I'm not here right now");
    struct contact_node_t *list = get_contact_list();
    struct contact_t *james = list->contact;
    
    assert_string_equals("I'm not here right now", james->status);
    destroy_list(list);
}

static void test_status_when_no_value(void)
{
    contact_list_add("James", NULL, NULL);
    struct contact_node_t *list = get_contact_list();
    struct contact_t *james = list->contact;
    
    assert_is_null(james->status);
    destroy_list(list);
}

static void update_show(void)
{
    contact_list_add("James", "away", NULL);
    contact_list_add("James", "dnd", NULL);
    struct contact_node_t *list = get_contact_list();

    assert_int_equals(1, get_size(list));
    struct contact_t *james = list->contact;
    assert_string_equals("James", james->name);
    assert_string_equals("dnd", james->show);
    destroy_list(list);
}

static void set_show_to_null(void)
{
    contact_list_add("James", "away", NULL);
    contact_list_add("James", NULL, NULL);
    struct contact_node_t *list = get_contact_list();

    assert_int_equals(1, get_size(list));
    struct contact_t *james = list->contact;
    assert_string_equals("James", james->name);
    assert_string_equals("online", james->show);
    destroy_list(list);
}

static void update_status(void)
{
    contact_list_add("James", NULL, "I'm not here right now");
    contact_list_add("James", NULL, "Gone to lunch");
    struct contact_node_t *list = get_contact_list();

    assert_int_equals(1, get_size(list));
    struct contact_t *james = list->contact;
    assert_string_equals("James", james->name);
    assert_string_equals("Gone to lunch", james->status);
    destroy_list(list);
}

static void set_status_to_null(void)
{
    contact_list_add("James", NULL, "Gone to lunch");
    contact_list_add("James", NULL, NULL);
    struct contact_node_t *list = get_contact_list();

    assert_int_equals(1, get_size(list));
    struct contact_t *james = list->contact;
    assert_string_equals("James", james->name);
    assert_is_null(james->status);
    destroy_list(list);
}

static void find_first_exists(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    char *search = (char *) malloc(2 * sizeof(char));
    strcpy(search, "B");

    char *result = find_contact(search);
    assert_string_equals("Bob", result);
    free(result);
    free(search);
}

static void find_second_exists(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    char *result = find_contact("Dav");
    assert_string_equals("Dave", result);
    free(result);
}

static void find_third_exists(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    char *result = find_contact("Ja");
    assert_string_equals("James", result);
    free(result);
}

static void find_returns_null(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    char *result = find_contact("Mike");
    assert_is_null(result);
}

static void find_on_empty_returns_null(void)
{
    char *result = find_contact("James");
    assert_is_null(result);
}

static void find_twice_returns_second_when_two_match(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Jamie", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    char *result1 = find_contact("Jam");
    char *result2 = find_contact(result1);
    assert_string_equals("Jamie", result2);
    free(result1);
    free(result2);
}

static void find_five_times_finds_fifth(void)
{
    contact_list_add("Jama", NULL, NULL);
    contact_list_add("Jamb", NULL, NULL);
    contact_list_add("Mike", NULL, NULL);
    contact_list_add("Dave", NULL, NULL);
    contact_list_add("Jamm", NULL, NULL);
    contact_list_add("Jamn", NULL, NULL);
    contact_list_add("Matt", NULL, NULL);
    contact_list_add("Jamo", NULL, NULL);
    contact_list_add("Jamy", NULL, NULL);
    contact_list_add("Jamz", NULL, NULL);

    char *result1 = find_contact("Jam");
    char *result2 = find_contact(result1);
    char *result3 = find_contact(result2);
    char *result4 = find_contact(result3);
    char *result5 = find_contact(result4);
    assert_string_equals("Jamo", result5);
    free(result1);
    free(result2);
    free(result3);
    free(result4);
    free(result5);
}

static void find_twice_returns_first_when_two_match_and_reset(void)
{
    contact_list_add("James", NULL, NULL);
    contact_list_add("Jamie", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);

    char *result1 = find_contact("Jam");
    reset_search_attempts();
    char *result2 = find_contact(result1);
    assert_string_equals("James", result2);
    free(result1);
    free(result2);
}

static void removed_contact_not_in_search(void)
{
    contact_list_add("Jamatron", NULL, NULL);
    contact_list_add("Bob", NULL, NULL);
    contact_list_add("Jambo", NULL, NULL);
    contact_list_add("James", NULL, NULL);
    contact_list_add("Jamie", NULL, NULL);

    char *result1 = find_contact("Jam"); // Jamatron
    char *result2 = find_contact(result1); // Jambo
    contact_list_remove("James");
    char *result3 = find_contact(result2);
    assert_string_equals("Jamie", result3);
    free(result1);
    free(result2);
    free(result3);
}

void register_contact_list_tests(void)
{
    TEST_MODULE("contact_list tests");
    BEFORETEST(beforetest);
    AFTERTEST(aftertest);
    TEST(empty_list_when_none_added);
    TEST(contains_one_element);
    TEST(first_element_correct);
    TEST(contains_two_elements);
    TEST(first_and_second_elements_correct);
    TEST(contains_three_elements);
    TEST(first_three_elements_correct);
    TEST(add_twice_at_beginning_adds_once);
    TEST(add_twice_in_middle_adds_once);
    TEST(add_twice_at_end_adds_once);
    TEST(remove_when_none_does_nothing);
    TEST(remove_when_one_removes);
    TEST(remove_first_when_two);
    TEST(remove_second_when_two);
    TEST(remove_first_when_three);
    TEST(remove_second_when_three);
    TEST(remove_third_when_three);
    TEST(test_show_when_value);
    TEST(test_show_online_when_no_value);
    TEST(test_show_online_when_empty_string);
    TEST(test_status_when_value);
    TEST(test_status_when_no_value);
    TEST(update_show);
    TEST(set_show_to_null);
    TEST(update_status);
    TEST(set_status_to_null);
    TEST(find_first_exists);
    TEST(find_second_exists);
    TEST(find_third_exists);
    TEST(find_returns_null);
    TEST(find_on_empty_returns_null);
    TEST(find_twice_returns_second_when_two_match);
    TEST(find_twice_returns_first_when_two_match_and_reset);
    TEST(removed_contact_not_in_search);
    TEST(find_five_times_finds_fifth);
}