Use same logic as add-build-to-channel when checking default channels (…

…#3846)
dotnet · Aug 20, 2024 · 4396011 · 4396011
1 parent 1497629
commit 4396011
Showing 1 changed file with 6 additions and 22 deletions.
diff --git a/src/Maestro/Microsoft.DotNet.Maestro.Tasks/src/PushMetadataToBuildAssetRegistry.cs b/src/Maestro/Microsoft.DotNet.Maestro.Tasks/src/PushMetadataToBuildAssetRegistry.cs
@@ -196,28 +196,12 @@ public async Task<bool> PushMetadataAsync(CancellationToken cancellationToken)
         private async Task<IEnumerable<DefaultChannel>> GetBuildDefaultChannelsAsync(IMaestroApi client,
             Client.Models.Build recordedBuild)
         {
-            var defaultChannels = new List<DefaultChannel>();
-            if (recordedBuild.GitHubBranch != null && recordedBuild.GitHubRepository != null)
-            {
-                defaultChannels.AddRange(
-                    await client.DefaultChannels.ListAsync(
-                        branch: recordedBuild.GitHubBranch,
-                        channelId: null,
-                        enabled: true,
-                        repository: recordedBuild.GitHubRepository
-                    ));
-            }
-
-            if (recordedBuild.AzureDevOpsBranch != null && recordedBuild.AzureDevOpsRepository != null)
-            {
-                defaultChannels.AddRange(
-                    await client.DefaultChannels.ListAsync(
-                        branch: recordedBuild.AzureDevOpsBranch,
-                        channelId: null,
-                        enabled: true,
-                        repository: recordedBuild.AzureDevOpsRepository
-                    ));
-            }
+            IEnumerable<DefaultChannel> defaultChannels = await client.DefaultChannels.ListAsync(
+                branch: recordedBuild.GetBranch(),
+                channelId: null,
+                enabled: true,
+                repository: recordedBuild.GetRepository()
+            );
 
             Log.LogMessage(MessageImportance.High, "Found the following default channels:");
             foreach (var defaultChannel in defaultChannels)