alexpete
82 lines
2.3 KiB
C++
82 lines
2.3 KiB
C++
/*
|
|
* 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 "Maestro_precompiled.h"
|
|
|
|
#include <AzTest/AzTest.h>
|
|
#include <AzCore/UnitTest/UnitTest.h>
|
|
#include <Mocks/ITimerMock.h>
|
|
#include <Mocks/ICryPakMock.h>
|
|
#include <Mocks/IConsoleMock.h>
|
|
#include <AzCore/Memory/OSAllocator.h>
|
|
|
|
#include <ISystem.h>
|
|
|
|
using ::testing::NiceMock;
|
|
|
|
class MaestroTestEnvironment
|
|
: public AZ::Test::ITestEnvironment
|
|
, public UnitTest::TraceBusRedirector
|
|
{
|
|
public:
|
|
AZ_TEST_CLASS_ALLOCATOR(MaestroTestEnvironment);
|
|
|
|
virtual ~MaestroTestEnvironment()
|
|
{}
|
|
|
|
protected:
|
|
|
|
struct MockHolder
|
|
{
|
|
AZ_TEST_CLASS_ALLOCATOR(MockHolder);
|
|
|
|
NiceMock<TimerMock> timer;
|
|
NiceMock<CryPakMock> pak;
|
|
NiceMock<ConsoleMock> console;
|
|
};
|
|
|
|
void SetupEnvironment() override
|
|
{
|
|
AZ::AllocatorInstance<AZ::SystemAllocator>::Create();
|
|
|
|
// Mocks need to be destroyed before the allocators are destroyed,
|
|
// but if they are member variables, they get destroyed *after*
|
|
// TeardownEnvironment when this Environment class is destroyed
|
|
// by the GoogleTest framework.
|
|
//
|
|
// Mocks also do not have any public destroy or release method to
|
|
// manage their lifetime, so this solution manages the lifetime
|
|
// and ordering via the heap.
|
|
m_mocks = new MockHolder();
|
|
m_stubEnv.pTimer = &m_mocks->timer;
|
|
m_stubEnv.pCryPak = &m_mocks->pak;
|
|
m_stubEnv.pConsole = &m_mocks->console;
|
|
gEnv = &m_stubEnv;
|
|
|
|
BusConnect();
|
|
}
|
|
|
|
void TeardownEnvironment() override
|
|
{
|
|
BusDisconnect();
|
|
// Destroy mocks before AZ allocators
|
|
delete m_mocks;
|
|
|
|
AZ::AllocatorInstance<AZ::SystemAllocator>::Destroy();
|
|
}
|
|
|
|
private:
|
|
SSystemGlobalEnvironment m_stubEnv;
|
|
MockHolder* m_mocks;
|
|
};
|
|
|
|
AZ_UNIT_TEST_HOOK(new MaestroTestEnvironment)
|