//MIT License
//
//Copyright (c) 2017 Mindaugas Vinkelis
//
//Permission is hereby granted, free of charge, to any person obtaining a copy
//of this software and associated documentation files (the "Software"), to deal
//in the Software without restriction, including without limitation the rights
//to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
//copies of the Software, and to permit persons to whom the Software is
//furnished to do so, subject to the following conditions:
//
//The above copyright notice and this permission notice shall be included in all
//copies or substantial portions of the Software.
//
//THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
//IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
//FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
//AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
//LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
//OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
//SOFTWARE.

#include <gmock/gmock.h>
#include "serialization_test_utils.h"

#include <bitsery/ext/std_map.h>
#include <bitsery/ext/entropy.h>
#include <unordered_map>
#include <bitsery/traits/string.h>

using StdMap = bitsery::ext::StdMap;

using testing::Eq;

template<typename Container>
Container createData() {
    return {};
}

template<>
std::unordered_map<std::string, MyStruct1> createData<std::unordered_map<std::string, MyStruct1>>() {
    return {
            std::make_pair("some key", MyStruct1{874,456}),
            std::make_pair("other key", MyStruct1{-34,8645}),
            std::make_pair("secret key", MyStruct1{-4878,3468975})
    };
}

template<>
std::unordered_multimap<int32_t, float> createData<std::unordered_multimap<int32_t, float>>() {
    return {
            std::pair<int32_t , float>(545, 45.485f),
            std::pair<int32_t , float>(6748, -7891.5f),
            std::pair<int32_t , float>(845, -457.0f)
    };
}

template<>
std::map<MyEnumClass, MyStruct1> createData<std::map<MyEnumClass, MyStruct1>>() {
    return {
            std::make_pair(MyEnumClass::E3, MyStruct1{874,456}),
            std::make_pair(MyEnumClass::E6, MyStruct1{-34,8645}),
            std::make_pair(MyEnumClass::E2, MyStruct1{-4878,3468975})
    };
}

template<>
std::multimap<int32_t ,int64_t> createData<std::multimap<int32_t ,int64_t>>() {
    return {//these are optimized with range and entropy
            std::pair<int32_t, int64_t>(-45, -984196845ll),
            std::pair<int32_t, int64_t>(54, 1ll),
            std::pair<int32_t, int64_t>(98, 3ll)
    };
}

template<typename T>
class SerializeExtensionStdMap : public testing::Test {
public:
    using TContainer = T;

    const TContainer src = createData<TContainer>();
    TContainer res{};
};

using SerializeExtensionStdMapTypes = ::testing::Types<
        std::unordered_map<std::string, MyStruct1>,
        std::unordered_multimap<int32_t, float>,
        std::map<MyEnumClass , MyStruct1>,
        std::multimap<int32_t ,int64_t>
>;

TYPED_TEST_CASE(SerializeExtensionStdMap, SerializeExtensionStdMapTypes);

namespace bitsery {

    template <typename S>
    void serialize(S& s, std::unordered_map<std::string, MyStruct1>& o) {
        s.ext(o, StdMap{10}, [&s](std::string& key, MyStruct1& value) {
            s.text1b(key, 100);
            s.object(value);
        });
    }

    template <typename S>
    void serialize(S& s, std::unordered_multimap<int32_t, float>& o) {
        s.ext(o, StdMap{10}, [&s](int32_t& key, float& value) {
            s.value4b(key);
            s.value4b(value);
        });
    }

    template <typename S>
    void serialize(S& s, std::map<MyEnumClass , MyStruct1>& o) {
        s.ext(o, StdMap{10}, [&s](MyEnumClass& key, MyStruct1& value) {
            s.value4b(key);
            s.object(value);
        });
    }

    template <typename S>
    void serialize(S& s, std::multimap<int32_t ,int64_t>& o) {
        s.ext(o, StdMap{10}, [&s](int32_t& key, int64_t& value) {
            s.enableBitPacking([&key, &value](typename S::BPEnabledType& sbp) {
                int64_t values[3]{1ll, 2ll, 3ll};
                sbp.ext(key, bitsery::ext::ValueRange<int32_t>{-100,100});
                sbp.ext8b(value, bitsery::ext::Entropy<int64_t[3]>{values});
            });
        });
    }


}

TYPED_TEST(SerializeExtensionStdMap, SerializeAndDeserializeEquals) {
    SerializationContext ctx1;
    ctx1.createSerializer().object(this->src);
    ctx1.createDeserializer().object(this->res);
    EXPECT_THAT(this->res, Eq(this->src));
}