AssetCatalog.cpp 61 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224
  1. /*
  2. * Copyright (c) Contributors to the Open 3D Engine Project.
  3. * For complete copyright and license terms please see the LICENSE at the root of this distribution.
  4. *
  5. * SPDX-License-Identifier: Apache-2.0 OR MIT
  6. *
  7. */
  8. #include <FileIOBaseTestTypes.h>
  9. #include <AzCore/Asset/AssetManagerBus.h>
  10. #include <AzCore/Asset/AssetTypeInfoBus.h>
  11. #include <AzCore/Component/TickBus.h>
  12. #include <AzCore/Interface/Interface.h>
  13. #include <AzCore/IO/Streamer/Streamer.h>
  14. #include <AzCore/IO/Streamer/StreamerComponent.h>
  15. #include <AzCore/Jobs/JobFunction.h>
  16. #include <AzCore/Jobs/JobManager.h>
  17. #include <AzCore/Jobs/JobContext.h>
  18. #include <AzCore/Math/Uuid.h>
  19. #include <AzCore/Memory/Memory.h>
  20. #include <AzCore/Memory/PoolAllocator.h>
  21. #include <AzCore/Settings/SettingsRegistryMergeUtils.h>
  22. #include <AzCore/std/containers/vector.h>
  23. #include <AzCore/std/parallel/binary_semaphore.h>
  24. #include <AzCore/UnitTest/TestTypes.h>
  25. #include <AzCore/UserSettings/UserSettingsComponent.h>
  26. #include <AzFramework/Asset/AssetCatalog.h>
  27. #include <AzFramework/Asset/AssetProcessorMessages.h>
  28. #include <AzFramework/Asset/GenericAssetHandler.h>
  29. #include <AzFramework/Asset/NetworkAssetNotification_private.h>
  30. #include <AzFramework/Application/Application.h>
  31. namespace UnitTest
  32. {
  33. namespace
  34. {
  35. AZ::Data::AssetId asset1;
  36. AZ::Data::AssetId asset2;
  37. AZ::Data::AssetId asset3;
  38. AZ::Data::AssetId asset4;
  39. AZ::Data::AssetId asset5;
  40. bool Search(const AZStd::vector<AZ::Data::ProductDependency>& dependencies, const AZ::Data::AssetId& assetId)
  41. {
  42. for (const auto& dependency : dependencies)
  43. {
  44. if (dependency.m_assetId == assetId)
  45. {
  46. return true;
  47. }
  48. }
  49. return false;
  50. }
  51. }
  52. class AssetCatalogDependencyTest
  53. : public LeakDetectionFixture
  54. {
  55. AzFramework::AssetCatalog* m_assetCatalog;
  56. public:
  57. const char* path1 = "SomeFolder/Asset1Path";
  58. const char* path2 = "SomeFolder/Asset2Path";
  59. const char* path3 = "OtherFolder/Asset3Path";
  60. const char* path4 = "OtherFolder/Asset4Path";
  61. const char* path5 = "OtherFolder/Asset5Path";
  62. void SetUp() override
  63. {
  64. AZ::Data::AssetManager::Descriptor desc;
  65. AZ::Data::AssetManager::Create(desc);
  66. m_assetCatalog = aznew AzFramework::AssetCatalog();
  67. m_assetCatalog->StartMonitoringAssets();
  68. AzFramework::AssetSystem::NetworkAssetUpdateInterface* notificationInterface = AZ::Interface<AzFramework::AssetSystem::NetworkAssetUpdateInterface>::Get();
  69. ASSERT_NE(notificationInterface, nullptr);
  70. asset1 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  71. asset2 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  72. asset3 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  73. asset4 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  74. asset5 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  75. AZ::Data::AssetInfo info1, info2, info3, info4, info5;
  76. info1.m_relativePath = path1;
  77. info1.m_sizeBytes = 1; // Need to initialize m_sizeBytes to non-zero number to avoid a FileIO call
  78. info2.m_relativePath = path2;
  79. info2.m_sizeBytes = 1;
  80. info3.m_relativePath = path3;
  81. info3.m_sizeBytes = 1;
  82. info4.m_relativePath = path4;
  83. info4.m_sizeBytes = 1;
  84. info5.m_relativePath = path5;
  85. info5.m_sizeBytes = 1;
  86. // asset1 -> asset2 -> asset3 -> asset5
  87. // --> asset4
  88. {
  89. m_assetCatalog->RegisterAsset(asset1, info1);
  90. AzFramework::AssetSystem::AssetNotificationMessage message("test", AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  91. message.m_assetId = asset1;
  92. message.m_dependencies.emplace_back(asset2, 0);
  93. notificationInterface->AssetChanged({ message });
  94. }
  95. {
  96. m_assetCatalog->RegisterAsset(asset2, info2);
  97. AzFramework::AssetSystem::AssetNotificationMessage message("test", AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  98. message.m_assetId = asset2;
  99. message.m_dependencies.emplace_back(asset3, 0);
  100. message.m_dependencies.emplace_back(asset4, 0);
  101. notificationInterface->AssetChanged({ message });
  102. }
  103. {
  104. m_assetCatalog->RegisterAsset(asset3, info3);
  105. AzFramework::AssetSystem::AssetNotificationMessage message("test", AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  106. message.m_assetId = asset3;
  107. message.m_dependencies.emplace_back(asset5, 0);
  108. notificationInterface->AssetChanged({ message });
  109. }
  110. {
  111. m_assetCatalog->RegisterAsset(asset4, info4);
  112. AzFramework::AssetSystem::AssetNotificationMessage message("test", AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  113. message.m_assetId = asset4;
  114. notificationInterface->AssetChanged({ message });
  115. }
  116. {
  117. m_assetCatalog->RegisterAsset(asset5, info5);
  118. AzFramework::AssetSystem::AssetNotificationMessage message("test", AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  119. message.m_assetId = asset5;
  120. notificationInterface->AssetChanged({ message });
  121. }
  122. }
  123. void TearDown() override
  124. {
  125. delete m_assetCatalog;
  126. AzFramework::LegacyAssetEventBus::ClearQueuedEvents();
  127. AZ::TickBus::ClearQueuedEvents();
  128. AZ::Data::AssetManager::Destroy();
  129. }
  130. void CheckDirectDependencies(AZ::Data::AssetId assetId, AZStd::initializer_list<AZ::Data::AssetId> expectedDependencies)
  131. {
  132. AZ::Outcome<AZStd::vector<AZ::Data::ProductDependency>, AZStd::string> result = AZ::Failure<AZStd::string>("No response");
  133. AZ::Data::AssetCatalogRequestBus::BroadcastResult(result, &AZ::Data::AssetCatalogRequestBus::Events::GetDirectProductDependencies, assetId);
  134. EXPECT_TRUE(result.IsSuccess());
  135. auto& actualDependencies = result.GetValue();
  136. EXPECT_TRUE(actualDependencies.size() == actualDependencies.size());
  137. for (const auto& dependency : expectedDependencies)
  138. {
  139. EXPECT_TRUE(Search(actualDependencies, dependency));
  140. }
  141. }
  142. void CheckAllDependencies(AZ::Data::AssetId assetId, AZStd::initializer_list<AZ::Data::AssetId> expectedDependencies)
  143. {
  144. AZ::Outcome<AZStd::vector<AZ::Data::ProductDependency>, AZStd::string> result = AZ::Failure<AZStd::string>("No response");
  145. AZ::Data::AssetCatalogRequestBus::BroadcastResult(result, &AZ::Data::AssetCatalogRequestBus::Events::GetAllProductDependencies, assetId);
  146. EXPECT_TRUE(result.IsSuccess());
  147. auto& actualDependencies = result.GetValue();
  148. EXPECT_TRUE(actualDependencies.size() == actualDependencies.size());
  149. for (const auto& dependency : expectedDependencies)
  150. {
  151. EXPECT_TRUE(Search(actualDependencies, dependency));
  152. }
  153. }
  154. void CheckAllDependenciesFilter(AZ::Data::AssetId assetId, const AZStd::unordered_set<AZ::Data::AssetId>& filterSet, const AZStd::vector<AZStd::string>& wildcardList, AZStd::initializer_list<AZ::Data::AssetId> expectedDependencies)
  155. {
  156. AZ::Outcome<AZStd::vector<AZ::Data::ProductDependency>, AZStd::string> result = AZ::Failure<AZStd::string>("No response");
  157. AZ::Data::AssetCatalogRequestBus::BroadcastResult(result, &AZ::Data::AssetCatalogRequestBus::Events::GetAllProductDependenciesFilter, assetId, filterSet, wildcardList);
  158. EXPECT_TRUE(result.IsSuccess());
  159. auto& actualDependencies = result.GetValue();
  160. EXPECT_TRUE(actualDependencies.size() == actualDependencies.size());
  161. for (const auto& dependency : expectedDependencies)
  162. {
  163. EXPECT_TRUE(Search(actualDependencies, dependency));
  164. }
  165. }
  166. };
  167. TEST_F(AssetCatalogDependencyTest, directDependencies)
  168. {
  169. CheckDirectDependencies(asset1, { asset2 });
  170. CheckDirectDependencies(asset2, { asset3, asset4 });
  171. CheckDirectDependencies(asset3, { asset5 });
  172. CheckDirectDependencies(asset4, { });
  173. CheckDirectDependencies(asset5, { });
  174. }
  175. TEST_F(AssetCatalogDependencyTest, allDependencies)
  176. {
  177. CheckAllDependencies(asset1, { asset2, asset3, asset4, asset5 });
  178. CheckAllDependencies(asset2, { asset3, asset4, asset5 });
  179. CheckAllDependencies(asset3, { asset5 });
  180. CheckAllDependencies(asset4, {});
  181. CheckAllDependencies(asset5, {});
  182. }
  183. TEST_F(AssetCatalogDependencyTest, allDependenciesFilter_AssetIdFilter)
  184. {
  185. CheckAllDependenciesFilter(asset1, { asset2 }, { }, { });
  186. CheckAllDependenciesFilter(asset1, { asset3 }, { }, { asset2 });
  187. CheckAllDependenciesFilter(asset1, { asset5 }, { }, { asset2, asset3, asset4 });
  188. CheckAllDependenciesFilter(asset1, { asset4, asset5 }, { }, { asset2, asset3 });
  189. }
  190. TEST_F(AssetCatalogDependencyTest, allDependenciesFilter_WildcardFilter)
  191. {
  192. CheckAllDependenciesFilter(asset1, { }, { "*/*" }, { });
  193. CheckAllDependenciesFilter(asset1, { }, { "*/Asset?Path" }, {});
  194. CheckAllDependenciesFilter(asset1, { }, { "other*" }, { asset2 });
  195. CheckAllDependenciesFilter(asset1, { }, { "*4*" }, { asset2, asset3, asset5 });
  196. CheckAllDependenciesFilter(asset1, { }, { "*4*", "*5*" }, { asset2, asset3 });
  197. }
  198. TEST_F(AssetCatalogDependencyTest, allDependenciesFilter_AssetIdAndWildcardFilter)
  199. {
  200. CheckAllDependenciesFilter(asset1, { asset5 }, { "*4*" }, { asset2, asset3 });
  201. }
  202. TEST_F(AssetCatalogDependencyTest, unregisterTest)
  203. {
  204. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::UnregisterAsset, asset4);
  205. CheckDirectDependencies(asset2, { asset3 });
  206. CheckAllDependencies(asset1, { asset2, asset3, asset5 });
  207. }
  208. class AssetCatalogDeltaTest :
  209. public LeakDetectionFixture
  210. {
  211. public:
  212. const char* path1 = "Asset1Path";
  213. const char* path2 = "Asset2Path";
  214. const char* path3 = "Asset3Path";
  215. const char* path4 = "Asset4Path";
  216. const char* path5 = "Asset5Path";
  217. const char* path6 = "Asset6Path";
  218. AZ::IO::FixedMaxPath sourceCatalogPath1;
  219. AZ::IO::FixedMaxPath sourceCatalogPath2;
  220. AZ::IO::FixedMaxPath sourceCatalogPath3;
  221. AZ::IO::FixedMaxPath baseCatalogPath;
  222. AZ::IO::FixedMaxPath deltaCatalogPath;
  223. AZ::IO::FixedMaxPath deltaCatalogPath2;
  224. AZ::IO::FixedMaxPath deltaCatalogPath3;
  225. AZStd::vector<AZStd::string> deltaCatalogFiles;
  226. AZStd::vector<AZStd::string> deltaCatalogFiles2;
  227. AZStd::vector<AZStd::string> deltaCatalogFiles3;
  228. AZStd::shared_ptr<AzFramework::AssetRegistry> baseCatalog;
  229. AZStd::shared_ptr<AzFramework::AssetRegistry> deltaCatalog;
  230. AZStd::shared_ptr<AzFramework::AssetRegistry> deltaCatalog2;
  231. AZStd::shared_ptr<AzFramework::AssetRegistry> deltaCatalog3;
  232. AZStd::unique_ptr<AzFramework::Application> m_app;
  233. AZ::Test::ScopedAutoTempDirectory m_tempDirectory;
  234. void SetUp() override
  235. {
  236. m_app.reset(aznew AzFramework::Application());
  237. AZ::ComponentApplication::Descriptor desc;
  238. desc.m_useExistingAllocator = true;
  239. AZ::SettingsRegistryInterface* registry = AZ::SettingsRegistry::Get();
  240. auto projectPathKey =
  241. AZ::SettingsRegistryInterface::FixedValueString(AZ::SettingsRegistryMergeUtils::BootstrapSettingsRootKey) + "/project_path";
  242. AZ::IO::FixedMaxPath enginePath;
  243. registry->Get(enginePath.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_EngineRootFolder);
  244. registry->Set(projectPathKey, (enginePath / "AutomatedTesting").Native());
  245. AZ::SettingsRegistryMergeUtils::MergeSettingsToRegistry_AddRuntimeFilePaths(*registry);
  246. AZ::ComponentApplication::StartupParameters startupParameters;
  247. startupParameters.m_loadAssetCatalog = false;
  248. startupParameters.m_loadSettingsRegistry = false;
  249. m_app->Start(desc, startupParameters);
  250. // Without this, the user settings component would attempt to save on finalize/shutdown. Since the file is
  251. // shared across the whole engine, if multiple tests are run in parallel, the saving could cause a crash
  252. // in the unit tests.
  253. AZ::UserSettingsComponentRequestBus::Broadcast(&AZ::UserSettingsComponentRequests::DisableSaveOnFinalize);
  254. asset1 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  255. asset2 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  256. asset3 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  257. asset4 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  258. asset5 = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  259. sourceCatalogPath1 = m_tempDirectory.GetDirectoryAsPath() / "AssetCatalogSource1.xml";
  260. sourceCatalogPath2 = m_tempDirectory.GetDirectoryAsPath() / "AssetCatalogSource2.xml";
  261. sourceCatalogPath3 = m_tempDirectory.GetDirectoryAsPath() / "AssetCatalogSource3.xml";
  262. baseCatalogPath = m_tempDirectory.GetDirectoryAsPath() / "AssetCatalogBase.xml";
  263. deltaCatalogPath = m_tempDirectory.GetDirectoryAsPath() / "AssetCatalogDelta.xml";
  264. deltaCatalogPath2 = m_tempDirectory.GetDirectoryAsPath() / "AssetCatalogDelta2.xml";
  265. deltaCatalogPath3 = m_tempDirectory.GetDirectoryAsPath() / "AssetCatalogDelta3.xml";
  266. AZ::Data::AssetInfo info1, info2, info3, info4, info5, info6;
  267. info1.m_relativePath = path1;
  268. info2.m_relativePath = path2;
  269. info3.m_relativePath = path3;
  270. info4.m_relativePath = path4;
  271. info5.m_relativePath = path5;
  272. info6.m_relativePath = path6;
  273. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset1, info1);
  274. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset2, info2);
  275. // baseCatalog - asset1 path1, asset2 path2
  276. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::SaveCatalog, baseCatalogPath.c_str());
  277. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset1, info3);
  278. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset4, info4);
  279. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::StartMonitoringAssets);
  280. AzFramework::AssetSystem::NetworkAssetUpdateInterface* notificationInterface = AZ::Interface<AzFramework::AssetSystem::NetworkAssetUpdateInterface>::Get();
  281. ASSERT_NE(notificationInterface, nullptr);
  282. {
  283. AzFramework::AssetSystem::AssetNotificationMessage message(path3, AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  284. message.m_assetId = asset1;
  285. message.m_dependencies.push_back(AZ::Data::ProductDependency(asset2, 0));
  286. notificationInterface->AssetChanged({ message });
  287. }
  288. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::StopMonitoringAssets);
  289. // sourcecatalog1 - asset1 path3 (depends on asset 2), asset2 path2, asset4 path4
  290. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::SaveCatalog, sourceCatalogPath1.c_str());
  291. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::UnregisterAsset, asset2);
  292. // deltacatalog - asset1 path3 (depends on asset 2), asset4 path4
  293. deltaCatalogFiles.push_back(path3);
  294. deltaCatalogFiles.push_back(path4);
  295. // deltacatalog - asset1 path3, asset4 path4
  296. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::SaveCatalog, deltaCatalogPath.c_str());
  297. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset2, info2);
  298. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset5, info5);
  299. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::StartMonitoringAssets);
  300. {
  301. AzFramework::AssetSystem::AssetNotificationMessage message(path5, AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  302. message.m_assetId = asset5;
  303. message.m_dependencies.push_back(AZ::Data::ProductDependency(asset2, 0));
  304. notificationInterface->AssetChanged({ message });
  305. }
  306. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::StopMonitoringAssets);
  307. // sourcecatalog2 - asset1 path3 (depends on asset 2), asset2 path2, asset4 path4, asset5 path5 (depends on asset 2)
  308. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::SaveCatalog, sourceCatalogPath2.c_str());
  309. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::UnregisterAsset, asset1);
  310. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::UnregisterAsset, asset2);
  311. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::UnregisterAsset, asset4);
  312. // deltacatalog2 - asset5 path5 (depends on asset 2)
  313. deltaCatalogFiles2.push_back(path5);
  314. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::SaveCatalog, deltaCatalogPath2.c_str());
  315. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset1, info6);
  316. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset2, info2);
  317. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RegisterAsset, asset5, info4);
  318. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::StartMonitoringAssets);
  319. {
  320. AzFramework::AssetSystem::AssetNotificationMessage message(path4, AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  321. message.m_assetId = asset5;
  322. message.m_dependencies.push_back(AZ::Data::ProductDependency(asset2, 0));
  323. notificationInterface->AssetChanged({ message });
  324. }
  325. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::StopMonitoringAssets);
  326. //sourcecatalog3 - asset1 path6, asset2 path2, asset5 path4 (depends on asset 2)
  327. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::SaveCatalog, sourceCatalogPath3.c_str());
  328. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::UnregisterAsset, asset2);
  329. // deltacatalog3 - asset1 path6 asset5 path4 (depends on asset 2)
  330. deltaCatalogFiles3.push_back(path6);
  331. deltaCatalogFiles3.push_back(path4);
  332. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::SaveCatalog, deltaCatalogPath3.c_str());
  333. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::UnregisterAsset, asset1);
  334. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::UnregisterAsset, asset5);
  335. baseCatalog = AzFramework::AssetCatalog::LoadCatalogFromFile(baseCatalogPath.c_str());
  336. deltaCatalog = AzFramework::AssetCatalog::LoadCatalogFromFile(deltaCatalogPath.c_str());
  337. deltaCatalog2 = AzFramework::AssetCatalog::LoadCatalogFromFile(deltaCatalogPath2.c_str());
  338. deltaCatalog3 = AzFramework::AssetCatalog::LoadCatalogFromFile(deltaCatalogPath3.c_str());
  339. }
  340. void TearDown() override
  341. {
  342. deltaCatalogFiles.set_capacity(0);
  343. deltaCatalogFiles2.set_capacity(0);
  344. deltaCatalogFiles3.set_capacity(0);
  345. baseCatalog.reset();
  346. deltaCatalog.reset();
  347. deltaCatalog2.reset();
  348. deltaCatalog3.reset();
  349. m_app->Stop();
  350. m_app.reset();
  351. AZ::AllocatorInstance<AZ::SystemAllocator>::Get().GarbageCollect();
  352. }
  353. void CheckDirectDependencies(AZ::Data::AssetId assetId, AZStd::initializer_list<AZ::Data::AssetId> expectedDependencies)
  354. {
  355. AZ::Outcome<AZStd::vector<AZ::Data::ProductDependency>, AZStd::string> result = AZ::Failure<AZStd::string>("No response");
  356. AZ::Data::AssetCatalogRequestBus::BroadcastResult(result, &AZ::Data::AssetCatalogRequestBus::Events::GetDirectProductDependencies, assetId);
  357. EXPECT_TRUE(result.IsSuccess());
  358. auto& actualDependencies = result.GetValue();
  359. EXPECT_TRUE(actualDependencies.size() == actualDependencies.size());
  360. for (const auto& dependency : expectedDependencies)
  361. {
  362. EXPECT_TRUE(Search(actualDependencies, dependency));
  363. }
  364. }
  365. void CheckNoDependencies(AZ::Data::AssetId assetId)
  366. {
  367. AZ::Outcome<AZStd::vector<AZ::Data::ProductDependency>, AZStd::string> result = AZ::Failure<AZStd::string>("No response");
  368. AZ::Data::AssetCatalogRequestBus::BroadcastResult(result, &AZ::Data::AssetCatalogRequestBus::Events::GetDirectProductDependencies, assetId);
  369. EXPECT_FALSE(result.IsSuccess());
  370. }
  371. };
  372. TEST_F(AssetCatalogDeltaTest, LoadCatalog_AssetChangedCatalogLoaded_AssetStillKnown)
  373. {
  374. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::StartMonitoringAssets);
  375. auto updatedAsset = AZ::Data::AssetId(AZ::Uuid::CreateRandom(), 0);
  376. AzFramework::AssetSystem::NetworkAssetUpdateInterface* notificationInterface = AZ::Interface<AzFramework::AssetSystem::NetworkAssetUpdateInterface>::Get();
  377. ASSERT_NE(notificationInterface, nullptr);
  378. {
  379. AzFramework::AssetSystem::AssetNotificationMessage message("test", AzFramework::AssetSystem::AssetNotificationMessage::AssetChanged, AZ::Uuid::CreateRandom(), "");
  380. message.m_assetId = updatedAsset;
  381. message.m_dependencies.emplace_back(asset2, 0);
  382. notificationInterface->AssetChanged({ message });
  383. }
  384. AZ::Data::AssetInfo assetInfo;
  385. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetInfo, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetInfoById, updatedAsset);
  386. EXPECT_TRUE(assetInfo.m_assetId.IsValid());
  387. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, baseCatalogPath.c_str());
  388. //Asset should still be known - the catalog should swap in the new catalog from disk
  389. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetInfo, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetInfoById, updatedAsset);
  390. EXPECT_TRUE(assetInfo.m_assetId.IsValid());
  391. }
  392. TEST_F(AssetCatalogDeltaTest, DeltaCatalogTest)
  393. {
  394. AZStd::string assetPath;
  395. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  396. EXPECT_EQ(assetPath, "");
  397. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::ClearCatalog);
  398. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, baseCatalogPath.c_str());
  399. // Topmost should have precedence
  400. // baseCatalog path1 path2
  401. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  402. EXPECT_EQ(assetPath, path1);
  403. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  404. EXPECT_EQ(assetPath, path2);
  405. CheckNoDependencies(asset1);
  406. CheckNoDependencies(asset2);
  407. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::AddDeltaCatalog, deltaCatalog);
  408. // deltacatalog path3 path4
  409. /// baseCatalog path1 path2
  410. /// asset1 asset2 asset3 asset4 asset5
  411. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  412. EXPECT_EQ(assetPath, path3);
  413. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  414. EXPECT_EQ(assetPath, path2);
  415. CheckDirectDependencies(asset1, { asset2 });
  416. CheckNoDependencies(asset2);
  417. CheckNoDependencies(asset4);
  418. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::AddDeltaCatalog, deltaCatalog2);
  419. // deltacatalog2 path5
  420. // deltacatalog path3 path4
  421. /// baseCatalog path1 path2
  422. /// asset1 asset2 asset3 asset4 asset5
  423. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  424. EXPECT_EQ(assetPath, path3);
  425. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  426. EXPECT_EQ(assetPath, path2);
  427. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  428. EXPECT_EQ(assetPath, path5);
  429. CheckDirectDependencies(asset1, { asset2 });
  430. CheckNoDependencies(asset2);
  431. CheckNoDependencies(asset4);
  432. CheckDirectDependencies(asset5, { asset2 });
  433. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RemoveDeltaCatalog, deltaCatalog);
  434. // deltacatalog2 path5
  435. /// baseCatalog path1 path2
  436. /// asset1 asset2 asset3 asset4 asset5
  437. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  438. EXPECT_EQ(assetPath, path1);
  439. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset3);
  440. EXPECT_EQ(assetPath, "");
  441. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  442. EXPECT_EQ(assetPath, path2);
  443. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  444. EXPECT_EQ(assetPath, path5);
  445. CheckNoDependencies(asset1);
  446. CheckNoDependencies(asset2);
  447. CheckDirectDependencies(asset5, { asset2 });
  448. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::AddDeltaCatalog, deltaCatalog);
  449. // deltacatalog path3 path4
  450. // deltacatalog2 path5
  451. /// baseCatalog path1 path2
  452. /// asset1 asset2 asset3 asset4 asset5
  453. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  454. EXPECT_EQ(assetPath, path3);
  455. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  456. EXPECT_EQ(assetPath, path2);
  457. CheckDirectDependencies(asset1, { asset2 });
  458. CheckNoDependencies(asset2);
  459. CheckNoDependencies(asset4);
  460. CheckDirectDependencies(asset5, { asset2 });
  461. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::InsertDeltaCatalog, deltaCatalog3, 0);
  462. // deltacatalog path3 path4
  463. // deltacatalog2 path5
  464. // deltacatalog3 path6 path4
  465. /// baseCatalog path1 path2
  466. /// asset1 asset2 asset3 asset4 asset5
  467. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  468. EXPECT_EQ(assetPath, path3);
  469. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  470. EXPECT_EQ(assetPath, path5);
  471. CheckDirectDependencies(asset1, { asset2 });
  472. CheckNoDependencies(asset2);
  473. CheckNoDependencies(asset4);
  474. CheckDirectDependencies(asset5, { asset2 });
  475. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RemoveDeltaCatalog, deltaCatalog3);
  476. // deltacatalog path3 path4
  477. // deltacatalog2 path5
  478. /// baseCatalog path1 path2
  479. /// asset1 asset2 asset3 asset4 asset5
  480. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  481. EXPECT_EQ(assetPath, path3);
  482. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  483. EXPECT_EQ(assetPath, path5);
  484. CheckDirectDependencies(asset1, { asset2 });
  485. CheckNoDependencies(asset2);
  486. CheckNoDependencies(asset4);
  487. CheckDirectDependencies(asset5, { asset2 });
  488. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::InsertDeltaCatalog, deltaCatalog3, 1);
  489. // deltacatalog path3 path4
  490. // deltacatalog3 path6 path4
  491. // deltacatalog2 path5
  492. /// baseCatalog path1 path2
  493. /// asset1 asset2 asset3 asset4 asset5
  494. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  495. EXPECT_EQ(assetPath, path3);
  496. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  497. EXPECT_EQ(assetPath, path4);
  498. CheckDirectDependencies(asset1, { asset2 });
  499. CheckNoDependencies(asset2);
  500. CheckNoDependencies(asset4);
  501. CheckDirectDependencies(asset5, { asset2 });
  502. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RemoveDeltaCatalog, baseCatalog);
  503. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::InsertDeltaCatalog, baseCatalog, 3);
  504. /// baseCatalog path1 path2
  505. // deltacatalog path3 path4
  506. // deltacatalog3 path6 path4
  507. // deltacatalog2 path5
  508. /// asset1 asset2 asset3 asset4 asset5
  509. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  510. EXPECT_EQ(assetPath, path1);
  511. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  512. EXPECT_EQ(assetPath, path2);
  513. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset3);
  514. EXPECT_EQ(assetPath, "");
  515. CheckNoDependencies(asset1);
  516. CheckNoDependencies(asset2);
  517. CheckNoDependencies(asset4);
  518. CheckDirectDependencies(asset5, { asset2 });
  519. }
  520. TEST_F(AssetCatalogDeltaTest, DeltaCatalogTest_AddDeltaCatalogNext_Success)
  521. {
  522. AZStd::string assetPath;
  523. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  524. EXPECT_EQ(assetPath, "");
  525. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::ClearCatalog);
  526. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, baseCatalogPath.c_str());
  527. // Topmost should have precedence
  528. // baseCatalog path1 path2
  529. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  530. EXPECT_EQ(assetPath, path1);
  531. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  532. EXPECT_EQ(assetPath, path2);
  533. CheckNoDependencies(asset1);
  534. CheckNoDependencies(asset2);
  535. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::AddDeltaCatalog, deltaCatalog);
  536. // deltacatalog path3 path4
  537. /// baseCatalog path1 path2
  538. /// asset1 asset2 asset3 asset4 asset5
  539. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  540. EXPECT_EQ(assetPath, path3);
  541. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  542. EXPECT_EQ(assetPath, path2);
  543. CheckDirectDependencies(asset1, { asset2 });
  544. CheckNoDependencies(asset2);
  545. CheckNoDependencies(asset4);
  546. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::AddDeltaCatalog, deltaCatalog2);
  547. // deltacatalog2 path5
  548. // deltacatalog path3 path4
  549. /// baseCatalog path1 path2
  550. /// asset1 asset2 asset3 asset4 asset5
  551. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  552. EXPECT_EQ(assetPath, path3);
  553. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  554. EXPECT_EQ(assetPath, path2);
  555. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  556. EXPECT_EQ(assetPath, path5);
  557. CheckDirectDependencies(asset1, { asset2 });
  558. CheckNoDependencies(asset2);
  559. CheckNoDependencies(asset4);
  560. CheckDirectDependencies(asset5, { asset2 });
  561. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RemoveDeltaCatalog, deltaCatalog);
  562. // deltacatalog2 path5
  563. /// baseCatalog path1 path2
  564. /// asset1 asset2 asset3 asset4 asset5
  565. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  566. EXPECT_EQ(assetPath, path1);
  567. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset3);
  568. EXPECT_EQ(assetPath, "");
  569. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  570. EXPECT_EQ(assetPath, path2);
  571. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  572. EXPECT_EQ(assetPath, path5);
  573. CheckNoDependencies(asset1);
  574. CheckNoDependencies(asset2);
  575. CheckDirectDependencies(asset5, { asset2 });
  576. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::AddDeltaCatalog, deltaCatalog);
  577. // deltacatalog path3 path4
  578. // deltacatalog2 path5
  579. /// baseCatalog path1 path2
  580. /// asset1 asset2 asset3 asset4 asset5
  581. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  582. EXPECT_EQ(assetPath, path3);
  583. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  584. EXPECT_EQ(assetPath, path2);
  585. CheckDirectDependencies(asset1, { asset2 });
  586. CheckNoDependencies(asset2);
  587. CheckNoDependencies(asset4);
  588. CheckDirectDependencies(asset5, { asset2 });
  589. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::InsertDeltaCatalogBefore, deltaCatalog3, deltaCatalog2);
  590. // deltacatalog path3 path4
  591. // deltacatalog2 path5
  592. // deltacatalog3 path6 path4
  593. /// baseCatalog path1 path2
  594. /// asset1 asset2 asset3 asset4 asset5
  595. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  596. EXPECT_EQ(assetPath, path3);
  597. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  598. EXPECT_EQ(assetPath, path5);
  599. CheckDirectDependencies(asset1, { asset2 });
  600. CheckNoDependencies(asset2);
  601. CheckNoDependencies(asset4);
  602. CheckDirectDependencies(asset5, { asset2 });
  603. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RemoveDeltaCatalog, deltaCatalog3);
  604. // deltacatalog path3 path4
  605. // deltacatalog2 path5
  606. /// baseCatalog path1 path2
  607. /// asset1 asset2 asset3 asset4 asset5
  608. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  609. EXPECT_EQ(assetPath, path3);
  610. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  611. EXPECT_EQ(assetPath, path5);
  612. CheckDirectDependencies(asset1, { asset2 });
  613. CheckNoDependencies(asset2);
  614. CheckNoDependencies(asset4);
  615. CheckDirectDependencies(asset5, { asset2 });
  616. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::InsertDeltaCatalogBefore, deltaCatalog3, deltaCatalog);
  617. // deltacatalog path3 path4
  618. // deltacatalog3 path6 path4
  619. // deltacatalog2 path5
  620. /// baseCatalog path1 path2
  621. /// asset1 asset2 asset3 asset4 asset5
  622. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  623. EXPECT_EQ(assetPath, path3);
  624. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  625. EXPECT_EQ(assetPath, path4);
  626. CheckDirectDependencies(asset1, { asset2 });
  627. CheckNoDependencies(asset2);
  628. CheckNoDependencies(asset4);
  629. CheckDirectDependencies(asset5, { asset2 });
  630. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::RemoveDeltaCatalog, baseCatalog);
  631. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::AddDeltaCatalog, baseCatalog);
  632. /// baseCatalog path1 path2
  633. // deltacatalog path3 path4
  634. // deltacatalog3 path6 path4
  635. // deltacatalog2 path5
  636. /// asset1 asset2 asset3 asset4 asset5
  637. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  638. EXPECT_EQ(assetPath, path1);
  639. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  640. EXPECT_EQ(assetPath, path2);
  641. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset3);
  642. EXPECT_EQ(assetPath, "");
  643. CheckNoDependencies(asset1);
  644. CheckNoDependencies(asset2);
  645. CheckNoDependencies(asset4);
  646. CheckDirectDependencies(asset5, { asset2 });
  647. }
  648. TEST_F(AssetCatalogDeltaTest, DeltaCatalogCreationTest)
  649. {
  650. bool result = false;
  651. AZStd::string assetPath;
  652. // load source catalog 1
  653. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::ClearCatalog);
  654. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, sourceCatalogPath1.c_str());
  655. // generate delta catalog 1
  656. AZ::IO::Path testDeltaCatalogPath1 = m_tempDirectory.GetDirectoryAsPath() / "TestAssetCatalogDelta.xml";
  657. AZ::Data::AssetCatalogRequestBus::BroadcastResult(result, &AZ::Data::AssetCatalogRequestBus::Events::CreateDeltaCatalog, deltaCatalogFiles, testDeltaCatalogPath1.Native());
  658. EXPECT_TRUE(result);
  659. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::ClearCatalog);
  660. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, testDeltaCatalogPath1.c_str());
  661. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  662. EXPECT_EQ(assetPath, path3);
  663. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset4);
  664. EXPECT_EQ(assetPath, path4);
  665. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  666. EXPECT_EQ(assetPath, "");
  667. // check against depenency info
  668. CheckDirectDependencies(asset1, { asset2 });
  669. // load source catalog 2
  670. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::ClearCatalog);
  671. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, sourceCatalogPath2.c_str());
  672. // generate delta catalog 3
  673. AZ::IO::Path testDeltaCatalogPath2 = m_tempDirectory.GetDirectoryAsPath() / "TestAssetCatalogDelta2.xml";
  674. AZ::Data::AssetCatalogRequestBus::BroadcastResult(result, &AZ::Data::AssetCatalogRequestBus::Events::CreateDeltaCatalog, deltaCatalogFiles2, testDeltaCatalogPath2.Native());
  675. EXPECT_TRUE(result);
  676. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::ClearCatalog);
  677. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, testDeltaCatalogPath2.c_str());
  678. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  679. EXPECT_EQ(assetPath, path5);
  680. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  681. EXPECT_EQ(assetPath, "");
  682. // check against dependency info
  683. CheckDirectDependencies(asset5, { asset2 });
  684. // load source catalog 3
  685. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::ClearCatalog);
  686. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, sourceCatalogPath3.c_str());
  687. // generate delta catalog 3
  688. AZ::IO::Path testDeltaCatalogPath3 = m_tempDirectory.GetDirectoryAsPath() / "TestAssetCatalogDelta3.xml";
  689. AZ::Data::AssetCatalogRequestBus::BroadcastResult(result, &AZ::Data::AssetCatalogRequestBus::Events::CreateDeltaCatalog, deltaCatalogFiles3, testDeltaCatalogPath3.Native());
  690. EXPECT_TRUE(result);
  691. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::ClearCatalog);
  692. AZ::Data::AssetCatalogRequestBus::Broadcast(&AZ::Data::AssetCatalogRequestBus::Events::LoadCatalog, testDeltaCatalogPath3.c_str());
  693. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset1);
  694. EXPECT_EQ(assetPath, path6);
  695. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset5);
  696. EXPECT_EQ(assetPath, path4);
  697. AZ::Data::AssetCatalogRequestBus::BroadcastResult(assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, asset2);
  698. EXPECT_EQ(assetPath, "");
  699. // check against dependency info
  700. CheckDirectDependencies(asset5, { asset2 });
  701. CheckNoDependencies(asset1);
  702. }
  703. class AssetCatalogAPITest
  704. : public LeakDetectionFixture
  705. {
  706. public:
  707. void SetUp() override
  708. {
  709. using namespace AZ::Data;
  710. m_application = new AzFramework::Application();
  711. auto settingsRegistry = AZ::SettingsRegistry::Get();
  712. ASSERT_NE(nullptr, settingsRegistry);
  713. EXPECT_TRUE(settingsRegistry->Get(m_assetRoot.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_CacheRootFolder));
  714. m_assetCatalog = aznew AzFramework::AssetCatalog();
  715. m_firstAssetId = AssetId(AZ::Uuid::CreateRandom(), 0);
  716. AssetInfo assetInfo;
  717. assetInfo.m_assetId = m_firstAssetId;
  718. assetInfo.m_relativePath = "AssetA.txt";
  719. assetInfo.m_sizeBytes = 1; //any non zero value
  720. m_assetCatalog->RegisterAsset(m_firstAssetId, assetInfo);
  721. m_secondAssetId = AssetId(AZ::Uuid::CreateRandom(), 0);
  722. assetInfo.m_assetId = m_secondAssetId;
  723. assetInfo.m_relativePath = "Foo/AssetA.txt";
  724. assetInfo.m_sizeBytes = 1; //any non zero value
  725. m_assetCatalog->RegisterAsset(m_secondAssetId, assetInfo);
  726. }
  727. void TearDown() override
  728. {
  729. delete m_assetCatalog;
  730. delete m_application;
  731. }
  732. AzFramework::Application* m_application;
  733. AzFramework::AssetCatalog* m_assetCatalog;
  734. AZ::Data::AssetId m_firstAssetId;
  735. AZ::Data::AssetId m_secondAssetId;
  736. AZ::IO::Path m_assetRoot;
  737. };
  738. TEST_F(AssetCatalogAPITest, GetAssetPathById_AbsolutePath_Valid)
  739. {
  740. AZ::IO::Path absPath = m_assetRoot / "AssetA.txt";
  741. EXPECT_EQ(m_firstAssetId, m_assetCatalog->GetAssetIdByPath(absPath.c_str(), AZ::Data::s_invalidAssetType, false));
  742. }
  743. TEST_F(AssetCatalogAPITest, GetAssetPathById_AbsolutePathWithSubFolder_Valid)
  744. {
  745. AZ::IO::Path absPath = (m_assetRoot / "Foo/AssetA.txt").LexicallyNormal();
  746. EXPECT_EQ(m_secondAssetId, m_assetCatalog->GetAssetIdByPath(absPath.c_str(), AZ::Data::s_invalidAssetType, false));
  747. }
  748. TEST_F(AssetCatalogAPITest, GetAssetPathById_RelPath_Valid)
  749. {
  750. EXPECT_EQ(m_firstAssetId, m_assetCatalog->GetAssetIdByPath("AssetA.txt", AZ::Data::s_invalidAssetType, false));
  751. }
  752. TEST_F(AssetCatalogAPITest, GetAssetPathById_RelPathWithSubFolders_Valid)
  753. {
  754. EXPECT_EQ(m_secondAssetId, m_assetCatalog->GetAssetIdByPath("Foo/AssetA.txt", AZ::Data::s_invalidAssetType, false));
  755. }
  756. TEST_F(AssetCatalogAPITest, GetAssetPathById_AbsPath_Invalid)
  757. {
  758. // A path that begins with a Posix path separator '/' is absolute not a relative path
  759. EXPECT_NE(m_firstAssetId, m_assetCatalog->GetAssetIdByPath("//AssetA.txt", AZ::Data::s_invalidAssetType, false));
  760. }
  761. TEST_F(AssetCatalogAPITest, GetRegisteredAssetPaths_TwoAssetsRegistered_RetrieveAssetPaths)
  762. {
  763. AZStd::vector<AZStd::string> assetPaths = m_assetCatalog->GetRegisteredAssetPaths();
  764. EXPECT_EQ(assetPaths.size(), 2);
  765. ASSERT_THAT(assetPaths, testing::ElementsAre(
  766. "AssetA.txt",
  767. "Foo/AssetA.txt"));
  768. }
  769. TEST_F(AssetCatalogAPITest, DoesAssetIdMatchWildcardPattern_Matches)
  770. {
  771. EXPECT_TRUE(m_assetCatalog->DoesAssetIdMatchWildcardPattern(m_firstAssetId, "*.txt"));
  772. EXPECT_TRUE(m_assetCatalog->DoesAssetIdMatchWildcardPattern(m_firstAssetId, "asset?.txt"));
  773. }
  774. TEST_F(AssetCatalogAPITest, DoesAssetIdMatchWildcardPattern_DoesNotMatch)
  775. {
  776. EXPECT_FALSE(m_assetCatalog->DoesAssetIdMatchWildcardPattern(AZ::Data::AssetId(), "*.txt"));
  777. EXPECT_FALSE(m_assetCatalog->DoesAssetIdMatchWildcardPattern(m_firstAssetId, "*.xml"));
  778. EXPECT_FALSE(m_assetCatalog->DoesAssetIdMatchWildcardPattern(m_firstAssetId, ""));
  779. }
  780. TEST_F(AssetCatalogAPITest, EnumerateAssetsListsCorrectAssets)
  781. {
  782. AZStd::vector<AZ::Data::AssetId> foundAssets;
  783. AZ::Data::AssetCatalogRequestBus::Broadcast(
  784. &AZ::Data::AssetCatalogRequestBus::Events::EnumerateAssets, nullptr,
  785. [&foundAssets](const AZ::Data::AssetId assetId, [[maybe_unused]] const AZ::Data::AssetInfo& assetInfo)
  786. {
  787. foundAssets.emplace_back(assetId);
  788. },
  789. nullptr);
  790. ASSERT_EQ(foundAssets.size(), 2);
  791. EXPECT_EQ(foundAssets[0], m_firstAssetId);
  792. EXPECT_EQ(foundAssets[1], m_secondAssetId);
  793. }
  794. TEST_F(AssetCatalogAPITest, EnumerateAssetsDoesNotBlockMutex)
  795. {
  796. // This test simulates a previously-occurring deadlock bug caused by having the main thread call AssetCatalog::EnumerateAssets
  797. // with a callback that calls the AssetManager, and a loading thread running underneath the AssetManager that calls the
  798. // AssetCatalog.
  799. // To reproduce this state, we will do the following:
  800. // Main Thread Job Thread
  801. // wait on mainToJobSync
  802. // call EnumerateAssets
  803. // unblock mainToJobSync
  804. // wait on jobToMainSync call AssetCatalog::X
  805. // unblock jobToMainSync
  806. //
  807. // These steps should ensure that we create a theoretical "lock inversion", if EnumerateAssets locked an AssetCatalog mutex.
  808. // Even though we're using binary semaphores instead of mutexes, we're creating the same blocking situation.
  809. // EnumerateAssets itself should no longer lock the AssetCatalog mutex, so this test should succeed since there won't be any
  810. // actual lock inversion.
  811. // Set up job manager with one thread that we can use to set up the concurrent mutex access.
  812. AZ::JobManagerDesc desc;
  813. AZ::JobManagerThreadDesc threadDesc;
  814. desc.m_workerThreads.push_back(threadDesc);
  815. auto jobManager = aznew AZ::JobManager(desc);
  816. auto jobContext = aznew AZ::JobContext(*jobManager);
  817. AZ::JobContext::SetGlobalContext(jobContext);
  818. AZStd::binary_semaphore mainToJobSync;
  819. AZStd::binary_semaphore jobToMainSync;
  820. // Create a job whose sole purpose is to wait for EnumerateAssets to get called, call AssetCatalog, then tell EnumerateAssets
  821. // to finish. If EnumerateAssets has a mutex held, this will deadlock.
  822. auto job = AZ::CreateJobFunction(
  823. [&mainToJobSync, &jobToMainSync]() mutable
  824. {
  825. // wait for the main thread to be inside the EnumerateAssets callback.
  826. mainToJobSync.acquire();
  827. // call AssetCatalog::X. This will deadlock if EnumerateAssets is holding an AssetCatalog mutex.
  828. [[maybe_unused]] AZStd::string assetPath;
  829. AZ::Data::AssetCatalogRequestBus::BroadcastResult(
  830. assetPath, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetPathById, AZ::Data::AssetId());
  831. // signal the main thread to continue
  832. jobToMainSync.release();
  833. },
  834. true);
  835. job->Start();
  836. bool testCompletedSuccessfully = false;
  837. // Call EnumerateAssets with a callback that also tries to lock the mutex.
  838. AZ::Data::AssetCatalogRequestBus::Broadcast(
  839. &AZ::Data::AssetCatalogRequestBus::Events::EnumerateAssets, nullptr,
  840. [this, &mainToJobSync, &jobToMainSync, &testCompletedSuccessfully]
  841. (const AZ::Data::AssetId assetId, [[maybe_unused]] const AZ::Data::AssetInfo& assetInfo)
  842. {
  843. // Only run this logic on the first assetId.
  844. if (assetId == m_firstAssetId)
  845. {
  846. // Tell the job it can continue
  847. mainToJobSync.release();
  848. // Wait for the job to finish calling AssetCatalog::X. This will deadlock if EnumerateAssets is holding an
  849. // AssetCatalog mutex.
  850. if (jobToMainSync.try_acquire_for(AZStd::chrono::seconds(5)))
  851. {
  852. // Only mark the test as completed successfully if we ran this logic and didn't deadlock.
  853. testCompletedSuccessfully = true;
  854. }
  855. }
  856. },
  857. nullptr);
  858. EXPECT_TRUE(testCompletedSuccessfully);
  859. // Clean up the job manager
  860. AZ::JobContext::SetGlobalContext(nullptr);
  861. delete jobContext;
  862. delete jobManager;
  863. }
  864. class AssetType1
  865. : public AZ::Data::AssetData
  866. {
  867. public:
  868. AZ_CLASS_ALLOCATOR(AssetType1, AZ::SystemAllocator);
  869. AZ_RTTI(AssetType1, "{64ECE3AE-2BEB-4502-9243-D17425249E08}", AZ::Data::AssetData);
  870. AssetType1()
  871. : m_data(nullptr)
  872. {
  873. }
  874. explicit AssetType1(const AZ::Data::AssetId& assetId, const AZ::Data::AssetData::AssetStatus assetStatus = AZ::Data::AssetData::AssetStatus::NotLoaded)
  875. : AZ::Data::AssetData(assetId, assetStatus)
  876. , m_data(nullptr)
  877. {
  878. }
  879. ~AssetType1() override
  880. {
  881. if (m_data)
  882. {
  883. azfree(m_data);
  884. }
  885. }
  886. char* m_data;
  887. };
  888. class AssetType2
  889. : public AZ::Data::AssetData
  890. {
  891. public:
  892. AZ_CLASS_ALLOCATOR(AssetType2, AZ::SystemAllocator);
  893. AZ_RTTI(AssetType2, "{F5EDAAAB-2398-4967-A2C7-6B7C9421C1CE}", AZ::Data::AssetData);
  894. AssetType2()
  895. : m_data(nullptr)
  896. {
  897. }
  898. explicit AssetType2(const AZ::Data::AssetId& assetId, const AZ::Data::AssetData::AssetStatus assetStatus = AZ::Data::AssetData::AssetStatus::NotLoaded)
  899. : AZ::Data::AssetData(assetId, assetStatus)
  900. , m_data(nullptr)
  901. {
  902. }
  903. ~AssetType2() override
  904. {
  905. if (m_data)
  906. {
  907. azfree(m_data);
  908. }
  909. }
  910. char* m_data;
  911. };
  912. class AssetCatalogDisplayNameTest
  913. : public LeakDetectionFixture
  914. {
  915. virtual AZ::IO::IStreamer* CreateStreamer() { return aznew AZ::IO::Streamer(AZStd::thread_desc{}, AZ::StreamerComponent::CreateStreamerStack()); }
  916. virtual void DestroyStreamer(AZ::IO::IStreamer* streamer) { delete streamer; }
  917. AZ::IO::FileIOBase* m_prevFileIO{ nullptr };
  918. TestFileIOBase m_fileIO;
  919. AZ::IO::IStreamer* m_streamer{ nullptr };
  920. AzFramework::AssetCatalog* m_assetCatalog;
  921. public:
  922. void SetUp() override
  923. {
  924. m_prevFileIO = AZ::IO::FileIOBase::GetInstance();
  925. AZ::IO::FileIOBase::SetInstance(&m_fileIO);
  926. m_streamer = CreateStreamer();
  927. if (m_streamer)
  928. {
  929. AZ::Interface<AZ::IO::IStreamer>::Register(m_streamer);
  930. }
  931. // create the database
  932. AZ::Data::AssetManager::Descriptor desc;
  933. AZ::Data::AssetManager::Create(desc);
  934. m_assetCatalog = aznew AzFramework::AssetCatalog();
  935. }
  936. void TearDown() override
  937. {
  938. delete m_assetCatalog;
  939. // destroy the database
  940. AZ::Data::AssetManager::Destroy();
  941. if (m_streamer)
  942. {
  943. AZ::Interface<AZ::IO::IStreamer>::Unregister(m_streamer);
  944. }
  945. DestroyStreamer(m_streamer);
  946. AZ::IO::FileIOBase::SetInstance(m_prevFileIO);
  947. }
  948. };
  949. TEST_F(AssetCatalogDisplayNameTest, GetAssetTypeByDisplayName_GetRegistered)
  950. {
  951. // Register an asset type
  952. AZ::TypeId assetTypeId = AZ::AzTypeInfo<AssetType1>::Uuid();
  953. AZStd::string assetTypeName = "Test Asset Type";
  954. auto handler = aznew AzFramework::GenericAssetHandler<AssetType1>(assetTypeName.data(), "SomeGroup", "someextension", assetTypeId);
  955. handler->Register();
  956. {
  957. // Get the Asset Type from the Display Name
  958. AZ::Data::AssetType id;
  959. AZ::Data::AssetCatalogRequestBus::BroadcastResult(id, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetTypeByDisplayName, assetTypeName);
  960. EXPECT_EQ(id, assetTypeId);
  961. }
  962. // Delete the handler
  963. handler->Unregister();
  964. delete handler;
  965. }
  966. TEST_F(AssetCatalogDisplayNameTest, GetAssetTypeByDisplayName_GetNotRegistered)
  967. {
  968. // Register an asset type
  969. AZ::TypeId assetTypeId = AZ::AzTypeInfo<AssetType1>::Uuid();
  970. AZStd::string assetTypeName = "Test Asset Type";
  971. auto handler = aznew AzFramework::GenericAssetHandler<AssetType1>(assetTypeName.data(), "SomeGroup", "someextension", assetTypeId);
  972. handler->Register();
  973. {
  974. // Try to get an Asset Type with a Display Name that was never registered
  975. AZ::Data::AssetType id;
  976. AZ::Data::AssetCatalogRequestBus::BroadcastResult(id, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetTypeByDisplayName, "Some Asset Type We Did Not Register");
  977. EXPECT_TRUE(id.IsNull());
  978. }
  979. // Delete the handler
  980. handler->Unregister();
  981. delete handler;
  982. }
  983. TEST_F(AssetCatalogDisplayNameTest, GetAssetTypeByDisplayName_GetWrongCasing)
  984. {
  985. // Register an asset type
  986. AZ::TypeId assetTypeId = AZ::AzTypeInfo<AssetType1>::Uuid();
  987. AZStd::string assetTypeName = "Test Asset Type";
  988. auto handler = aznew AzFramework::GenericAssetHandler<AssetType1>(assetTypeName.data(), "SomeGroup", "someextension", assetTypeId);
  989. handler->Register();
  990. {
  991. // Try to get the Asset Type we registered, but with a lowercase Display Name
  992. AZ::Data::AssetType id;
  993. AZ::Data::AssetCatalogRequestBus::BroadcastResult(id, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetTypeByDisplayName, "test asset type");
  994. EXPECT_TRUE(id.IsNull());
  995. }
  996. // Delete the handler
  997. handler->Unregister();
  998. delete handler;
  999. }
  1000. TEST_F(AssetCatalogDisplayNameTest, GetAssetTypeByDisplayName_NoRegisteredType)
  1001. {
  1002. // Get the Asset Type from a Display Name we did not register
  1003. AZ::Data::AssetType id;
  1004. AZ::Data::AssetCatalogRequestBus::BroadcastResult(id, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetTypeByDisplayName, "Some Asset Type We Did Not Register");
  1005. EXPECT_TRUE(id.IsNull());
  1006. }
  1007. TEST_F(AssetCatalogDisplayNameTest, GetAssetTypeByDisplayName_MultiRegisteredType_SameDisplayName)
  1008. {
  1009. AZStd::string assetTypeName = "Test Asset Type";
  1010. // Register an asset type
  1011. AZ::TypeId asset1TypeId = AZ::AzTypeInfo<AssetType1>::Uuid();
  1012. auto handler1 = aznew AzFramework::GenericAssetHandler<AssetType1>(assetTypeName.data(), "SomeGroup", "someextension", asset1TypeId);
  1013. handler1->Register();
  1014. // Register a different type handler with the same Display Name
  1015. AZ::TypeId asset2TypeId = AZ::AzTypeInfo<AssetType2>::Uuid();
  1016. auto handler2 = aznew AzFramework::GenericAssetHandler<AssetType2>(assetTypeName.data(), "SomeGroup", "someextension", asset2TypeId);
  1017. handler2->Register();
  1018. // Verify one of the types is returned.
  1019. // Note that we have no control over which one is returned as it's an implementation detail of the AssetCatalog
  1020. {
  1021. AZ::Data::AssetType id;
  1022. AZ::Data::AssetCatalogRequestBus::BroadcastResult(id, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetTypeByDisplayName, assetTypeName);
  1023. EXPECT_TRUE((id == asset1TypeId) || (id == asset2TypeId));
  1024. }
  1025. // Delete both handlers
  1026. handler1->Unregister();
  1027. handler2->Unregister();
  1028. delete handler1;
  1029. delete handler2;
  1030. }
  1031. }