#include "test_suite.h"
#include "string_functions.h"

#include <iomanip>
#include <iostream>

using namespace std;

const int MESSAGE_WIDTH = 45;

int test_int(int& testNum, const string MESSAGE, const int LHS, const int RHS)  {
    cout << "Test #" << setw(3) << right << ++testNum;
    cout << setw(MESSAGE_WIDTH) << right;
    cout << MESSAGE << ": ";
    cout << left;
    if (LHS == RHS)
        cout << "    PASSED   \n";
    else
        cout << " !!>FAILED<!! Returned: \"" << LHS << "\" != Expected: \"" << RHS << "\"\n";
    return LHS == RHS ? 1 : 0;
}

int test_size_t(int& testNum, const string MESSAGE, const size_t LHS, const size_t RHS)  {
    cout << "Test #" << setw(3) << right << ++testNum;
    cout << setw(MESSAGE_WIDTH) << right;
    cout << MESSAGE << ": ";
    cout << left;
    if (LHS == RHS)
        cout << "    PASSED   \n";
    else
        cout << " !!>FAILED<!! Returned: \"" << LHS << "\" != Expected: \"" << RHS << "\"\n";
    return LHS == RHS ? 1 : 0;
}

int test_char(int& testNum, const string MESSAGE, const char LHS, const char RHS)  {
    cout << "Test #" << setw(3) << right << ++testNum;
    cout << setw(MESSAGE_WIDTH) << right;
    cout << MESSAGE << ": ";
    cout << left;
    if (LHS == RHS)
        cout << "    PASSED   \n";
    else
        cout << " !!>FAILED<!! Returned: \"" << LHS << "\" != Expected: \"" << RHS << "\"\n";
    return LHS == RHS ? 1 : 0;
}

int test_string(int& testNum, const string MESSAGE, const string LHS, const string RHS)  {
    cout << "Test #" << setw(3) << right << ++testNum;
    cout << setw(MESSAGE_WIDTH) << right;
    cout << MESSAGE << ": ";
    cout << left;
    if (LHS == RHS)
        cout << "    PASSED   \n";
    else
        cout << " !!>FAILED<!! Returned: \"" << LHS << "\" != Expected: \"" << RHS << "\"\n";
    return LHS == RHS ? 1 : 0;
}

int test_vector_string(int& testNum, const string MESSAGE, const vector<string>& LHS, const vector<string>& RHS) {
    cout << "Test #" << setw(3) << right << ++testNum;
    cout << setw(MESSAGE_WIDTH) << right;
    cout << MESSAGE << ": ";
    cout << left;

    bool vectorsMatch = (LHS.size() == RHS.size());
    if (vectorsMatch) {
        for(unsigned int i = 0; i < LHS.size(); i++) {
            if(LHS.at(i) != RHS.at(i)) {
                vectorsMatch = false;
                break;
            }
        }
    }
    if (vectorsMatch) {
        cout << "    PASSED   \n";
    } else {
        cout << " !!>FAILED<!! Returned: {";
        for(unsigned int i = 0; i < LHS.size(); i++) {
            cout << "\"" << LHS.at(i) << "\"";
            if(i < LHS.size() - 1) {
                cout << ", ";
            }
        }
        cout << "} != Expected: {";
        for(unsigned int i = 0; i < RHS.size(); i++) {
            cout << "\"" << RHS.at(i) << "\"";
            if(i < RHS.size() - 1) {
                cout << ", ";
            }
        }
        cout << "}\n";
    }
    return vectorsMatch ? 1 : 0;
}

