ItemImageProviderTests.cs 28 KB

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