diff --git a/Code/Tools/AssetProcessor/native/tests/assetmanager/AssetProcessorManagerTest.cpp b/Code/Tools/AssetProcessor/native/tests/assetmanager/AssetProcessorManagerTest.cpp index 48a86dc5f6..da6e995c97 100644 --- a/Code/Tools/AssetProcessor/native/tests/assetmanager/AssetProcessorManagerTest.cpp +++ b/Code/Tools/AssetProcessor/native/tests/assetmanager/AssetProcessorManagerTest.cpp @@ -9,6 +9,7 @@ #include "AssetProcessorManagerTest.h" #include "native/AssetManager/PathDependencyManager.h" #include +#include #include #include @@ -4130,11 +4131,21 @@ struct LockedFileTest MOCK_METHOD2(SendResponse, size_t (unsigned, const AzFramework::AssetSystem::BaseAssetProcessorMessage&)); MOCK_METHOD1(RemoveResponseHandler, void (unsigned)); - size_t Send(unsigned, const AzFramework::AssetSystem::BaseAssetProcessorMessage&) override + size_t Send(unsigned, const AzFramework::AssetSystem::BaseAssetProcessorMessage& message) override { - if(m_callback) + using SourceFileNotificationMessage = AzToolsFramework::AssetSystem::SourceFileNotificationMessage; + switch (message.GetMessageType()) { - m_callback(); + case SourceFileNotificationMessage::MessageType: + if (const auto sourceFileMessage = azrtti_cast(&message); + sourceFileMessage != nullptr && sourceFileMessage->m_type == SourceFileNotificationMessage::NotificationType::FileRemoved + && m_callback) + { + m_callback(); + } + break; + default: + break; } return 0;