/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <boost/test/unit_test.hpp>

#include "ignite/ignition.h"
#include "ignite/test_utils.h"

using namespace ignite;
using namespace cache;
using namespace boost::unit_test;

namespace
{
    /** Test put affinity key Java task. */
    const std::string TEST_PUT_AFFINITY_KEY_TASK("org.apache.ignite.platform.PlatformComputePutAffinityKeyTask");
}

class InteropTestSuiteFixture
{
public:
    InteropTestSuiteFixture()
    {
        // No-op.
    }

    ~InteropTestSuiteFixture()
    {
        ignite::Ignition::StopAll(false);
    }
};


/**
 * Affinity key class.
 */
struct AffinityKey
{
    /** Key */
    int32_t key;

    /** Affinity key */
    int32_t aff;

    /**
     * Default constructor.
     */
    AffinityKey() :
            key(0),
            aff(0)
    {
        // No-op.
    }

    /**
     * Constructor.
     * @param key Key.
     * @param aff Affinity key.
     */
    AffinityKey(int32_t key, int32_t aff) :
            key(key),
            aff(aff)
    {
        // No-op.
    }
};

namespace ignite
{
    namespace binary
    {
        template<>
        struct BinaryType<AffinityKey> : BinaryTypeDefaultAll<AffinityKey>
        {
            static void GetTypeName(std::string& dst)
            {
                dst = "AffinityKey";
            }

            static void Write(BinaryWriter& writer, const AffinityKey& obj)
            {
                writer.WriteInt32("key", obj.key);
                writer.WriteInt32("aff", obj.aff);
            }

            static void Read(BinaryReader& reader, AffinityKey& dst)
            {
                dst.key = reader.ReadInt32("key");
                dst.aff = reader.ReadInt32("aff");
            }

            static void GetAffinityFieldName(std::string& dst)
            {
                dst = "aff";
            }
        };
    }
}

BOOST_FIXTURE_TEST_SUITE(InteropTestSuite, InteropTestSuiteFixture)

#ifdef ENABLE_STRING_SERIALIZATION_VER_2_TESTS

BOOST_AUTO_TEST_CASE(StringUtfInvalidSequence)
{
    Ignite ignite = ignite_test::StartNode("cache-test.xml");

    Cache<std::string, std::string> cache = ignite.CreateCache<std::string, std::string>("Test");

    std::string initialValue;

    initialValue.push_back(static_cast<unsigned char>(0xD8));
    initialValue.push_back(static_cast<unsigned char>(0x00));

    try
    {
        cache.Put("key", initialValue);

        std::string cachedValue = cache.Get("key");

        BOOST_ERROR("Exception is expected due to invalid format.");
    }
    catch (const IgniteError&)
    {
        // Expected in this mode.
    }

    Ignition::StopAll(false);
}

BOOST_AUTO_TEST_CASE(StringUtfInvalidCodePoint)
{
    putenv("IGNITE_BINARY_MARSHALLER_USE_STRING_SERIALIZATION_VER_2=true");

    Ignite ignite = ignite_test::StartNode("cache-test.xml");

    Cache<std::string, std::string> cache = ignite.CreateCache<std::string, std::string>("Test");

    std::string initialValue;

    //    1110xxxx 10xxxxxx 10xxxxxx |
    // <=          11011000 00000000 | U+D8
    //  = 11101101 10100000 10000000 | ED A0 80
    initialValue.push_back(static_cast<unsigned char>(0xED));
    initialValue.push_back(static_cast<unsigned char>(0xA0));
    initialValue.push_back(static_cast<unsigned char>(0x80));

    cache.Put("key", initialValue);
    std::string cachedValue = cache.Get("key");

    // This is a valid case. Invalid code points are supported in this mode.
    BOOST_CHECK_EQUAL(initialValue, cachedValue);

    Ignition::StopAll(false);
}

#endif

BOOST_AUTO_TEST_CASE(StringUtfValid4ByteCodePoint)
{
    Ignite ignite = ignite_test::StartPlatformNode("cache-test.xml", "ServerNode");

    Cache<std::string, std::string> cache = ignite.CreateCache<std::string, std::string>("Test");

    std::string initialValue;

    //    11110xxx 10xxxxxx 10xxxxxx 10xxxxxx |
    // <=             00001 00000001 01001011 | U+1014B
    // <=      000   010000   000101   001011 | U+1014B
    //  = 11110000 10010000 10000101 10001011 | F0 90 85 8B
    initialValue.push_back(static_cast<unsigned char>(0xF0));
    initialValue.push_back(static_cast<unsigned char>(0x90));
    initialValue.push_back(static_cast<unsigned char>(0x85));
    initialValue.push_back(static_cast<unsigned char>(0x8B));

    cache.Put("key", initialValue);
    std::string cachedValue = cache.Get("key");

    // This is a valid UTF-8 code point. Should be supported in default mode.
    BOOST_CHECK_EQUAL(initialValue, cachedValue);

    Ignition::StopAll(false);
}

BOOST_AUTO_TEST_CASE(PutObjectByCppThenByJava)
{
    Ignite ignite = ignite_test::StartPlatformNode("interop.xml", "ServerNode");

    cache::Cache<AffinityKey, AffinityKey> cache = ignite.GetOrCreateCache<AffinityKey, AffinityKey>("default");

    AffinityKey key1(2, 3);
    cache.Put(key1, key1);

    compute::Compute compute = ignite.GetCompute();

    compute.ExecuteJavaTask<int*>(TEST_PUT_AFFINITY_KEY_TASK);

    AffinityKey key2(1, 2);
    AffinityKey val = cache.Get(key2);

    BOOST_CHECK_EQUAL(val.key, 1);
    BOOST_CHECK_EQUAL(val.aff, 2);
}

BOOST_AUTO_TEST_CASE(PutObjectPointerByCppThenByJava)
{
    Ignite ignite = ignite_test::StartPlatformNode("interop.xml", "ServerNode");

    cache::Cache<AffinityKey*, AffinityKey*> cache =
            ignite.GetOrCreateCache<AffinityKey*, AffinityKey*>("default");

    AffinityKey* key1 = new AffinityKey(2, 3);
    cache.Put(key1, key1);

    delete key1;

    compute::Compute compute = ignite.GetCompute();

    compute.ExecuteJavaTask<int*>(TEST_PUT_AFFINITY_KEY_TASK);

    AffinityKey* key2 = new AffinityKey(1, 2);
    AffinityKey* val = cache.Get(key2);

    BOOST_CHECK_EQUAL(val->key, 1);
    BOOST_CHECK_EQUAL(val->aff, 2);

    delete key2;
    delete val;
}

BOOST_AUTO_TEST_SUITE_END()