/*
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Library General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 * StringUtilsTest.cpp
 * Unittest for String functions.
 * Copyright (C) 2005-2008 Simon Newton
 */

#include <cppunit/extensions/HelperMacros.h>
#include <iostream>
#include <string>
#include <vector>
#include "ola/StringUtils.h"

using std::string;
using std::vector;
using ola::Escape;
using ola::EscapeString;
using ola::HexStringToUInt16;
using ola::HexStringToUInt;
using ola::IntToString;
using ola::ShortenString;
using ola::StringSplit;
using ola::StringToUInt16;
using ola::StringToUInt8;
using ola::StringToUInt;
using ola::StringTrim;
using ola::ToLower;

class StringUtilsTest: public CppUnit::TestFixture {
  CPPUNIT_TEST_SUITE(StringUtilsTest);
  CPPUNIT_TEST(testSplit);
  CPPUNIT_TEST(testTrim);
  CPPUNIT_TEST(testShorten);
  CPPUNIT_TEST(testIntToString);
  CPPUNIT_TEST(testEscape);
  CPPUNIT_TEST(testStringToUInt);
  CPPUNIT_TEST(testStringToUInt16);
  CPPUNIT_TEST(testStringToUInt8);
  CPPUNIT_TEST(testHexStringToUInt);
  CPPUNIT_TEST(testToLower);
  CPPUNIT_TEST_SUITE_END();

  public:
    void testSplit();
    void testTrim();
    void testShorten();
    void testIntToString();
    void testEscape();
    void testStringToUInt();
    void testStringToUInt16();
    void testStringToUInt8();
    void testHexStringToUInt();
    void testToLower();
};


CPPUNIT_TEST_SUITE_REGISTRATION(StringUtilsTest);

/*
 * Test the split function
 */
void StringUtilsTest::testSplit() {
  vector<string> tokens;
  string input = "";
  StringSplit(input, tokens);
  CPPUNIT_ASSERT_EQUAL((size_t) 1, tokens.size());
  CPPUNIT_ASSERT_EQUAL(string(""), tokens[0]);

  input = "1 2 345";
  tokens.clear();
  StringSplit(input, tokens);

  CPPUNIT_ASSERT_EQUAL((size_t) 3, tokens.size());
  CPPUNIT_ASSERT_EQUAL(string("1"), tokens[0]);
  CPPUNIT_ASSERT_EQUAL(string("2"), tokens[1]);
  CPPUNIT_ASSERT_EQUAL(string("345"), tokens[2]);

  input = "1,2,345";
  tokens.clear();
  StringSplit(input, tokens, ",");

  CPPUNIT_ASSERT_EQUAL((size_t) 3, tokens.size());
  CPPUNIT_ASSERT_EQUAL(string("1"), tokens[0]);
  CPPUNIT_ASSERT_EQUAL(string("2"), tokens[1]);
  CPPUNIT_ASSERT_EQUAL(string("345"), tokens[2]);

  input = ",1,2,345,,";
  tokens.clear();
  StringSplit(input, tokens, ",");

  CPPUNIT_ASSERT_EQUAL((size_t) 6, tokens.size());
  CPPUNIT_ASSERT_EQUAL(string(""), tokens[0]);
  CPPUNIT_ASSERT_EQUAL(string("1"), tokens[1]);
  CPPUNIT_ASSERT_EQUAL(string("2"), tokens[2]);
  CPPUNIT_ASSERT_EQUAL(string("345"), tokens[3]);
  CPPUNIT_ASSERT_EQUAL(string(""), tokens[4]);
  CPPUNIT_ASSERT_EQUAL(string(""), tokens[5]);

  input = "1 2,345";
  tokens.clear();
  StringSplit(input, tokens, " ,");

  CPPUNIT_ASSERT_EQUAL((size_t) 3, tokens.size());
  CPPUNIT_ASSERT_EQUAL(string("1"), tokens[0]);
  CPPUNIT_ASSERT_EQUAL(string("2"), tokens[1]);
  CPPUNIT_ASSERT_EQUAL(string("345"), tokens[2]);

  input = "1, 2,345";
  tokens.clear();
  StringSplit(input, tokens, " ,");

  CPPUNIT_ASSERT_EQUAL((size_t) 4, tokens.size());
  CPPUNIT_ASSERT_EQUAL(string("1"), tokens[0]);
  CPPUNIT_ASSERT_EQUAL(string(""), tokens[1]);
  CPPUNIT_ASSERT_EQUAL(string("2"), tokens[2]);
  CPPUNIT_ASSERT_EQUAL(string("345"), tokens[3]);

  input = "1";
  tokens.clear();
  StringSplit(input, tokens, ".");

  CPPUNIT_ASSERT_EQUAL((size_t) 1, tokens.size());
  CPPUNIT_ASSERT_EQUAL(string("1"), tokens[0]);
}


