ItemImageProviderTests.cs 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Globalization;
  4. using System.IO;
  5. using System.Linq;
  6. using System.Net;
  7. using System.Net.Http;
  8. using System.Net.Mime;
  9. using System.Text;
  10. using System.Text.RegularExpressions;
  11. using System.Threading;
  12. using System.Threading.Tasks;
  13. using MediaBrowser.Controller.Entities;
  14. using MediaBrowser.Controller.Library;
  15. using MediaBrowser.Controller.Providers;
  16. using MediaBrowser.Model.Configuration;
  17. using MediaBrowser.Model.Drawing;
  18. using MediaBrowser.Model.Entities;
  19. using MediaBrowser.Model.IO;
  20. using MediaBrowser.Model.MediaInfo;
  21. using MediaBrowser.Model.Providers;
  22. using MediaBrowser.Providers.Manager;
  23. using Microsoft.Extensions.Logging.Abstractions;
  24. using Moq;
  25. using Xunit;
  26. namespace Jellyfin.Providers.Tests.Manager
  27. {
  28. public partial class ItemImageProviderTests
  29. {
  30. private static readonly CompositeFormat _testDataImagePath = CompositeFormat.Parse("Test Data/Images/blank{0}.jpg");
  31. [GeneratedRegex("[0-9]+")]
  32. private static partial Regex NumbersRegex();
  33. [Fact]
  34. public void ValidateImages_PhotoEmptyProviders_NoChange()
  35. {
  36. var itemImageProvider = GetItemImageProvider(null, null);
  37. var changed = itemImageProvider.ValidateImages(new Photo(), Enumerable.Empty<ILocalImageProvider>(), null);
  38. Assert.False(changed);
  39. }
  40. [Fact]
  41. public void ValidateImages_EmptyItemEmptyProviders_NoChange()
  42. {
  43. ValidateImages_Test(ImageType.Primary, 0, true, 0, false, 0);
  44. }
  45. public static TheoryData<ImageType, int> GetImageTypesWithCount()
  46. {
  47. var theoryTypes = new TheoryData<ImageType, int>
  48. {
  49. // minimal test cases that hit different handling
  50. { ImageType.Primary, 1 },
  51. { ImageType.Backdrop, 2 }
  52. };
  53. return theoryTypes;
  54. }
  55. [Theory]
  56. [MemberData(nameof(GetImageTypesWithCount))]
  57. public void ValidateImages_EmptyItemAndPopulatedProviders_AddsImages(ImageType imageType, int imageCount)
  58. {
  59. ValidateImages_Test(imageType, 0, true, imageCount, true, imageCount);
  60. }
  61. [Theory]
  62. [MemberData(nameof(GetImageTypesWithCount))]
  63. public void ValidateImages_PopulatedItemWithGoodPathsAndEmptyProviders_NoChange(ImageType imageType, int imageCount)
  64. {
  65. ValidateImages_Test(imageType, imageCount, true, 0, false, imageCount);
  66. }
  67. [Theory]
  68. [MemberData(nameof(GetImageTypesWithCount))]
  69. public void ValidateImages_PopulatedItemWithBadPathsAndEmptyProviders_RemovesImage(ImageType imageType, int imageCount)
  70. {
  71. ValidateImages_Test(imageType, imageCount, false, 0, true, 0);
  72. }
  73. private void ValidateImages_Test(ImageType imageType, int initialImageCount, bool initialPathsValid, int providerImageCount, bool expectedChange, int expectedImageCount)
  74. {
  75. var item = GetItemWithImages(imageType, initialImageCount, initialPathsValid);
  76. var imageProvider = GetImageProvider(imageType, providerImageCount, true);
  77. var itemImageProvider = GetItemImageProvider(null, null);
  78. var actualChange = itemImageProvider.ValidateImages(item, new[] { imageProvider }, null);
  79. Assert.Equal(expectedChange, actualChange);
  80. Assert.Equal(expectedImageCount, item.GetImages(imageType).Count());
  81. }
  82. [Fact]
  83. public void MergeImages_EmptyItemNewImagesEmpty_NoChange()
  84. {
  85. var itemImageProvider = GetItemImageProvider(null, null);
  86. var changed = itemImageProvider.MergeImages(new Video(), Array.Empty<LocalImageInfo>(), new ImageRefreshOptions(Mock.Of<IDirectoryService>()));
  87. Assert.False(changed);
  88. }
  89. [Theory]
  90. [MemberData(nameof(GetImageTypesWithCount))]
  91. public void MergeImages_PopulatedItemWithGoodPathsAndPopulatedNewImages_AddsUpdatesImages(ImageType imageType, int imageCount)
  92. {
  93. // valid and not valid paths - should replace the valid paths with the invalid ones
  94. var item = GetItemWithImages(imageType, imageCount, true);
  95. var images = GetImages(imageType, imageCount, false);
  96. var itemImageProvider = GetItemImageProvider(null, null);
  97. var changed = itemImageProvider.MergeImages(item, images, new ImageRefreshOptions(Mock.Of<IDirectoryService>()));
  98. Assert.True(changed);
  99. // adds for types that allow multiple, replaces singular type images
  100. if (item.AllowsMultipleImages(imageType))
  101. {
  102. Assert.Equal(imageCount * 2, item.GetImages(imageType).Count());
  103. }
  104. else
  105. {
  106. Assert.Single(item.GetImages(imageType));
  107. Assert.Same(images[0].FileInfo.FullName, item.GetImages(imageType).First().Path);
  108. }
  109. }
  110. [Theory]
  111. [InlineData(ImageType.Primary, 1, false)]
  112. [InlineData(ImageType.Backdrop, 2, false)]
  113. [InlineData(ImageType.Primary, 1, true)]
  114. [InlineData(ImageType.Backdrop, 2, true)]
  115. public void MergeImages_PopulatedItemWithGoodPathsAndSameNewImages_ResetIfTimeChanges(ImageType imageType, int imageCount, bool updateTime)
  116. {
  117. var oldTime = new DateTime(1970, 1, 1);
  118. var updatedTime = updateTime ? new DateTime(2021, 1, 1) : oldTime;
  119. var fileSystem = new Mock<IFileSystem>();
  120. fileSystem.Setup(fs => fs.GetLastWriteTimeUtc(It.IsAny<FileSystemMetadata>()))
  121. .Returns(updatedTime);
  122. BaseItem.FileSystem = fileSystem.Object;
  123. // all valid paths - matching for strictly updating
  124. var item = GetItemWithImages(imageType, imageCount, true);
  125. // set size to non-zero to allow for image size reset to occur
  126. foreach (var image in item.GetImages(imageType))
  127. {
  128. image.DateModified = oldTime;
  129. image.Height = 1;
  130. image.Width = 1;
  131. }
  132. var images = GetImages(imageType, imageCount, true);
  133. var itemImageProvider = GetItemImageProvider(null, fileSystem);
  134. var changed = itemImageProvider.MergeImages(item, images, new ImageRefreshOptions(Mock.Of<IDirectoryService>()));
  135. if (updateTime)
  136. {
  137. Assert.True(changed);
  138. // before and after paths are the same, verify updated by size reset to 0
  139. var typedImages = item.GetImages(imageType).ToArray();
  140. Assert.Equal(imageCount, typedImages.Length);
  141. foreach (var image in typedImages)
  142. {
  143. Assert.Equal(updatedTime, image.DateModified);
  144. Assert.Equal(0, image.Height);
  145. Assert.Equal(0, image.Width);
  146. }
  147. }
  148. else
  149. {
  150. Assert.False(changed);
  151. }
  152. }
  153. [Theory]
  154. [InlineData(ImageType.Primary, 0)]
  155. [InlineData(ImageType.Primary, 1)]
  156. [InlineData(ImageType.Backdrop, 2)]
  157. public void RemoveImages_DeletesImages_WhenFound(ImageType imageType, int imageCount)
  158. {
  159. var item = GetItemWithImages(imageType, imageCount, false);
  160. var mockFileSystem = new Mock<IFileSystem>(MockBehavior.Strict);
  161. if (imageCount > 0)
  162. {
  163. mockFileSystem.Setup(fs => fs.DeleteFile("invalid path 0"))
  164. .Verifiable();
  165. }
  166. if (imageCount > 1)
  167. {
  168. mockFileSystem.Setup(fs => fs.DeleteFile("invalid path 1"))
  169. .Verifiable();
  170. }
  171. var itemImageProvider = GetItemImageProvider(Mock.Of<IProviderManager>(), mockFileSystem);
  172. var result = itemImageProvider.RemoveImages(item);
  173. Assert.Equal(imageCount != 0, result);
  174. Assert.Empty(item.GetImages(imageType));
  175. mockFileSystem.Verify();
  176. }
  177. [Theory]
  178. [InlineData(ImageType.Primary, 1, false)]
  179. [InlineData(ImageType.Backdrop, 2, false)]
  180. [InlineData(ImageType.Primary, 1, true)]
  181. [InlineData(ImageType.Backdrop, 2, true)]
  182. public async Task RefreshImages_PopulatedItemPopulatedProviderDynamic_UpdatesImagesIfForced(ImageType imageType, int imageCount, bool forceRefresh)
  183. {
  184. var item = GetItemWithImages(imageType, imageCount, false);
  185. var libraryOptions = GetLibraryOptions(item, imageType, imageCount);
  186. var imageResponse = new DynamicImageResponse
  187. {
  188. HasImage = true,
  189. Format = ImageFormat.Jpg,
  190. Path = "url path",
  191. Protocol = MediaProtocol.Http
  192. };
  193. var dynamicProvider = new Mock<IDynamicImageProvider>(MockBehavior.Strict);
  194. dynamicProvider.Setup(rp => rp.Name).Returns("MockDynamicProvider");
  195. dynamicProvider.Setup(rp => rp.GetSupportedImages(item))
  196. .Returns(new[] { imageType });
  197. dynamicProvider.Setup(rp => rp.GetImage(item, imageType, It.IsAny<CancellationToken>()))
  198. .ReturnsAsync(imageResponse);
  199. var refreshOptions = forceRefresh
  200. ? new ImageRefreshOptions(Mock.Of<IDirectoryService>())
  201. {
  202. ImageRefreshMode = MetadataRefreshMode.FullRefresh,
  203. ReplaceAllImages = true
  204. }
  205. : new ImageRefreshOptions(Mock.Of<IDirectoryService>());
  206. var itemImageProvider = GetItemImageProvider(null, new Mock<IFileSystem>());
  207. var result = await itemImageProvider.RefreshImages(item, libraryOptions, new List<IImageProvider> { dynamicProvider.Object }, refreshOptions, CancellationToken.None);
  208. Assert.Equal(forceRefresh, result.UpdateType.HasFlag(ItemUpdateType.ImageUpdate));
  209. if (forceRefresh)
  210. {
  211. // replaces multi-types
  212. Assert.Single(item.GetImages(imageType));
  213. }
  214. else
  215. {
  216. // adds to multi-types if room
  217. Assert.Equal(imageCount, item.GetImages(imageType).Count());
  218. }
  219. }
  220. [Theory]
  221. [InlineData(ImageType.Primary, 1, true, MediaProtocol.Http)]
  222. [InlineData(ImageType.Backdrop, 2, true, MediaProtocol.Http)]
  223. [InlineData(ImageType.Primary, 1, true, MediaProtocol.File)]
  224. [InlineData(ImageType.Backdrop, 2, true, MediaProtocol.File)]
  225. [InlineData(ImageType.Primary, 1, false, MediaProtocol.File)]
  226. [InlineData(ImageType.Backdrop, 2, false, MediaProtocol.File)]
  227. public async Task RefreshImages_EmptyItemPopulatedProviderDynamic_AddsImages(ImageType imageType, int imageCount, bool responseHasPath, MediaProtocol protocol)
  228. {
  229. // Has to exist for querying DateModified time on file, results stored but not checked so not populating
  230. BaseItem.FileSystem = Mock.Of<IFileSystem>();
  231. var item = new Video();
  232. var libraryOptions = GetLibraryOptions(item, imageType, imageCount);
  233. // Path must exist if set: is read in as a stream by AsyncFile.OpenRead
  234. var imageResponse = new DynamicImageResponse
  235. {
  236. HasImage = true,
  237. Format = ImageFormat.Jpg,
  238. Path = responseHasPath ? string.Format(CultureInfo.InvariantCulture, _testDataImagePath, 0) : null,
  239. Protocol = protocol
  240. };
  241. var dynamicProvider = new Mock<IDynamicImageProvider>(MockBehavior.Strict);
  242. dynamicProvider.Setup(rp => rp.Name).Returns("MockDynamicProvider");
  243. dynamicProvider.Setup(rp => rp.GetSupportedImages(item))
  244. .Returns(new[] { imageType });
  245. dynamicProvider.Setup(rp => rp.GetImage(item, imageType, It.IsAny<CancellationToken>()))
  246. .ReturnsAsync(imageResponse);
  247. var refreshOptions = new ImageRefreshOptions(Mock.Of<IDirectoryService>());
  248. var providerManager = new Mock<IProviderManager>(MockBehavior.Strict);
  249. providerManager.Setup(pm => pm.SaveImage(item, It.IsAny<Stream>(), It.IsAny<string>(), imageType, null, It.IsAny<CancellationToken>()))
  250. .Callback<BaseItem, Stream, string, ImageType, int?, CancellationToken>((callbackItem, _, _, callbackType, _, _) => callbackItem.SetImagePath(callbackType, 0, new FileSystemMetadata()))
  251. .Returns(Task.CompletedTask);
  252. providerManager.Setup(pm => pm.SaveImage(item, It.IsAny<string>(), It.IsAny<string>(), imageType, null, null, It.IsAny<CancellationToken>()))
  253. .Callback<BaseItem, string, string, ImageType, int?, bool?, CancellationToken>((callbackItem, _, _, callbackType, _, _, _) => callbackItem.SetImagePath(callbackType, 0, new FileSystemMetadata()))
  254. .Returns(Task.CompletedTask);
  255. var itemImageProvider = GetItemImageProvider(providerManager.Object, null);
  256. var result = await itemImageProvider.RefreshImages(item, libraryOptions, new List<IImageProvider> { dynamicProvider.Object }, refreshOptions, CancellationToken.None);
  257. Assert.True(result.UpdateType.HasFlag(ItemUpdateType.ImageUpdate));
  258. // dynamic provider unable to return multiple images
  259. Assert.Single(item.GetImages(imageType));
  260. if (protocol == MediaProtocol.Http)
  261. {
  262. Assert.Equal(imageResponse.Path, item.GetImagePath(imageType, 0));
  263. }
  264. }
  265. [Theory]
  266. [InlineData(ImageType.Primary, 1, false)]
  267. [InlineData(ImageType.Backdrop, 1, false)]
  268. [InlineData(ImageType.Backdrop, 2, false)]
  269. [InlineData(ImageType.Primary, 1, true)]
  270. [InlineData(ImageType.Backdrop, 1, true)]
  271. [InlineData(ImageType.Backdrop, 2, true)]
  272. public async Task RefreshImages_PopulatedItemPopulatedProviderRemote_UpdatesImagesIfForced(ImageType imageType, int imageCount, bool forceRefresh)
  273. {
  274. var item = GetItemWithImages(imageType, imageCount, false);
  275. var libraryOptions = GetLibraryOptions(item, imageType, imageCount);
  276. var remoteProvider = new Mock<IRemoteImageProvider>(MockBehavior.Strict);
  277. remoteProvider.Setup(rp => rp.Name).Returns("MockRemoteProvider");
  278. remoteProvider.Setup(rp => rp.GetSupportedImages(item))
  279. .Returns(new[] { imageType });
  280. var refreshOptions = forceRefresh
  281. ? new ImageRefreshOptions(Mock.Of<IDirectoryService>())
  282. {
  283. ImageRefreshMode = MetadataRefreshMode.FullRefresh,
  284. ReplaceAllImages = true
  285. }
  286. : new ImageRefreshOptions(Mock.Of<IDirectoryService>());
  287. var remoteInfo = new RemoteImageInfo[imageCount];
  288. for (int i = 0; i < imageCount; i++)
  289. {
  290. remoteInfo[i] = new RemoteImageInfo
  291. {
  292. Type = imageType,
  293. Url = "image url " + i
  294. };
  295. }
  296. var providerManager = new Mock<IProviderManager>(MockBehavior.Strict);
  297. providerManager.Setup(pm => pm.GetAvailableRemoteImages(It.IsAny<BaseItem>(), It.IsAny<RemoteImageQuery>(), It.IsAny<CancellationToken>()))
  298. .ReturnsAsync(remoteInfo);
  299. var itemImageProvider = GetItemImageProvider(providerManager.Object, new Mock<IFileSystem>());
  300. var result = await itemImageProvider.RefreshImages(item, libraryOptions, new List<IImageProvider> { remoteProvider.Object }, refreshOptions, CancellationToken.None);
  301. Assert.Equal(forceRefresh, result.UpdateType.HasFlag(ItemUpdateType.ImageUpdate));
  302. Assert.Equal(imageCount, item.GetImages(imageType).Count());
  303. foreach (var image in item.GetImages(imageType))
  304. {
  305. if (forceRefresh)
  306. {
  307. Assert.Matches("image url [0-9]", image.Path);
  308. }
  309. else
  310. {
  311. Assert.DoesNotMatch("image url [0-9]", image.Path);
  312. }
  313. }
  314. }
  315. [Theory]
  316. [InlineData(ImageType.Primary, 0, false)] // singular type only fetches if type is missing from item, no caching
  317. [InlineData(ImageType.Backdrop, 0, false)] // empty item, no cache to check
  318. [InlineData(ImageType.Backdrop, 1, false)] // populated item, cached so no download
  319. [InlineData(ImageType.Backdrop, 1, true)] // populated item, forced to download
  320. public async Task RefreshImages_NonStubItemPopulatedProviderRemote_DownloadsIfNecessary(ImageType imageType, int initialImageCount, bool fullRefresh)
  321. {
  322. var targetImageCount = 1;
  323. // Set path and media source manager so images will be downloaded (EnableImageStub will return false)
  324. var item = GetItemWithImages(imageType, initialImageCount, false);
  325. item.Path = "non-empty path";
  326. BaseItem.MediaSourceManager = Mock.Of<IMediaSourceManager>();
  327. // seek 2 so it won't short-circuit out of downloading when populated
  328. var libraryOptions = GetLibraryOptions(item, imageType, 2);
  329. const string Content = "Content";
  330. var remoteProvider = new Mock<IRemoteImageProvider>(MockBehavior.Strict);
  331. remoteProvider.Setup(rp => rp.Name).Returns("MockRemoteProvider");
  332. remoteProvider.Setup(rp => rp.GetSupportedImages(item))
  333. .Returns(new[] { imageType });
  334. remoteProvider.Setup(rp => rp.GetImageResponse(It.IsAny<string>(), It.IsAny<CancellationToken>()))
  335. .ReturnsAsync((string url, CancellationToken _) => new HttpResponseMessage
  336. {
  337. ReasonPhrase = url,
  338. StatusCode = HttpStatusCode.OK,
  339. Content = new StringContent(Content, Encoding.UTF8, MediaTypeNames.Image.Jpeg)
  340. });
  341. var refreshOptions = fullRefresh
  342. ? new ImageRefreshOptions(Mock.Of<IDirectoryService>())
  343. {
  344. ImageRefreshMode = MetadataRefreshMode.FullRefresh,
  345. ReplaceAllImages = true
  346. }
  347. : new ImageRefreshOptions(Mock.Of<IDirectoryService>());
  348. var remoteInfo = new RemoteImageInfo[targetImageCount];
  349. for (int i = 0; i < targetImageCount; i++)
  350. {
  351. remoteInfo[i] = new RemoteImageInfo
  352. {
  353. Type = imageType,
  354. Url = "image url " + i
  355. };
  356. }
  357. var providerManager = new Mock<IProviderManager>(MockBehavior.Strict);
  358. providerManager.Setup(pm => pm.GetAvailableRemoteImages(It.IsAny<BaseItem>(), It.IsAny<RemoteImageQuery>(), It.IsAny<CancellationToken>()))
  359. .ReturnsAsync(remoteInfo);
  360. providerManager.Setup(pm => pm.SaveImage(item, It.IsAny<Stream>(), It.IsAny<string>(), imageType, null, It.IsAny<CancellationToken>()))
  361. .Callback<BaseItem, Stream, string, ImageType, int?, CancellationToken>((callbackItem, _, _, callbackType, _, _) =>
  362. callbackItem.SetImagePath(callbackType, callbackItem.AllowsMultipleImages(callbackType) ? callbackItem.GetImages(callbackType).Count() : 0, new FileSystemMetadata()))
  363. .Returns(Task.CompletedTask);
  364. var fileSystem = new Mock<IFileSystem>();
  365. // match reported file size to image content length - condition for skipping already downloaded multi-images
  366. fileSystem.Setup(fs => fs.GetFileInfo(It.IsAny<string>()))
  367. .Returns(new FileSystemMetadata { Length = Content.Length });
  368. var itemImageProvider = GetItemImageProvider(providerManager.Object, fileSystem);
  369. var result = await itemImageProvider.RefreshImages(item, libraryOptions, new List<IImageProvider> { remoteProvider.Object }, refreshOptions, CancellationToken.None);
  370. Assert.Equal(initialImageCount == 0 || fullRefresh, result.UpdateType.HasFlag(ItemUpdateType.ImageUpdate));
  371. Assert.Equal(targetImageCount, item.GetImages(imageType).Count());
  372. }
  373. [Theory]
  374. [MemberData(nameof(GetImageTypesWithCount))]
  375. public async Task RefreshImages_EmptyItemPopulatedProviderRemoteExtras_LimitsImages(ImageType imageType, int imageCount)
  376. {
  377. var item = new Video();
  378. var libraryOptions = GetLibraryOptions(item, imageType, imageCount);
  379. var remoteProvider = new Mock<IRemoteImageProvider>(MockBehavior.Strict);
  380. remoteProvider.Setup(rp => rp.Name).Returns("MockRemoteProvider");
  381. remoteProvider.Setup(rp => rp.GetSupportedImages(item))
  382. .Returns(new[] { imageType });
  383. var refreshOptions = new ImageRefreshOptions(Mock.Of<IDirectoryService>());
  384. // populate remote with double the required images to verify count is trimmed to the library option count
  385. var remoteInfoCount = imageCount * 2;
  386. var remoteInfo = new RemoteImageInfo[remoteInfoCount];
  387. for (int i = 0; i < remoteInfoCount; i++)
  388. {
  389. remoteInfo[i] = new RemoteImageInfo
  390. {
  391. Type = imageType,
  392. Url = "image url " + i
  393. };
  394. }
  395. var providerManager = new Mock<IProviderManager>(MockBehavior.Strict);
  396. providerManager.Setup(pm => pm.GetAvailableRemoteImages(It.IsAny<BaseItem>(), It.IsAny<RemoteImageQuery>(), It.IsAny<CancellationToken>()))
  397. .ReturnsAsync(remoteInfo);
  398. var itemImageProvider = GetItemImageProvider(providerManager.Object, null);
  399. var result = await itemImageProvider.RefreshImages(item, libraryOptions, new List<IImageProvider> { remoteProvider.Object }, refreshOptions, CancellationToken.None);
  400. Assert.True(result.UpdateType.HasFlag(ItemUpdateType.ImageUpdate));
  401. var actualImages = item.GetImages(imageType).ToList();
  402. Assert.Equal(imageCount, actualImages.Count);
  403. // images from the provider manager are sorted by preference (earlier images are higher priority) so we can verify that low url numbers are chosen
  404. foreach (var image in actualImages)
  405. {
  406. var index = int.Parse(NumbersRegex().Match(image.Path).ValueSpan, NumberStyles.Integer, CultureInfo.InvariantCulture);
  407. Assert.True(index < imageCount);
  408. }
  409. }
  410. [Theory]
  411. [MemberData(nameof(GetImageTypesWithCount))]
  412. public async Task RefreshImages_PopulatedItemEmptyProviderRemoteFullRefresh_DoesntClearImages(ImageType imageType, int imageCount)
  413. {
  414. var item = GetItemWithImages(imageType, imageCount, false);
  415. var libraryOptions = GetLibraryOptions(item, imageType, imageCount);
  416. var remoteProvider = new Mock<IRemoteImageProvider>(MockBehavior.Strict);
  417. remoteProvider.Setup(rp => rp.Name).Returns("MockRemoteProvider");
  418. remoteProvider.Setup(rp => rp.GetSupportedImages(item))
  419. .Returns(new[] { imageType });
  420. var refreshOptions = new ImageRefreshOptions(Mock.Of<IDirectoryService>())
  421. {
  422. ImageRefreshMode = MetadataRefreshMode.FullRefresh,
  423. ReplaceAllImages = true
  424. };
  425. var itemImageProvider = GetItemImageProvider(Mock.Of<IProviderManager>(), null);
  426. var result = await itemImageProvider.RefreshImages(item, libraryOptions, new List<IImageProvider> { remoteProvider.Object }, refreshOptions, CancellationToken.None);
  427. Assert.False(result.UpdateType.HasFlag(ItemUpdateType.ImageUpdate));
  428. Assert.Equal(imageCount, item.GetImages(imageType).Count());
  429. }
  430. [Theory]
  431. [InlineData(9, false)]
  432. [InlineData(10, true)]
  433. [InlineData(null, true)]
  434. public async Task RefreshImages_ProviderRemote_FiltersByWidth(int? remoteImageWidth, bool expectedToUpdate)
  435. {
  436. var imageType = ImageType.Primary;
  437. var item = new Video();
  438. var libraryOptions = new LibraryOptions
  439. {
  440. TypeOptions = new[]
  441. {
  442. new TypeOptions
  443. {
  444. Type = item.GetType().Name,
  445. ImageOptions = new[]
  446. {
  447. new ImageOption
  448. {
  449. Type = imageType,
  450. MinWidth = 10
  451. }
  452. }
  453. }
  454. }
  455. };
  456. var remoteProvider = new Mock<IRemoteImageProvider>(MockBehavior.Strict);
  457. remoteProvider.Setup(rp => rp.Name).Returns("MockRemoteProvider");
  458. remoteProvider.Setup(rp => rp.GetSupportedImages(item))
  459. .Returns(new[] { imageType });
  460. var refreshOptions = new ImageRefreshOptions(Mock.Of<IDirectoryService>());
  461. // set width on image from remote
  462. var remoteInfo = new[]
  463. {
  464. new RemoteImageInfo()
  465. {
  466. Type = imageType,
  467. Url = "image url",
  468. Width = remoteImageWidth
  469. }
  470. };
  471. var providerManager = new Mock<IProviderManager>(MockBehavior.Strict);
  472. providerManager.Setup(pm => pm.GetAvailableRemoteImages(It.IsAny<BaseItem>(), It.IsAny<RemoteImageQuery>(), It.IsAny<CancellationToken>()))
  473. .ReturnsAsync(remoteInfo);
  474. var itemImageProvider = GetItemImageProvider(providerManager.Object, null);
  475. var result = await itemImageProvider.RefreshImages(item, libraryOptions, new List<IImageProvider> { remoteProvider.Object }, refreshOptions, CancellationToken.None);
  476. Assert.Equal(expectedToUpdate, result.UpdateType.HasFlag(ItemUpdateType.ImageUpdate));
  477. }
  478. private static ItemImageProvider GetItemImageProvider(IProviderManager? providerManager, Mock<IFileSystem>? mockFileSystem)
  479. {
  480. // strict to ensure this isn't accidentally used where a prepared mock is intended
  481. providerManager ??= Mock.Of<IProviderManager>(MockBehavior.Strict);
  482. // BaseItem.ValidateImages depends on the directory service being able to list directory contents, give it the expected valid file paths
  483. mockFileSystem ??= new Mock<IFileSystem>(MockBehavior.Strict);
  484. mockFileSystem.Setup(fs => fs.GetFilePaths(It.IsAny<string>(), It.IsAny<bool>()))
  485. .Returns(new[]
  486. {
  487. string.Format(CultureInfo.InvariantCulture, _testDataImagePath, 0),
  488. string.Format(CultureInfo.InvariantCulture, _testDataImagePath, 1)
  489. });
  490. return new ItemImageProvider(new NullLogger<ItemImageProvider>(), providerManager, mockFileSystem.Object);
  491. }
  492. private static Video GetItemWithImages(ImageType type, int count, bool validPaths)
  493. {
  494. // Has to exist for querying DateModified time on file, results stored but not checked so not populating
  495. BaseItem.FileSystem ??= Mock.Of<IFileSystem>();
  496. var item = new Mock<Video>
  497. {
  498. CallBase = true
  499. };
  500. item.Setup(m => m.IsSaveLocalMetadataEnabled()).Returns(false);
  501. item.Setup(m => m.GetInternalMetadataPath()).Returns(string.Empty);
  502. var path = validPaths ? _testDataImagePath.Format : "invalid path {0}";
  503. for (int i = 0; i < count; i++)
  504. {
  505. item.Object.SetImagePath(type, i, new FileSystemMetadata
  506. {
  507. FullName = string.Format(CultureInfo.InvariantCulture, path, i),
  508. });
  509. }
  510. return item.Object;
  511. }
  512. private static ILocalImageProvider GetImageProvider(ImageType type, int count, bool validPaths)
  513. {
  514. var images = GetImages(type, count, validPaths);
  515. var imageProvider = new Mock<ILocalImageProvider>();
  516. imageProvider.Setup(ip => ip.GetImages(It.IsAny<BaseItem>(), It.IsAny<IDirectoryService>()))
  517. .Returns(images);
  518. return imageProvider.Object;
  519. }
  520. /// <summary>
  521. /// Creates a list of <see cref="LocalImageInfo"/> references of the specified type and size, optionally pointing to files that exist.
  522. /// </summary>
  523. private static LocalImageInfo[] GetImages(ImageType type, int count, bool validPaths)
  524. {
  525. var path = validPaths ? _testDataImagePath.Format : "invalid path {0}";
  526. var images = new LocalImageInfo[count];
  527. for (int i = 0; i < count; i++)
  528. {
  529. images[i] = new LocalImageInfo
  530. {
  531. Type = type,
  532. FileInfo = new FileSystemMetadata
  533. {
  534. FullName = string.Format(CultureInfo.InvariantCulture, path, i)
  535. }
  536. };
  537. }
  538. return images;
  539. }
  540. /// <summary>
  541. /// Generates a <see cref="LibraryOptions"/> object that will allow for the requested number of images for the target type.
  542. /// </summary>
  543. private static LibraryOptions GetLibraryOptions(BaseItem item, ImageType type, int count)
  544. {
  545. return new LibraryOptions
  546. {
  547. TypeOptions = new[]
  548. {
  549. new TypeOptions
  550. {
  551. Type = item.GetType().Name,
  552. ImageOptions = new[]
  553. {
  554. new ImageOption
  555. {
  556. Type = type,
  557. Limit = count,
  558. }
  559. }
  560. }
  561. }
  562. };
  563. }
  564. }
  565. }