/*
 *
 *    Copyright (c) 2024 Project CHIP Authors
 *    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.
 */

#include <stdint.h>

#include <app/TestEventTriggerDelegate.h>
#include <lib/core/StringBuilderAdapters.h>
#include <lib/support/Span.h>
#include <pw_unit_test/framework.h>

using namespace chip;

namespace {

class TestEventHandler : public TestEventTriggerHandler
{
public:
    TestEventHandler() = delete;

    explicit TestEventHandler(uint64_t supportedEventTriggerValue) : mSupportedEventTriggerValue(supportedEventTriggerValue) {}

    CHIP_ERROR HandleEventTrigger(uint64_t eventTrigger) override
    {
        if (eventTrigger == mSupportedEventTriggerValue)
        {
            ++mCount;
            return CHIP_NO_ERROR;
        }
        return CHIP_ERROR_INVALID_ARGUMENT;
    }

    int GetCount() const { return mCount; }
    void ClearCount() { mCount = 0; }

private:
    uint64_t mSupportedEventTriggerValue;
    int mCount = 0;
};

class TestEventDelegate : public TestEventTriggerDelegate
{
public:
    explicit TestEventDelegate(const ByteSpan & enableKey) : mEnableKey(enableKey) {}

    bool DoesEnableKeyMatch(const ByteSpan & enableKey) const override
    {
        return !mEnableKey.empty() && mEnableKey.data_equal(enableKey);
    }

private:
    ByteSpan mEnableKey;
};

TEST(TestTestEventTriggerDelegate, TestKeyChecking)
{
    const uint8_t kTestKey[16]       = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 };
    const uint8_t kBadKey[16]        = { 255, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 };
    const uint8_t kDiffLenBadKey[17] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16 };
    TestEventDelegate delegate{ ByteSpan{ kTestKey } };

    EXPECT_TRUE(delegate.DoesEnableKeyMatch(ByteSpan{ kTestKey }));
    EXPECT_FALSE(delegate.DoesEnableKeyMatch(ByteSpan{ kBadKey }));
    EXPECT_FALSE(delegate.DoesEnableKeyMatch(ByteSpan{ kDiffLenBadKey }));
    EXPECT_FALSE(delegate.DoesEnableKeyMatch(ByteSpan{}));
}

TEST(TestTestEventTriggerDelegate, TestHandlerManagement)
{
    const uint8_t kTestKey[16] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 };

    TestEventDelegate delegate{ ByteSpan{ kTestKey } };

    TestEventHandler event1Handler{ 1 };
    TestEventHandler event2Handler{ 2 };

    // Add 2, check 2 works 1 doesn't.
    EXPECT_NE(delegate.HandleEventTriggers(1), CHIP_NO_ERROR);

    EXPECT_EQ(delegate.AddHandler(&event2Handler), CHIP_NO_ERROR);
    EXPECT_NE(delegate.AddHandler(&event2Handler), CHIP_NO_ERROR);

    EXPECT_EQ(delegate.HandleEventTriggers(2), CHIP_NO_ERROR);
    EXPECT_NE(delegate.HandleEventTriggers(1), CHIP_NO_ERROR);
    EXPECT_EQ(delegate.HandleEventTriggers(2), CHIP_NO_ERROR);

    EXPECT_EQ(event1Handler.GetCount(), 0);
    EXPECT_EQ(event2Handler.GetCount(), 2);

    event1Handler.ClearCount();
    event2Handler.ClearCount();

    // Add 1, check 1 and 2 work.
    EXPECT_EQ(delegate.AddHandler(&event1Handler), CHIP_NO_ERROR);
    EXPECT_NE(delegate.AddHandler(&event1Handler), CHIP_NO_ERROR);

    EXPECT_EQ(delegate.HandleEventTriggers(1), CHIP_NO_ERROR);
    EXPECT_EQ(delegate.HandleEventTriggers(2), CHIP_NO_ERROR);
    EXPECT_EQ(delegate.HandleEventTriggers(1), CHIP_NO_ERROR);

    EXPECT_EQ(event1Handler.GetCount(), 2);
    EXPECT_EQ(event2Handler.GetCount(), 1);

    event1Handler.ClearCount();
    event2Handler.ClearCount();

    // Remove 2, check 1 works.
    delegate.RemoveHandler(&event2Handler);

    EXPECT_EQ(delegate.HandleEventTriggers(1), CHIP_NO_ERROR);
    EXPECT_NE(delegate.HandleEventTriggers(2), CHIP_NO_ERROR);

    EXPECT_EQ(event1Handler.GetCount(), 1);
    EXPECT_EQ(event2Handler.GetCount(), 0);

    // Remove again, should be NO-OP.
    delegate.RemoveHandler(&event2Handler);
    EXPECT_NE(delegate.HandleEventTriggers(2), CHIP_NO_ERROR);
    EXPECT_EQ(event2Handler.GetCount(), 0);

    event1Handler.ClearCount();
    event2Handler.ClearCount();

    // Add 2 again, check 1 and 2 work.
    EXPECT_EQ(delegate.AddHandler(&event2Handler), CHIP_NO_ERROR);
    EXPECT_NE(delegate.AddHandler(&event2Handler), CHIP_NO_ERROR);

    EXPECT_EQ(delegate.HandleEventTriggers(1), CHIP_NO_ERROR);
    EXPECT_EQ(delegate.HandleEventTriggers(2), CHIP_NO_ERROR);

    EXPECT_EQ(event1Handler.GetCount(), 1);
    EXPECT_EQ(event2Handler.GetCount(), 1);

    event1Handler.ClearCount();
    event2Handler.ClearCount();

    // Remove all handlers, check neither works.
    delegate.ClearAllHandlers();

    EXPECT_NE(delegate.HandleEventTriggers(1), CHIP_NO_ERROR);
    EXPECT_NE(delegate.HandleEventTriggers(2), CHIP_NO_ERROR);

    EXPECT_EQ(event1Handler.GetCount(), 0);
    EXPECT_EQ(event2Handler.GetCount(), 0);

    // Add a handler at the end: having it remaining should not cause crashes/leaks.
    EXPECT_EQ(delegate.AddHandler(&event2Handler), CHIP_NO_ERROR);
}

} // namespace