/*
 * Test the trim function.
 */
void StringUtilsTest::testTrim() {
  string input = "foo bar baz";
  StringTrim(&input);
  CPPUNIT_ASSERT_EQUAL(string("foo bar baz"), input);
  input = "  \rfoo bar\t\n";
  StringTrim(&input);
  CPPUNIT_ASSERT_EQUAL(string("foo bar"), input);
  input = "  \r\t\n";
  StringTrim(&input);
  CPPUNIT_ASSERT_EQUAL(string(""), input);
}


/*
 * Test the shorten function.
 */
void StringUtilsTest::testShorten() {
  string input = "foo bar baz";
  ShortenString(&input);
  CPPUNIT_ASSERT_EQUAL(string("foo bar baz"), input);
  input = "foo \0bar";
  ShortenString(&input);
  CPPUNIT_ASSERT_EQUAL(string("foo "), input);
  input = "foo\0bar\0baz";
  StringTrim(&input);
  CPPUNIT_ASSERT_EQUAL(string("foo"), input);
}


/*
 * test the IntToString function.
 */
void StringUtilsTest::testIntToString() {
  CPPUNIT_ASSERT_EQUAL(string("0"), IntToString(0));
  CPPUNIT_ASSERT_EQUAL(string("1234"), IntToString(1234));
  CPPUNIT_ASSERT_EQUAL(string("-1234"), IntToString(-1234));
  unsigned int i = 42;
  CPPUNIT_ASSERT_EQUAL(string("42"), IntToString(i));
}

void StringUtilsTest::testEscape() {
  string s1 = "foo\"";
  Escape(&s1);
  CPPUNIT_ASSERT_EQUAL(string("foo\\\""), s1);

  s1 = "he said \"foo\"";
  Escape(&s1);
  CPPUNIT_ASSERT_EQUAL(string("he said \\\"foo\\\""), s1);

  s1 = "backslash\\test";
  Escape(&s1);
  CPPUNIT_ASSERT_EQUAL(string("backslash\\\\test"), s1);

  s1 = "newline\ntest";
  Escape(&s1);
  CPPUNIT_ASSERT_EQUAL(string("newline\\ntest"), s1);

  s1 = "tab\ttest";
  Escape(&s1);
  CPPUNIT_ASSERT_EQUAL(string("tab\\ttest"), s1);

  s1 = "one\"two\\three/four\bfive\fsix\nseven\reight\tnine";
  Escape(&s1);
  CPPUNIT_ASSERT_EQUAL(
      string("one\\\"two\\\\three\\/four\\bfive\\fsix\\nseven\\reight\\tnine"),
      s1);

  s1 = "one\"two\\three/four\bfive\fsix\nseven\reight\tnine";
  string result = EscapeString(s1);
  CPPUNIT_ASSERT_EQUAL(
      string("one\\\"two\\\\three\\/four\\bfive\\fsix\\nseven\\reight\\tnine"),
      result);
}


