You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
265 lines
12 KiB
C++
265 lines
12 KiB
C++
/*
|
|
* Copyright (c) Contributors to the Open 3D Engine Project.
|
|
* For complete copyright and license terms please see the LICENSE at the root of this distribution.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 OR MIT
|
|
*
|
|
*/
|
|
|
|
#include <AzCore/Component/TransformBus.h>
|
|
#include <AzCore/UnitTest/TestTypes.h>
|
|
#include <AzTest/AzTest.h>
|
|
#include <AzToolsFramework/Entity/EditorEntityHelpers.h>
|
|
#include <AzToolsFramework/Entity/EditorEntityInfoBus.h>
|
|
#include <AzToolsFramework/ToolsComponents/EditorVisibilityBus.h>
|
|
#include <AzToolsFramework/UnitTest/AzToolsFrameworkTestHelpers.h>
|
|
|
|
#include <Plugins/ComponentEntityEditorPlugin/Objects/ComponentEntityObject.h>
|
|
|
|
using namespace AzToolsFramework;
|
|
|
|
namespace UnitTest
|
|
{
|
|
class ComponentEntityObjectVisibilityFixture
|
|
: public ToolsApplicationFixture
|
|
, private EditorEntityVisibilityNotificationBus::Router
|
|
, private EditorEntityInfoNotificationBus::Handler
|
|
{
|
|
public:
|
|
void SetUpEditorFixtureImpl() override
|
|
{
|
|
EditorEntityVisibilityNotificationBus::Router::BusRouterConnect();
|
|
EditorEntityInfoNotificationBus::Handler::BusConnect();
|
|
}
|
|
|
|
void TearDownEditorFixtureImpl() override
|
|
{
|
|
EditorEntityInfoNotificationBus::Handler::BusDisconnect();
|
|
EditorEntityVisibilityNotificationBus::Router::BusRouterDisconnect();
|
|
}
|
|
|
|
TestEditorActions m_editorActions;
|
|
AZ::EntityId m_layerId;
|
|
|
|
private:
|
|
// EditorEntityVisibilityNotificationBus ...
|
|
void OnEntityVisibilityChanged(bool /*visibility*/) override
|
|
{
|
|
}
|
|
|
|
// EditorEntityInfoNotificationBus ...
|
|
void OnEntityInfoUpdatedVisibility(AZ::EntityId /*entityId*/, bool /*visible*/) override
|
|
{
|
|
}
|
|
};
|
|
|
|
TEST_F(ComponentEntityObjectVisibilityFixture, ViewportComponentEntityObjectRespectsLayerVisibility)
|
|
{
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Given
|
|
AZ::Entity* entityA = nullptr;
|
|
const AZ::EntityId a = CreateDefaultEditorEntity("A", &entityA);
|
|
AZ::Entity* entityB = nullptr;
|
|
const AZ::EntityId b = CreateDefaultEditorEntity("B", &entityB);
|
|
AZ::Entity* entityC = nullptr;
|
|
const AZ::EntityId c = CreateDefaultEditorEntity("C", &entityC);
|
|
|
|
m_layerId = CreateEditorLayerEntity("Layer");
|
|
|
|
entityA->Deactivate();
|
|
entityB->Deactivate();
|
|
entityC->Deactivate();
|
|
|
|
CComponentEntityObject componentEntityObjectA;
|
|
componentEntityObjectA.AssignEntity(entityA);
|
|
|
|
CComponentEntityObject componentEntityObjectB;
|
|
componentEntityObjectB.AssignEntity(entityB);
|
|
|
|
CComponentEntityObject componentEntityObjectC;
|
|
componentEntityObjectC.AssignEntity(entityC);
|
|
|
|
entityC->Activate();
|
|
entityB->Activate();
|
|
entityA->Activate();
|
|
|
|
AZ::TransformBus::Event(a, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
AZ::TransformBus::Event(b, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
AZ::TransformBus::Event(c, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// When
|
|
SetEntityVisibility(a, false);
|
|
SetEntityVisibility(b, false);
|
|
SetEntityVisibility(c, false);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Then
|
|
EXPECT_TRUE(componentEntityObjectA.IsHidden());
|
|
EXPECT_TRUE(componentEntityObjectB.IsHidden());
|
|
EXPECT_TRUE(componentEntityObjectC.IsHidden());
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// When
|
|
SetEntityVisibility(a, true);
|
|
SetEntityVisibility(b, true);
|
|
SetEntityVisibility(c, true);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Then
|
|
EXPECT_FALSE(componentEntityObjectA.IsHidden());
|
|
EXPECT_FALSE(componentEntityObjectB.IsHidden());
|
|
EXPECT_FALSE(componentEntityObjectC.IsHidden());
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// When
|
|
SetEntityVisibility(m_layerId, false);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Then
|
|
EXPECT_TRUE(componentEntityObjectA.IsHidden());
|
|
EXPECT_TRUE(componentEntityObjectB.IsHidden());
|
|
EXPECT_TRUE(componentEntityObjectC.IsHidden());
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
}
|
|
|
|
TEST_F(ComponentEntityObjectVisibilityFixture, ComponentEntityObjectDoesNotOverrideVisibility)
|
|
{
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Given
|
|
AZ::Entity* entityA = nullptr;
|
|
const AZ::EntityId a = CreateDefaultEditorEntity("A", &entityA);
|
|
AZ::Entity* entityB = nullptr;
|
|
const AZ::EntityId b = CreateDefaultEditorEntity("B", &entityB);
|
|
AZ::Entity* entityC = nullptr;
|
|
const AZ::EntityId c = CreateDefaultEditorEntity("C", &entityC);
|
|
|
|
m_layerId = CreateEditorLayerEntity("Layer");
|
|
|
|
entityA->Deactivate();
|
|
entityB->Deactivate();
|
|
entityC->Deactivate();
|
|
|
|
CComponentEntityObject componentEntityObjectA;
|
|
componentEntityObjectA.AssignEntity(entityA);
|
|
|
|
CComponentEntityObject componentEntityObjectB;
|
|
componentEntityObjectB.AssignEntity(entityB);
|
|
|
|
CComponentEntityObject componentEntityObjectC;
|
|
componentEntityObjectC.AssignEntity(entityC);
|
|
|
|
entityC->Activate();
|
|
entityB->Activate();
|
|
entityA->Activate();
|
|
|
|
AZ::TransformBus::Event(a, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
AZ::TransformBus::Event(b, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
AZ::TransformBus::Event(c, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// When
|
|
SetEntityVisibility(m_layerId, false);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Then
|
|
EXPECT_TRUE(componentEntityObjectA.IsHidden());
|
|
EXPECT_TRUE(componentEntityObjectB.IsHidden());
|
|
EXPECT_TRUE(componentEntityObjectC.IsHidden());
|
|
|
|
EXPECT_TRUE(IsEntitySetToBeVisible(a));
|
|
EXPECT_FALSE(IsEntityVisible(a));
|
|
|
|
EXPECT_TRUE(IsEntitySetToBeVisible(a));
|
|
EXPECT_FALSE(IsEntityVisible(b));
|
|
|
|
EXPECT_TRUE(IsEntitySetToBeVisible(a));
|
|
EXPECT_FALSE(IsEntityVisible(c));
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
}
|
|
|
|
TEST_F(ComponentEntityObjectVisibilityFixture, ViewportComponentEntityObjectRespectsLayerLock)
|
|
{
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Given
|
|
AZ::Entity* entityA = nullptr;
|
|
const AZ::EntityId a = CreateDefaultEditorEntity("A", &entityA);
|
|
AZ::Entity* entityB = nullptr;
|
|
const AZ::EntityId b = CreateDefaultEditorEntity("B", &entityB);
|
|
AZ::Entity* entityC = nullptr;
|
|
const AZ::EntityId c = CreateDefaultEditorEntity("C", &entityC);
|
|
|
|
m_layerId = CreateEditorLayerEntity("Layer");
|
|
|
|
entityA->Deactivate();
|
|
entityB->Deactivate();
|
|
entityC->Deactivate();
|
|
|
|
CComponentEntityObject componentEntityObjectA;
|
|
componentEntityObjectA.AssignEntity(entityA);
|
|
|
|
CComponentEntityObject componentEntityObjectB;
|
|
componentEntityObjectB.AssignEntity(entityB);
|
|
|
|
CComponentEntityObject componentEntityObjectC;
|
|
componentEntityObjectC.AssignEntity(entityC);
|
|
|
|
entityC->Activate();
|
|
entityB->Activate();
|
|
entityA->Activate();
|
|
|
|
AZ::TransformBus::Event(a, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
AZ::TransformBus::Event(b, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
AZ::TransformBus::Event(c, &AZ::TransformBus::Events::SetParent, m_layerId);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// When
|
|
SetEntityLockState(a, true);
|
|
SetEntityLockState(b, true);
|
|
SetEntityLockState(c, true);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Then
|
|
EXPECT_TRUE(componentEntityObjectA.IsFrozen());
|
|
EXPECT_TRUE(componentEntityObjectB.IsFrozen());
|
|
EXPECT_TRUE(componentEntityObjectC.IsFrozen());
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// When
|
|
SetEntityLockState(a, false);
|
|
SetEntityLockState(b, false);
|
|
SetEntityLockState(c, false);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Then
|
|
EXPECT_FALSE(componentEntityObjectA.IsFrozen());
|
|
EXPECT_FALSE(componentEntityObjectB.IsFrozen());
|
|
EXPECT_FALSE(componentEntityObjectC.IsFrozen());
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// When
|
|
SetEntityLockState(m_layerId, true);
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Then
|
|
EXPECT_TRUE(componentEntityObjectA.IsFrozen());
|
|
EXPECT_TRUE(componentEntityObjectB.IsFrozen());
|
|
EXPECT_TRUE(componentEntityObjectC.IsFrozen());
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
}
|
|
} // namespace UnitTest
|