#include "CppUnitTest.h"
#include "JsonArray.h"
#include "JsonHashTable.h"

using namespace Microsoft::VisualStudio::CppUnitTestFramework;
using namespace ArduinoJson::Generator;

namespace JsonGeneratorTests
{		
    TEST_CLASS(JsonArrayTests)
    {
        JsonArray<2> arr;
        char buffer[256];

    public:
        
        TEST_METHOD(Empty)
        {
            returnValueIs(2);
            jsonIs("[]");
        }

        TEST_METHOD(Null)
        {
            addValue((char*)0);

            returnValueIs(6);
            jsonIs("[null]");
        }

        TEST_METHOD(OneString)
        {
            addValue("hello");

            returnValueIs(9);
            jsonIs("[\"hello\"]");
        }

        TEST_METHOD(TwoStrings)
        {
            addValue("hello");
            addValue("world");

            returnValueIs(17);
            jsonIs("[\"hello\",\"world\"]");
        }

        TEST_METHOD(OneStringOverCapacity)
        {
            addValue("hello");
            addValue("world");
            addValue("lost");

            returnValueIs(17);
            jsonIs("[\"hello\",\"world\"]");
        }

        TEST_METHOD(OneDoubleDefaultDigits)
        {
            addValue(3.14159265358979323846);
            jsonIs("[3.14]");
        }

        TEST_METHOD(OneDoubleFourDigits)
        {
            addValue<4>(3.14159265358979323846);
            jsonIs("[3.1416]");
        }

        TEST_METHOD(OneInteger)
        {
            addValue(1);

            returnValueIs(3);
            jsonIs("[1]");
        }

        TEST_METHOD(TwoIntegers)
        {
            addValue(1);
            addValue(2);

            returnValueIs(5);
            jsonIs("[1,2]");
        }

        TEST_METHOD(OneIntegerOverCapacity)
        {
            addValue(1);
            addValue(2);
            addValue(3);

            returnValueIs(5);
            jsonIs("[1,2]");
        }

        TEST_METHOD(OneTrue)
        {
            addValue(true);

            returnValueIs(6);
            jsonIs("[true]");
        }

        TEST_METHOD(OneFalse)
        {
            addValue(false);

            returnValueIs(7);
            jsonIs("[false]");
        }

        TEST_METHOD(TwoBooleans)
        {
            addValue(false);
            addValue(true);

            returnValueIs(12);
            jsonIs("[false,true]");
        }

        TEST_METHOD(OneBooleanOverCapacity)
        {
            addValue(false);
            addValue(true);
            addValue(false);

            returnValueIs(12);
            jsonIs("[false,true]");
        }

        TEST_METHOD(OneEmptyNestedArray)
        {
            JsonArray<1> nestedArray;
            
            addNested(nestedArray);

            returnValueIs(4);
            jsonIs("[[]]");
        }

        TEST_METHOD(OneEmptyNestedHash)
        {
            JsonHashTable<1> nestedHash;

            addNested(nestedHash);

            returnValueIs(4);
            jsonIs("[{}]");
        }

        TEST_METHOD(OneNestedArrayWithOneInteger)
        {
            JsonArray<1> nestedArray;
            nestedArray.add(1);

            addNested(nestedArray);

            returnValueIs(5);
            jsonIs("[[1]]");
        }

    private:

        void addNested(JsonObjectBase& value)
        {
            arr.add<JsonObjectBase&>(value);
        }

        template<typename T>
        void addValue(T value)
        {
            arr.add(value);
        }

        template<int DIGITS>
        void addValue(double value)
        {
            arr.add<DIGITS>(value);
        }

        void jsonIs(const char* expected)
        {      
            arr.printTo(buffer, sizeof(buffer));
            Assert::AreEqual(expected, buffer);
        }

        void returnValueIs(size_t expected)
        {
            size_t actual = arr.printTo(buffer, sizeof(buffer));
            Assert::AreEqual(expected, actual);
        }
    };
}