// Copyright (c) 2019 - 2020 by Robert Bosch GmbH. All rights reserved.
// Copyright (c) 2020 - 2021 by Apex.AI Inc. All rights reserved.
//
// Licensed 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.
//
// SPDX-License-Identifier: Apache-2.0

#include "test.hpp"

#include "iceoryx_hoofs/concurrent/lockfree_queue.hpp"
#include "iceoryx_hoofs/concurrent/resizeable_lockfree_queue.hpp"

// We test the common functionality of LockFreeQueue and ResizableLockFreeQueue here
// in typed tests to reduce code duplication.

namespace
{
using namespace ::testing;

// use a non-POD type for testing (just a boxed version of int).
// We use implicit conversions of int to Integer to be able use the same test structure
// for Integer and int (primarily for tryPush).
// This allows testing PODs and Custom Types with the same test structure.
struct Integer
{
    Integer(int value = 0)
        : value(value)
    {
    }

    int value{0};

    // so that it behaves like an int for comparison purposes
    operator int() const
    {
        return value;
    }
};

template <typename Config>
class LockFreeQueueTest : public ::testing::Test
{
  protected:
    LockFreeQueueTest()
    {
    }

    ~LockFreeQueueTest()
    {
    }

    void SetUp()
    {
        // reduce capacity before running the tests if required by config
        setCapacity<Config>();
    }

    void TearDown()
    {
    }

    void fillQueue(int start = 0)
    {
        int data{start};
        for (uint64_t i = 0; i < queue.capacity(); ++i)
        {
            queue.tryPush(data);
            data++;
        }
    }

    // only set Capacity if DynamicCapacity is smaller
    // since some queue types may not even provide the option to call setCapacity
    // this must be done at compile time
    // (the additional template indirection via Config_ is required for SFINAE)
    template <typename Config_>
    typename std::enable_if<Config_::SetCapacityInitially, void>::type setCapacity()
    {
        queue.setCapacity(Config::DynamicCapacity);
    }

    template <typename Config_>
    typename std::enable_if<!Config_::SetCapacityInitially, void>::type setCapacity()
    {
    }

    using Queue = typename Config::QueueType;
    Queue queue;
};

template <size_t Capacity>
using IntegerQueue = iox::concurrent::LockFreeQueue<Integer, Capacity>;

// define the test configurations with varying types, capacities and dynamically reduced capacities

template <template <typename, uint64_t> class QueueType_,
          typename ElementType_,
          uint64_t Capacity_,
          uint64_t DynamicCapacity_ = Capacity_>
struct Config
{
    static constexpr uint64_t Capacity = Capacity_;
    static constexpr uint64_t DynamicCapacity = DynamicCapacity_;
    static_assert(DynamicCapacity <= Capacity, "DynamicCapacity can be at most Capacity");

