test: cover all remote model collections

Author: ungrav

StabilityMatrix.Tests/Core/RemoteModelsTests.cs

@@ -1,4 +1,6 @@
+using System.Reflection;
 using StabilityMatrix.Core.Helper;
+using StabilityMatrix.Core.Models;
 
 namespace StabilityMatrix.Tests.Core;
 
@@ -15,12 +17,27 @@ public void ExecutableHuggingFaceModelsUsePinnedRevisions()
             ".pth",
         };
 
-        var unpinnedModels = RemoteModels
-            .UltralyticsModelFiles
-            .Where(model => model.DownloadableResource?.Url.Host == "huggingface.co")
-            .Where(model => executableExtensions.Contains(Path.GetExtension(model.DownloadableResource!.Value.Url.AbsolutePath)))
-            .Where(model => model.DownloadableResource!.Value.Url.AbsolutePath.Contains("/resolve/main/", StringComparison.Ordinal))
-            .Select(model => model.DownloadableResource!.Value.Url)
+        var resources = typeof(RemoteModels)
+            .GetProperties(BindingFlags.Public | BindingFlags.Static)
+            .Select(property => property.GetValue(null))
+            .SelectMany(
+                value =>
+                    value switch
+                    {
+                        IEnumerable<RemoteResource> remoteResources => remoteResources,
+                        IEnumerable<HybridModelFile> hybridModels => hybridModels
+                            .Where(model => model.DownloadableResource.HasValue)
+                            .Select(model => model.DownloadableResource!.Value),
+                        HybridModelFile { DownloadableResource: { } resource } => [resource],
+                        _ => [],
+                    }
+            );
+
+        var unpinnedModels = resources
+            .Where(resource => resource.Url.Host == "huggingface.co")
+            .Where(resource => executableExtensions.Contains(Path.GetExtension(resource.Url.AbsolutePath)))
+            .Where(resource => resource.Url.AbsolutePath.Contains("/resolve/main/", StringComparison.Ordinal))
+            .Select(resource => resource.Url)
             .ToArray();
 
         Assert.Empty(unpinnedModels);