|
2 | 2 | // Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information. |
3 | 3 |
|
4 | 4 | using System; |
| 5 | +using System.Collections.Generic; |
5 | 6 | using System.Threading.Tasks; |
6 | 7 | using Microsoft.Extensions.Logging; |
7 | 8 | using Moq; |
@@ -132,6 +133,121 @@ public async Task SkipsRepositorySignedPackages() |
132 | 133 | _telemetryService.Verify(t => t.TrackPackageRevalidationMarkedAsCompleted("Repository.Signed.Package", "1.0.0"), Times.Once); |
133 | 134 | } |
134 | 135 |
|
| 136 | + [Theory] |
| 137 | + [MemberData(nameof(SkipsPackagesWithTooManyVersionsData))] |
| 138 | + public async Task SkipsPackagesWithTooManyVersions(int? maximumPackageVersions, bool skipsPackageWithManyVersions) |
| 139 | + { |
| 140 | + _config.MaximumPackageVersions = maximumPackageVersions; |
| 141 | + |
| 142 | + // Arrange |
| 143 | + _validationContext.Mock(packageRevalidations: new[] |
| 144 | + { |
| 145 | + new PackageRevalidation |
| 146 | + { |
| 147 | + Key = 1, |
| 148 | + PackageId = "Package.With.Many.Versions", |
| 149 | + PackageNormalizedVersion = "1.0.0", |
| 150 | + Enqueued = null, |
| 151 | + Completed = false, |
| 152 | + }, |
| 153 | + new PackageRevalidation |
| 154 | + { |
| 155 | + Key = 2, |
| 156 | + PackageId = "Package.With.Many.Versions", |
| 157 | + PackageNormalizedVersion = "2.0.0", |
| 158 | + Enqueued = null, |
| 159 | + Completed = false, |
| 160 | + }, |
| 161 | + new PackageRevalidation |
| 162 | + { |
| 163 | + Key = 3, |
| 164 | + PackageId = "Package.With.Many.Versions", |
| 165 | + PackageNormalizedVersion = "3.0.0", |
| 166 | + Enqueued = null, |
| 167 | + Completed = false, |
| 168 | + }, |
| 169 | + new PackageRevalidation |
| 170 | + { |
| 171 | + Key = 4, |
| 172 | + PackageId = "Package", |
| 173 | + PackageNormalizedVersion = "1.0.0", |
| 174 | + Enqueued = null, |
| 175 | + Completed = false, |
| 176 | + }, |
| 177 | + }); |
| 178 | + |
| 179 | + _galleryContext.Mock(packages: new[] |
| 180 | + { |
| 181 | + new Package |
| 182 | + { |
| 183 | + PackageRegistration = new PackageRegistration { Id = "Package.With.Many.Versions" }, |
| 184 | + NormalizedVersion = "1.0.0", |
| 185 | + PackageStatusKey = PackageStatus.Available, |
| 186 | + }, |
| 187 | + new Package |
| 188 | + { |
| 189 | + PackageRegistration = new PackageRegistration { Id = "Package.With.Many.Versions" }, |
| 190 | + NormalizedVersion = "2.0.0", |
| 191 | + PackageStatusKey = PackageStatus.Available, |
| 192 | + }, |
| 193 | + new Package |
| 194 | + { |
| 195 | + PackageRegistration = new PackageRegistration { Id = "Package.With.Many.Versions" }, |
| 196 | + NormalizedVersion = "3.0.0", |
| 197 | + PackageStatusKey = PackageStatus.Available, |
| 198 | + }, |
| 199 | + new Package |
| 200 | + { |
| 201 | + PackageRegistration = new PackageRegistration { Id = "Package" }, |
| 202 | + NormalizedVersion = "1.0.0", |
| 203 | + PackageStatusKey = PackageStatus.Available, |
| 204 | + }, |
| 205 | + }); |
| 206 | + |
| 207 | + // Act |
| 208 | + var next = await _target.NextOrNullAsync(); |
| 209 | + |
| 210 | + // Assert |
| 211 | + if (skipsPackageWithManyVersions) |
| 212 | + { |
| 213 | + Assert.Equal("Package", next.PackageId); |
| 214 | + Assert.Equal("1.0.0", next.PackageNormalizedVersion); |
| 215 | + |
| 216 | + _telemetryService.Verify(t => t.TrackPackageRevalidationMarkedAsCompleted("Package.With.Many.Versions", "1.0.0"), Times.Never); |
| 217 | + _telemetryService.Verify(t => t.TrackPackageRevalidationMarkedAsCompleted("Package.With.Many.Versions", "2.0.0"), Times.Never); |
| 218 | + _telemetryService.Verify(t => t.TrackPackageRevalidationMarkedAsCompleted("Package.With.Many.Versions", "3.0.0"), Times.Never); |
| 219 | + } |
| 220 | + else |
| 221 | + { |
| 222 | + Assert.Equal("Package.With.Many.Versions", next.PackageId); |
| 223 | + Assert.Equal("1.0.0", next.PackageNormalizedVersion); |
| 224 | + } |
| 225 | + } |
| 226 | + |
| 227 | + public static IEnumerable<object[]> SkipsPackagesWithTooManyVersionsData() |
| 228 | + { |
| 229 | + // If the "MaximumPackageVersions" is null, no packages should be skipped for having too many versions. |
| 230 | + yield return new object[] |
| 231 | + { |
| 232 | + null, |
| 233 | + false |
| 234 | + }; |
| 235 | + |
| 236 | + // If "MaximumPackageVersions" is set, packages with less versions than the value should not be skipped. |
| 237 | + yield return new object[] |
| 238 | + { |
| 239 | + 100, |
| 240 | + false |
| 241 | + }; |
| 242 | + |
| 243 | + // If "MaximumPackageVersions" is set, packages with more versions than the value should be skipped. |
| 244 | + yield return new object[] |
| 245 | + { |
| 246 | + 2, |
| 247 | + true |
| 248 | + }; |
| 249 | + } |
| 250 | + |
135 | 251 | [Fact] |
136 | 252 | public async Task SkipsDeletedPackages() |
137 | 253 | { |
|
0 commit comments