    using QueueType = QueueType_<ElementType_, Capacity>;
    static constexpr bool SetCapacityInitially = DynamicCapacity < Capacity;
};

template <typename T, uint64_t C>
using LFQueue = iox::concurrent::LockFreeQueue<T, C>;

template <typename T, uint64_t C>
using RLFQueue = iox::concurrent::ResizeableLockFreeQueue<T, C>;


template <template <typename, uint64_t> class QueueType, typename ElementType, uint64_t Capacity>
using Full = Config<QueueType, ElementType, Capacity>;

template <template <typename, uint64_t> class QueueType, typename ElementType, uint64_t Capacity>
using AlmostFull = Config<QueueType, ElementType, Capacity, (Capacity > 1) ? (Capacity - 1) : Capacity>;

template <template <typename, uint64_t> class QueueType, typename ElementType, uint64_t Capacity>
using HalfFull = Config<QueueType, ElementType, Capacity, (Capacity > 1) ? (Capacity / 2) : Capacity>;

template <template <typename, uint64_t> class QueueType, typename ElementType, uint64_t Capacity>
using AlmostEmpty = Config<QueueType, ElementType, Capacity, 1>;

// configs of the lockfree queue without resize
using LFFull1 = Full<LFQueue, int, 1>;
using LFFull2 = Full<LFQueue, int, 1000>;
using LFFull3 = Full<LFQueue, Integer, 100>;

// configs of the resizeable lockfree queue
using Full1 = Full<RLFQueue, Integer, 1>;
using Full2 = Full<RLFQueue, Integer, 10>;
using Full3 = Full<RLFQueue, int, 1000>;

using AlmostFull1 = AlmostFull<RLFQueue, Integer, 10>;
using AlmostFull2 = AlmostFull<RLFQueue, int, 1000>;

using HalfFull1 = HalfFull<RLFQueue, Integer, 10>;
using HalfFull2 = HalfFull<RLFQueue, int, 1000>;

using AlmostEmpty1 = AlmostEmpty<RLFQueue, Integer, 10>;
using AlmostEmpty2 = AlmostEmpty<RLFQueue, int, 1000>;

typedef ::testing::Types<LFFull1,
                         LFFull2,
                         LFFull3,
                         Full1,
                         Full2,
                         Full3,
                         AlmostFull1,
                         AlmostFull2,
                         HalfFull1,
                         HalfFull2,
                         AlmostEmpty1,
                         AlmostEmpty2>
    TestConfigs;

TYPED_TEST_SUITE(LockFreeQueueTest, TestConfigs);


TEST(LockFreeQueueTest, capacityIsConsistent)
{
    ::testing::Test::RecordProperty("TEST_ID", "0b56ef76-2eac-4174-9999-e26495758e6a");
    constexpr uint64_t CAPACITY{37};
    IntegerQueue<CAPACITY> q;
    EXPECT_EQ(q.capacity(), CAPACITY);
}

TYPED_TEST(LockFreeQueueTest, constructedQueueIsEmpty)
{
    ::testing::Test::RecordProperty("TEST_ID", "9bb8a86e-c3d0-44ef-9fb7-999f50f0c4ac");
    auto& q = this->queue;
    EXPECT_TRUE(q.empty());
    EXPECT_EQ(q.size(), 0);
}

TYPED_TEST(LockFreeQueueTest, pushAndPopSingleElement)
{
    ::testing::Test::RecordProperty("TEST_ID", "381e13c3-d5dc-40c1-b4fd-65634578bdc8");
    auto& q = this->queue;

    int data = 42;
    EXPECT_TRUE(q.tryPush(data));
    EXPECT_EQ(q.size(), 1);

    auto x = q.pop();
    ASSERT_TRUE(x.has_value());
    EXPECT_EQ(x.value(), 42);
    EXPECT_EQ(q.size(), 0);
}

TYPED_TEST(LockFreeQueueTest, popFromEmptyQueueReturnsNothing)
{
    ::testing::Test::RecordProperty("TEST_ID", "f9d4f232-e9e7-4089-bf61-2696a4bdc8f3");
    auto& q = this->queue;

    int data = 24;
    q.tryPush(data);
    q.pop();
    EXPECT_FALSE(q.pop().has_value());
    EXPECT_EQ(q.size(), 0);
}

TYPED_TEST(LockFreeQueueTest, tryPushUntilFullCapacityIsUsed)
{
    ::testing::Test::RecordProperty("TEST_ID", "6c6bb533-aab5-46bc-8368-977fc2503a74");
    auto& q = this->queue;
    auto capacity = q.capacity();

    int data{0};
    for (uint64_t i = 0; i < capacity; ++i)
    {
        EXPECT_EQ(q.size(), i);
        EXPECT_TRUE(q.tryPush(data));
        data++;
    }

    EXPECT_EQ(q.size(), capacity);
}

TYPED_TEST(LockFreeQueueTest, tryPushInFullQueueFails)
{
    ::testing::Test::RecordProperty("TEST_ID", "0b793a3e-1e47-46d9-91c2-4967569b508b");
    auto& q = this->queue;
    this->fillQueue(38);
    int data{37};
    EXPECT_FALSE(q.tryPush(data));
}

TYPED_TEST(LockFreeQueueTest, poppedElementsAreInFifoOrder)
{
    ::testing::Test::RecordProperty("TEST_ID", "fecc997a-ae8e-49d3-a1ea-7ec07b5a93ed");
    auto& q = this->queue;
    auto capacity = q.capacity();

    // scramble the start value to avoid false positives
    // due to memory values of previous tests on the stack ...
    int value = 73;
    this->fillQueue(value);

    for (uint64_t i = capacity; i > 0; --i)
    {
        EXPECT_EQ(q.size(), i);
        auto x = q.pop();
        ASSERT_TRUE(x.has_value());
        EXPECT_EQ(x.value(), value);
        ++value;
    }
    EXPECT_FALSE(q.pop().has_value());
    EXPECT_EQ(q.size(), 0);
}

TYPED_TEST(LockFreeQueueTest, pushDoesNotOverflowIfQueueIsNotFull)
{
    ::testing::Test::RecordProperty("TEST_ID", "2096033c-5631-480e-8b9c-a8b472721cdb");
    auto& q = this->queue;
    auto capacity = q.capacity();

    int start{66};
    int data{start};
    for (uint64_t i = 0; i < capacity; ++i)
    {
        auto x = q.push(data);
        EXPECT_FALSE(x.has_value());
        data++;
    }
}

TYPED_TEST(LockFreeQueueTest, pushReturnsOldestElementOnOverflow)
{
    ::testing::Test::RecordProperty("TEST_ID", "df40eac8-11ba-4352-aef0-c1c4785a43f8");
    auto& q = this->queue;
    auto capacity = q.capacity();

    int start{666};
    this->fillQueue(start);

    int data{-start};
    for (uint64_t i = 0; i < capacity; ++i)
    {
        auto x = q.push(data);
        ASSERT_TRUE(x.has_value());
        EXPECT_EQ(x.value(), start);
        data--;
        start++;
    }
}

TYPED_TEST(LockFreeQueueTest, pushInsertsInFifoOrder)
{
    ::testing::Test::RecordProperty("TEST_ID", "18698a63-de51-407a-a2f0-dce591c92223");
    auto& q = this->queue;
    auto capacity = q.capacity();

    int start{69};
    this->fillQueue(start);

    int value{-start};
    for (uint64_t i = 0; i < capacity; ++i)
    {
        q.push(value);
        value--;
    }

    value = -start;
    for (uint64_t i = 0; i < capacity; ++i)
    {
        auto x = q.pop();
        ASSERT_TRUE(x.has_value());
        EXPECT_EQ(x.value(), value);
        value--;
    }
}

TYPED_TEST(LockFreeQueueTest, checkEmptynessAfterOneElementWasPushedandPopped)
{
    ::testing::Test::RecordProperty("TEST_ID", "ef49cb74-9631-4804-b040-36cc28c2abfc");
    auto& q = this->queue;

    q.tryPush(37);
    auto x = q.pop();

    EXPECT_TRUE(q.empty());
    EXPECT_EQ(q.size(), 0);
}

TYPED_TEST(LockFreeQueueTest, checkEmptynessAfterFullQueueWasEmptied)
{
    ::testing::Test::RecordProperty("TEST_ID", "fb0d2c27-25e6-41c6-a51a-909a01ec0052");
    auto& q = this->queue;
    auto capacity = q.capacity();

    int start{73};
    this->fillQueue(start);

    for (uint64_t i = 0; i < capacity; ++i)
    {
        auto x = q.pop();
    }

    EXPECT_TRUE(q.empty());
    EXPECT_EQ(q.size(), 0);
}

} // namespace