/*
* 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 <Prefab/PrefabTestFixture.h>

#include <AzCore/std/containers/unordered_set.h>
#include <AzToolsFramework/Prefab/Spawnable/SpawnableUtils.h>

namespace UnitTest
{
    using SpawnableCreateTest = PrefabTestFixture;

    TEST_F(SpawnableCreateTest, SpawnableCreate_NoNestingPrefabDom_CreateSucceeds)
    {
        AZStd::vector<AZ::Entity*> entitiesCreated;
        AZStd::unordered_set<AZStd::string> expectedEntityNameSet;
        for (int i = 0; i < 3; i++)
        {
            entitiesCreated.push_back(CreateEntity(AZStd::string::format("Entity_%i", i).c_str()));
            expectedEntityNameSet.insert(entitiesCreated.back()->GetName());
        }

        AZStd::unique_ptr<AzToolsFramework::Prefab::Instance> instance(
            m_prefabSystemComponent->CreatePrefab(entitiesCreated, {}, "test/path"));
        ASSERT_TRUE(instance);

        //Create Spawnable
        auto& prefabDom = m_prefabSystemComponent->FindTemplateDom(instance->GetTemplateId());
        auto spawnable = ::AzToolsFramework::Prefab::SpawnableUtils::CreateSpawnable(prefabDom);

        EXPECT_EQ(spawnable.GetEntities().size(), 3);
        const auto& spawnableEntities = spawnable.GetEntities();
        AZStd::unordered_set<AZStd::string> actualEntityNameSet;

        for (int i = 0; i < 3; i++)
        {
            actualEntityNameSet.insert(spawnableEntities[i]->GetName());
        }

        EXPECT_EQ(expectedEntityNameSet, actualEntityNameSet);
    }

    TEST_F(SpawnableCreateTest, SpawnableCreate_TripleNestingPrefabDom_CreateSucceeds)
    {
        AZStd::vector<AZ::Entity*> entitiesCreated;
        AZStd::unordered_set<AZStd::string> expectedEntityNameSet;
        for (int i = 0; i < 3; i++)
        {
            entitiesCreated.push_back(CreateEntity(AZStd::string::format("Entity_%i", i).c_str()));
            expectedEntityNameSet.insert(entitiesCreated.back()->GetName());
        }

        // Build a 3 level deep nested Template
        AZStd::unique_ptr<AzToolsFramework::Prefab::Instance> firstInstance(
            m_prefabSystemComponent->CreatePrefab({ entitiesCreated[0] }, {}, "test/path1"));
        ASSERT_TRUE(firstInstance);

        AZStd::unique_ptr<AzToolsFramework::Prefab::Instance> secondInstance(
            m_prefabSystemComponent->CreatePrefab({ entitiesCreated[1] }, MakeInstanceList(AZStd::move(firstInstance)), "test/path2"));
        ASSERT_TRUE(secondInstance);

        AZStd::unique_ptr<AzToolsFramework::Prefab::Instance> thirdInstance(
            m_prefabSystemComponent->CreatePrefab({ entitiesCreated[2] }, MakeInstanceList(AZStd::move(secondInstance)), "test/path3"));
        ASSERT_TRUE(thirdInstance);

        //Create Spawnable
        auto& prefabDom = m_prefabSystemComponent->FindTemplateDom(thirdInstance->GetTemplateId());
        auto spawnable = ::AzToolsFramework::Prefab::SpawnableUtils::CreateSpawnable(prefabDom);

        EXPECT_EQ(spawnable.GetEntities().size(), 3);
        const auto& spawnableEntities = spawnable.GetEntities();
        AZStd::unordered_set<AZStd::string> actualEntityNameSet;

        for (int i = 0; i < 3; i++)
        {
            actualEntityNameSet.insert(spawnableEntities[i]->GetName());
        }

        EXPECT_EQ(expectedEntityNameSet, actualEntityNameSet);
    }
}