/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/

#include <AzNetworking/DataStructures/RingBufferBitset.h>
#include <AzCore/UnitTest/TestTypes.h>

namespace UnitTest
{
    AzNetworking::RingbufferBitset<128> test;

    TEST(RingbufferBitset, Push2Bits)
    {
        test.PushBackBits(2);

        test.SetBit(0, true);
        test.SetBit(1, true);

        EXPECT_TRUE(test.GetBit(0));
        EXPECT_TRUE(test.GetBit(1));
    }

    TEST(RingbufferBitset, Push4Bits)
    {
        test.PushBackBits(2);

        EXPECT_FALSE(test.GetBit(0));
        EXPECT_FALSE(test.GetBit(1));
        EXPECT_TRUE (test.GetBit(2));
        EXPECT_TRUE (test.GetBit(3));
    }

    TEST(RingbufferBitset, Push44Bits)
    {
        test.PushBackBits(40);
        test.SetBit(41, true);

        EXPECT_FALSE(test.GetBit(40));
        EXPECT_TRUE (test.GetBit(41));
        EXPECT_TRUE (test.GetBit(42));
        EXPECT_TRUE (test.GetBit(43));
    }

    TEST(RingbufferBitset, Push84Bits)
    {
        test.PushBackBits(40);
        test.SetBit(80, true);

        EXPECT_TRUE(test.GetBit(80));
        EXPECT_TRUE(test.GetBit(81));
        EXPECT_TRUE(test.GetBit(82));
        EXPECT_TRUE(test.GetBit(83));
    }

    TEST(RingbufferBitset, Push124Bits)
    {
        test.PushBackBits(40);

        EXPECT_TRUE(test.GetBit(120));
        EXPECT_TRUE(test.GetBit(121));
        EXPECT_TRUE(test.GetBit(122));
        EXPECT_TRUE(test.GetBit(123));
    }

    TEST(RingbufferBitset, TestWrap)
    {
        // Bits should reset to false after wrapping and falling out of our window
        test.PushBackBits(40);

        EXPECT_FALSE(test.GetBit(160 - 128));
        EXPECT_FALSE(test.GetBit(161 - 128));
        EXPECT_FALSE(test.GetBit(162 - 128));
        EXPECT_FALSE(test.GetBit(163 - 128));
    }
}