bool run_all_tests()  {
    int totalPassed = 0, totalNumTests = 0;
    totalPassed += test_size_t(        totalNumTests,  "Testing string_length()",              string_length("Now") , 3 );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_length()",              string_length("Tablet") , 6 );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_length()",              string_length("The length is 17.") , 17 );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_length()",              string_length("A") , 1 );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_length()",              string_length("") , 0 );
    cout << endl;

    totalPassed += test_char(          totalNumTests,  "Testing string_char_at()",             string_char_at("Elephant", 3) , 'p' );
    totalPassed += test_char(          totalNumTests,  "Testing string_char_at()",             string_char_at("Giraffe", 2) , 'r' );
    totalPassed += test_char(          totalNumTests,  "Testing string_char_at()",             string_char_at("Armadillo", 4) , 'd' );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_append()",              string_append("There's a ", "natural mystic.") , "There's a natural mystic." );
    totalPassed += test_string(        totalNumTests,  "Testing string_append()",              string_append("It's the ", "eye of the tiger.") , "It's the eye of the tiger." );
    totalPassed += test_string(        totalNumTests,  "Testing string_append()",              string_append("Some", "thing") , "Something" );
    totalPassed += test_string(        totalNumTests,  "Testing string_append()",              string_append("", "Nothing to something.") , "Nothing to something." );
    totalPassed += test_string(        totalNumTests,  "Testing string_append()",              string_append("Not adding anything.", "") , "Not adding anything." );
    totalPassed += test_string(        totalNumTests,  "Testing string_append()",              string_append("", "") , "" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_insert()",              string_insert("If you carefully.", "listen ", 7) , "If you listen carefully." );
    totalPassed += test_string(        totalNumTests,  "Testing string_insert()",              string_insert("carefully.", "Watch ", 0) , "Watch carefully." );
    totalPassed += test_string(        totalNumTests,  "Testing string_insert()",              string_insert("Iner", "t", 2) , "Inter" );
    totalPassed += test_string(        totalNumTests,  "Testing string_insert()",              string_insert("", "TA DA!", 0) , "TA DA!" );
    totalPassed += test_string(        totalNumTests,  "Testing string_insert()",              string_insert("TA DA!", "", 0) , "TA DA!" );
    totalPassed += test_string(        totalNumTests,  "Testing string_insert()",              string_insert("", "", 0) , "" );
    cout << endl;

    totalPassed += test_size_t(        totalNumTests,  "Testing string_find()",                string_find("Have to face reality now.", 'o') , 6 );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_find()",                string_find("Have to face reality now.", 'e') , 3 );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_find()",                string_find("Have to face reality now.", 'r') , 13 );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_find()",                string_find("Have to face reality now.", 'a') , 1 );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_find()",                string_find("Have to face reality now.", 'q') , string::npos );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_find()",                string_find("Have to face reality now.", 'h') , string::npos );
    totalPassed += test_size_t(        totalNumTests,  "Testing string_find()",                string_find("", '?') , string::npos );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_substring()",           string_substring("Such a natural mystic", 7, 7) , "natural" );
    totalPassed += test_string(        totalNumTests,  "Testing string_substring()",           string_substring("Such a natural mystic", 8, 2) , "at" );
    totalPassed += test_string(        totalNumTests,  "Testing string_substring()",           string_substring("Such a natural mystic", 0, 4) , "Such" );
    totalPassed += test_string(        totalNumTests,  "Testing string_substring()",           string_substring("Something to nothing.", 0, 0) , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_substring()",           string_substring("Something to nothing.", 10, 0) , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_substring()",           string_substring("", 0, 0) , "" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_replace()",             string_replace("Strings are not the way", "Strings", "Things") , "Things are not the way" );
    totalPassed += test_string(        totalNumTests,  "Testing string_replace()",             string_replace("Strings are not the way", "not the", "the") , "Strings are the way" );
    totalPassed += test_string(        totalNumTests,  "Testing string_replace()",             string_replace("Show me the things", "things", "way") , "Show me the way" );
    totalPassed += test_string(        totalNumTests,  "Testing string_replace()",             string_replace("Show me the things", "car", "way") , "Show me the things" );
    totalPassed += test_string(        totalNumTests,  "Testing string_replace()",             string_replace("Show me the things", "the ", "") , "Show me things" );
    totalPassed += test_string(        totalNumTests,  "Testing string_replace()",             string_replace("", "now what?", "hmm") , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_replace()",             string_replace("", "now", "") , "" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_first_word()",          string_first_word("The quick brown fox jumped over the lazy dog") , "The" );
    totalPassed += test_string(        totalNumTests,  "Testing string_first_word()",          string_first_word("A man a plan a canal Panama") , "A" );
    totalPassed += test_string(        totalNumTests,  "Testing string_first_word()",          string_first_word("I have the hang of this") , "I" );
    totalPassed += test_string(        totalNumTests,  "Testing string_first_word()",          string_first_word("Testing string_first_word()") , "Testing" );
    totalPassed += test_string(        totalNumTests,  "Testing string_first_word()",          string_first_word("Single") , "Single" );
    totalPassed += test_string(        totalNumTests,  "Testing string_first_word()",          string_first_word("Uh oh") , "Uh" );
    totalPassed += test_string(        totalNumTests,  "Testing string_first_word()",          string_first_word("") , "" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_remove_first_word()",   string_remove_first_word("The quick brown fox jumped over the lazy dog") , "quick brown fox jumped over the lazy dog" );
    totalPassed += test_string(        totalNumTests,  "Testing string_remove_first_word()",   string_remove_first_word("Testing string_remove_first_word()") , "string_remove_first_word()" );
    totalPassed += test_string(        totalNumTests,  "Testing string_remove_first_word()",   string_remove_first_word("Goodbye") , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_remove_first_word()",   string_remove_first_word("") , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_remove_first_word() twice",string_remove_first_word(string_remove_first_word("The quick brown fox jumped over the lazy dog")) , "brown fox jumped over the lazy dog" );
    totalPassed += test_string(        totalNumTests,  "Testing string_remove_first_word() twice",string_remove_first_word("Testing string_remove_first_word()") , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_remove_first_word() twice",string_remove_first_word(string_remove_first_word("Goodbye")) , "" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_second_word()",         string_second_word("The quick brown fox jumped over the lazy dog") , "quick" );
    totalPassed += test_string(        totalNumTests,  "Testing string_second_word()",         string_second_word("A man a plan a canal Panama") , "man" );
    totalPassed += test_string(        totalNumTests,  "Testing string_second_word()",         string_second_word("Testing string_second_word()") , "string_second_word()" );
    totalPassed += test_string(        totalNumTests,  "Testing string_second_word()",         string_second_word("I have the hang of this") , "have" );
    totalPassed += test_string(        totalNumTests,  "Testing string_second_word()",         string_second_word("Uh oh") , "oh" );
    totalPassed += test_string(        totalNumTests,  "Testing string_second_word()",         string_second_word("Single") , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_second_word()",         string_second_word("") , "" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_third_word()",          string_third_word("The quick brown fox jumped over the lazy dog") , "brown" );
    totalPassed += test_string(        totalNumTests,  "Testing string_third_word()",          string_third_word("A man a plan a canal Panama") , "a" );
    totalPassed += test_string(        totalNumTests,  "Testing string_third_word()", 	       string_third_word("I have the hang of this") , "the" );
    totalPassed += test_string(        totalNumTests,  "Testing string_third_word()", 	       string_third_word("Uh oh no") , "no" );
    totalPassed += test_string(        totalNumTests,  "Testing string_third_word()", 	       string_third_word("Uh oh") , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_third_word()", 	       string_third_word("Single") , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_third_word()", 	       string_third_word("") , "" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(1)",           string_nth_word("The quick brown fox jumped over the lazy dog", 1) , "The" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(2)",           string_nth_word("The quick brown fox jumped over the lazy dog", 2) , "quick" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(3)",           string_nth_word("The quick brown fox jumped over the lazy dog", 3) , "brown" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(4)",           string_nth_word("The quick brown fox jumped over the lazy dog", 4) , "fox" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(5)",           string_nth_word("The quick brown fox jumped over the lazy dog", 5) , "jumped" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(6)",           string_nth_word("The quick brown fox jumped over the lazy dog", 6) , "over" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(7)",           string_nth_word("The quick brown fox jumped over the lazy dog", 7) , "the" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(8)",           string_nth_word("The quick brown fox jumped over the lazy dog", 8) , "lazy" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(9)",           string_nth_word("The quick brown fox jumped over the lazy dog", 9) , "dog" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(10)",          string_nth_word("The quick brown fox jumped over the lazy dog", 10) , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(11)",          string_nth_word("The quick brown fox jumped over the lazy dog", 11) , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(1)",           string_nth_word("", 1) , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(2)",           string_nth_word("", 2) , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(3)",           string_nth_word("", 3) , "" );
    totalPassed += test_string(        totalNumTests,  "Testing string_nth_word(10)",          string_nth_word("", 10) , "" );
    cout << endl;

    vector<string> tokens1 = {"The", "quick", "brown", "fox", "jumped", "over", "the", "lazy", "dog"};
    totalPassed += test_vector_string( totalNumTests,  "Testing string_tokenize()",            string_tokenize("The quick brown fox jumped over the lazy dog", ' '), tokens1 );
    totalPassed += test_vector_string( totalNumTests,  "Testing string_tokenize()",            string_tokenize("The@quick@brown@fox@jumped@over@the@lazy@dog", '@'), tokens1 );

    vector<string> tokens2 = {"The quick brown fox jumped over the lazy dog"};
    totalPassed += test_vector_string( totalNumTests,  "Testing string_tokenize()",            string_tokenize("The quick brown fox jumped over the lazy dog", '*'), tokens2 );
    
    vector<string> tokens3 = {""};
    totalPassed += test_vector_string( totalNumTests,  "Testing string_tokenize()",            string_tokenize("", '*'), tokens3 );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_substitute()",          string_substitute("The Gxxgle", 'x', 'o') , "The Google" );
    totalPassed += test_string(        totalNumTests,  "Testing string_substitute()",          string_substitute("$chool of Mine$", '$', 's') , "school of Mines" );
    totalPassed += test_string(        totalNumTests,  "Testing string_substitute()",          string_substitute("$chool of Mine$", '+', '*') , "$chool of Mine$" );
    totalPassed += test_string(        totalNumTests,  "Testing string_substitute() twice",    string_substitute(string_substitute("D--", '-', '+'), 'D', 'C') , "C++" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_to_lower()",            string_to_lower("This SHOULD be LOWER case") , "this should be lower case" );
    totalPassed += test_string(        totalNumTests,  "Testing string_to_lower()",            string_to_lower("MNASDF874792L[]//.;[\t],") , "mnasdf874792l[]//.;[\t]," );
    totalPassed += test_string(        totalNumTests,  "Testing string_to_lower()",            string_to_lower("C++") , "c++" );
    totalPassed += test_string(        totalNumTests,  "Testing string_to_lower()",            string_to_lower("this is already lower case") , "this is already lower case" );
    totalPassed += test_string(        totalNumTests,  "Testing string_to_lower()",            string_to_lower("1234567890,./;'[]") , "1234567890,./;'[]" );
    cout << endl;

    totalPassed += test_string(        totalNumTests,  "Testing string_to_upper()",            string_to_upper("This SHOULD be upper case") , "THIS SHOULD BE UPPER CASE" );
    totalPassed += test_string(        totalNumTests,  "Testing string_to_upper()",            string_to_upper("mnasdf874792l[]//.;[\t],") , "MNASDF874792L[]//.;[\t]," );
    totalPassed += test_string(        totalNumTests,  "Testing string_to_upper()",            string_to_upper("c++") , "C++" );
    totalPassed += test_string(        totalNumTests,  "Testing string_to_upper()",            string_to_upper("THIS IS ALREADY UPPER CASE") , "THIS IS ALREADY UPPER CASE" );
    totalPassed += test_string(        totalNumTests,  "Testing string_to_upper()",            string_to_upper("1234567890,./;'[]") , "1234567890,./;'[]" );
    cout << endl;

    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("C++", "c++") , -1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("C++", "C++") , 0 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("c++", "c++") , 0 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("c++", "C++") , 1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("short", "shorter") , -1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("longer", "long") , 1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("after", "later") , -1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("later", "after") , 1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("G", "g") , -1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("g", "G") , 1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("GoOse", "Bye") , 1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("Equal", "Equal") , 0 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("Not Empty", "") , 1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("", "Not Empty") , -1 );
    totalPassed += test_int(           totalNumTests,  "Testing string_compare()",             string_compare("", "") , 0 );
    cout << endl;

    cout << "Tests Passed: " << setw(3) << right << totalPassed << " / " << setw(3) << totalNumTests << " (" << setprecision(0) << fixed << totalPassed * 100.0f / totalNumTests << "%)" << endl << endl;
    
    return (totalPassed == totalNumTests);
}