void StringUtilsTest::testStringToUInt() {
  unsigned int value;
  CPPUNIT_ASSERT(!StringToUInt("", &value));
  CPPUNIT_ASSERT(!StringToUInt("-1", &value));
  CPPUNIT_ASSERT(StringToUInt("0", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 0, value);
  CPPUNIT_ASSERT(StringToUInt("1", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 1, value);
  CPPUNIT_ASSERT(StringToUInt("65537", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 65537, value);
  CPPUNIT_ASSERT(!StringToUInt("foo", &value));
}


void StringUtilsTest::testHexStringToUInt() {
  unsigned int value;
  CPPUNIT_ASSERT(!HexStringToUInt("", &value));
  CPPUNIT_ASSERT(!HexStringToUInt("-1", &value));

  CPPUNIT_ASSERT(HexStringToUInt("0", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 0, value);
  CPPUNIT_ASSERT(HexStringToUInt("1", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 1, value);
  CPPUNIT_ASSERT(HexStringToUInt("a", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 10, value);
  CPPUNIT_ASSERT(HexStringToUInt("f", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 15, value);
  CPPUNIT_ASSERT(HexStringToUInt("a1", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 161, value);
  CPPUNIT_ASSERT(HexStringToUInt("ff", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 255, value);
  CPPUNIT_ASSERT(HexStringToUInt("a1", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 161, value);
  CPPUNIT_ASSERT(HexStringToUInt("ff", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 255, value);
  CPPUNIT_ASSERT(HexStringToUInt("ffff", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 65535, value);

  CPPUNIT_ASSERT(HexStringToUInt("ffffff", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 16777215, value);
  CPPUNIT_ASSERT(HexStringToUInt("ffffffff", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 4294967295UL, value);
  CPPUNIT_ASSERT(HexStringToUInt("ef123456", &value));
  CPPUNIT_ASSERT_EQUAL((unsigned int) 4010947670UL, value);
  CPPUNIT_ASSERT(!HexStringToUInt("fz", &value));
  CPPUNIT_ASSERT(!HexStringToUInt("zfff", &value));
  CPPUNIT_ASSERT(!HexStringToUInt("0xf", &value));

  // test HexStringToUInt16
  uint16_t value2;
  CPPUNIT_ASSERT(!HexStringToUInt16("", &value2));
  CPPUNIT_ASSERT(!HexStringToUInt16("-1", &value2));

  CPPUNIT_ASSERT(HexStringToUInt16("0", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 0, value2);
  CPPUNIT_ASSERT(HexStringToUInt16("1", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 1, value2);
  CPPUNIT_ASSERT(HexStringToUInt16("a", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 10, value2);
  CPPUNIT_ASSERT(HexStringToUInt16("f", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 15, value2);
  CPPUNIT_ASSERT(HexStringToUInt16("a1", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 161, value2);
  CPPUNIT_ASSERT(HexStringToUInt16("ff", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 255, value2);
  CPPUNIT_ASSERT(HexStringToUInt16("a1", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 161, value2);
  CPPUNIT_ASSERT(HexStringToUInt16("ff", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 255, value2);
  CPPUNIT_ASSERT(HexStringToUInt16("ffff", &value2));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 65535, value2);

  CPPUNIT_ASSERT(!HexStringToUInt16("ffffff", &value2));
  CPPUNIT_ASSERT(!HexStringToUInt16("ffffffff", &value2));
  CPPUNIT_ASSERT(!HexStringToUInt16("ef123456", &value2));
  CPPUNIT_ASSERT(!HexStringToUInt16("fz", &value2));
  CPPUNIT_ASSERT(!HexStringToUInt16("zfff", &value2));
  CPPUNIT_ASSERT(!HexStringToUInt16("0xf", &value2));
}


void StringUtilsTest::testStringToUInt16() {
  uint16_t value;

  CPPUNIT_ASSERT(!StringToUInt16("", &value));
  CPPUNIT_ASSERT(!StringToUInt16("-1", &value));
  CPPUNIT_ASSERT(!StringToUInt16("65536", &value));

  CPPUNIT_ASSERT(StringToUInt16("0", &value));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 0, value);
  CPPUNIT_ASSERT(StringToUInt16("1", &value));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 1, value);
  CPPUNIT_ASSERT(StringToUInt16("143", &value));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 143, value);
  CPPUNIT_ASSERT(StringToUInt16("65535", &value));
  CPPUNIT_ASSERT_EQUAL((uint16_t) 65535, value);
}


void StringUtilsTest::testStringToUInt8() {
  uint8_t value;

  CPPUNIT_ASSERT(!StringToUInt8("", &value));
  CPPUNIT_ASSERT(!StringToUInt8("-1", &value));
  CPPUNIT_ASSERT(!StringToUInt8("256", &value));

  CPPUNIT_ASSERT(StringToUInt8("0", &value));
  CPPUNIT_ASSERT_EQUAL((uint8_t) 0, value);
  CPPUNIT_ASSERT(StringToUInt8("1", &value));
  CPPUNIT_ASSERT_EQUAL((uint8_t) 1, value);
  CPPUNIT_ASSERT(StringToUInt8("143", &value));
  CPPUNIT_ASSERT_EQUAL((uint8_t) 143, value);
  CPPUNIT_ASSERT(StringToUInt8("255", &value));
  CPPUNIT_ASSERT_EQUAL((uint8_t) 255, value);
}


void StringUtilsTest::testToLower() {
  string s = "HelLo There";
  ToLower(&s);
  CPPUNIT_ASSERT_EQUAL(string("hello there"), s);
}