WinGet Source COM Api

src/AppInstallerCLIE2ETests/Interop/GroupPolicyForInterop.cs

@@ -9,6 +9,7 @@ namespace AppInstallerCLIE2ETests.Interop
     using System;
     using System.IO;
     using System.Text;
+    using System.Threading;
     using System.Threading.Tasks;
     using AppInstallerCLIE2ETests.Helpers;
     using Microsoft.Management.Deployment;
@@ -51,6 +52,16 @@ public void CleanUp()
             GroupPolicyHelper.DeleteExistingPolicies();
         }
 
+        /// <summary>
+        /// Test class Tear down.
+        /// </summary>
+        [OneTimeTearDown]
+        public void TestClassTearDown()
+        {
+            // Restore the tests source if it was removed as the affects subsequent tests.
+            TestCommon.SetupTestSource();
+        }
+
         /// <summary>
         /// Validates disabling WinGetPolicy should block COM/WinRT Objects creation (InProcess and OutOfProcess).
         /// </summary>
@@ -91,6 +102,14 @@ public void DisableWinGetPolicy()
             Assert.AreEqual(Constants.BlockByWinGetPolicyErrorMessage, groupPolicyException.Message);
             Assert.AreEqual(Constants.ErrorCode.ERROR_BLOCKED_BY_POLICY, groupPolicyException.HResult);
 
+            groupPolicyException = Assert.Catch<GroupPolicyException>(() => { AddPackageCatalogOptions packageManagerSettings = this.TestFactory.CreateAddPackageCatalogOptions(); });
+            Assert.AreEqual(Constants.BlockByWinGetPolicyErrorMessage, groupPolicyException.Message);
+            Assert.AreEqual(Constants.ErrorCode.ERROR_BLOCKED_BY_POLICY, groupPolicyException.HResult);
+
+            groupPolicyException = Assert.Catch<GroupPolicyException>(() => { RemovePackageCatalogOptions packageManagerSettings = this.TestFactory.CreateRemovePackageCatalogOptions(); });
+            Assert.AreEqual(Constants.BlockByWinGetPolicyErrorMessage, groupPolicyException.Message);
+            Assert.AreEqual(Constants.ErrorCode.ERROR_BLOCKED_BY_POLICY, groupPolicyException.HResult);
+
             // PackageManagerSettings is not implemented in context OutOfProcDev
             if (this.TestFactory.Context == ClsidContext.InProc)
             {
@@ -159,6 +178,66 @@ public async Task DisableWinGetCommandLineInterfacesPolicy()
             var packageVersion = "2.0.0.0";
             string downloadDir = Path.Combine(TestCommon.GetDefaultDownloadDirectory(), $"{Constants.ModifyRepairInstaller}_{packageVersion}");
             TestCommon.AssertInstallerDownload(downloadDir, "TestModifyRepair", packageVersion, ProcessorArchitecture.X86, TestCommon.Scope.Unknown, PackageInstallerType.Burn, "en-US");
+
+            // Add, update and remove package catalog
+            await this.AddUpdateRemovePackageCatalog();
+        }
+
+        private async Task AddUpdateRemovePackageCatalog()
+        {
+            // Remove the tests source if it exists.
+            await this.RemovePackageCatalog();
+
+            PackageManager packageManager = this.TestFactory.CreatePackageManager();
+
+            // Add package catalog
+            AddPackageCatalogOptions options = this.TestFactory.CreateAddPackageCatalogOptions();
+            options.SourceUri = Constants.TestSourceUrl;
+            options.Name = Constants.TestSourceName;
+            options.TrustLevel = PackageCatalogTrustLevel.Trusted;
+            options.AcceptSourceAgreements = true;
+
+            var addCatalogResult = await packageManager.AddPackageCatalogAsync(options);
+            Assert.IsNotNull(addCatalogResult);
+            Assert.AreEqual(AddPackageCatalogStatus.Ok, addCatalogResult.Status);
+
+            // Get package catalog
+            var packageCatalog = packageManager.GetPackageCatalogByName(options.Name);
+
+            Assert.IsNotNull(packageCatalog);
+            Assert.AreEqual(options.Name, packageCatalog.Info.Name);
+            Assert.AreEqual(options.SourceUri, packageCatalog.Info.Argument);
+            var lastUpdatedTime = packageCatalog.Info.LastUpdateTime;
+
+            // Update package catalog
+            // Sleep for 30 seconds to make sure the last updated time is different after the refresh.
+            Thread.Sleep(TimeSpan.FromSeconds(30));
+
+            var updateResult = await packageCatalog.RefreshPackageCatalogAsync();
+            Assert.IsNotNull(updateResult);
+            Assert.AreEqual(RefreshPackageCatalogStatus.Ok, updateResult.Status);
+
+            packageCatalog = packageManager.GetPackageCatalogByName(options.Name);
+            Assert.IsTrue(packageCatalog.Info.LastUpdateTime > lastUpdatedTime);
+
+            // Remove package catalog
+            await this.RemovePackageCatalog();
+        }
+
+        private async Task RemovePackageCatalog()
+        {
+            PackageManager packageManager = this.TestFactory.CreatePackageManager();
+
+            // Remove the tests source if it exists.
+            RemovePackageCatalogOptions removePackageCatalogOptions = this.TestFactory.CreateRemovePackageCatalogOptions();
+            removePackageCatalogOptions.Name = Constants.TestSourceName;
+
+            var removeCatalogResult = await packageManager.RemovePackageCatalogAsync(removePackageCatalogOptions);
+            Assert.IsNotNull(removeCatalogResult);
+            Assert.AreEqual(RemovePackageCatalogStatus.Ok, removeCatalogResult.Status);
+
+            var packageCatalog = packageManager.GetPackageCatalogByName(removePackageCatalogOptions.Name);
+            Assert.IsNull(packageCatalog);
         }
     }
 }