Fix null check in IncludeFlagUtils and update tests with proper types

Co-authored-by: nkolev92 <[email protected]>

Author: Copilot

src/NuGet.Core/NuGet.Commands/RestoreCommand/Utility/IncludeFlagUtils.cs

@@ -65,6 +65,7 @@ internal static Dictionary<string, LibraryIncludeFlags> FlattenDependencyTypes(
             static bool IsDependencyPruned(LibraryDependency dependency, IReadOnlyDictionary<string, PrunePackageReference> packagesToPrune)
             {
                 if (packagesToPrune?.TryGetValue(dependency.Name, out PrunePackageReference packageToPrune) == true
+                    && dependency.LibraryRange.VersionRange != null
                     && dependency.LibraryRange.VersionRange.Satisfies(packageToPrune.VersionRange.MaxVersion))
                 {
                     return true;

test/NuGet.Core.FuncTests/NuGet.Commands.FuncTest/RestoreCommand_PrunePackageReference.cs

@@ -3,13 +3,16 @@
 
 using System;
 using System.Collections.Generic;
+using System.Collections.Immutable;
 using System.IO;
 using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using FluentAssertions;
 using NuGet.Commands.Test;
 using NuGet.Common;
+using NuGet.Configuration;
+using NuGet.Frameworks;
 using NuGet.LibraryModel;
 using NuGet.Packaging;
 using NuGet.ProjectModel;
@@ -1456,23 +1459,27 @@ public void AnalyzePruningResults_WithMultiTargetingProjectAndVariousVersions_Wa
         public void AnalyzePruningResults_WithCPMAndNullVersionRange_DoesNotThrowNullReferenceException()
         {
             // Test for issue: NullReferenceException in package pruning when using CPM and a version is not specified
-            var rootProject = @"
+            var projectSpec = new PackageSpec(new List<TargetFrameworkInformation>
+            {
+                new TargetFrameworkInformation
                 {
-                  ""frameworks"": {
-                    ""net10.0"": {
-                        ""dependencies"": {
-                                ""A"": {
-                                    ""target"": ""Package"",
-                                },
-                        },
-                        ""packagesToPrune"": {
-                            ""A"" : ""(,1.0.0]"" 
-                        }
+                    FrameworkName = NuGetFramework.Parse("net10.0"),
+                    TargetAlias = "net10.0",
+                    Dependencies = ImmutableArray.Create(new LibraryDependency
+                    {
+                        LibraryRange = new LibraryRange("A", null, LibraryDependencyTarget.Package)
+                    }),
+                    PackagesToPrune = new Dictionary<string, PrunePackageReference>
+                    {
+                        { "A", new PrunePackageReference("A", VersionRange.Parse("(,1.0.0]")) }
                     }
-                  }
-                }";
+                }
+            })
+            {
+                Name = "Project1",
+                FilePath = Path.Combine(Path.GetTempPath(), "Project1.csproj")
+            };
 
-            var projectSpec = ProjectTestHelpers.GetPackageSpecWithProjectNameAndSpec("Project1", Path.GetTempPath(), rootProject);
             var testLogger = new TestLogger();
             var testEvent = new TelemetryEvent("dummyEvent");
 
@@ -1501,24 +1508,43 @@ await SimpleTestPackageUtility.CreateFolderFeedV3Async(
                 PackageSaveMode.Defaultv3,
                 packageA);
 
-            var rootProject = @"
-        {
-          ""frameworks"": {
-            ""net10.0"": {
-                ""dependencies"": {
-                        ""A"": {
-                            ""target"": ""Package"",
-                        },
-                },
-                ""packagesToPrune"": {
-                    ""A"" : ""(,1.0.0]"" 
+            // Create PackageSpec manually with null version range to simulate CPM without version
+            var projectPath = Path.Combine(pathContext.SolutionRoot, "Project1", "Project1.csproj");
+            var projectSpec = new PackageSpec(new List<TargetFrameworkInformation>
+            {
+                new TargetFrameworkInformation
+                {
+                    FrameworkName = NuGetFramework.Parse("net10.0"),
+                    TargetAlias = "net10.0",
+                    Dependencies = ImmutableArray.Create(new LibraryDependency
+                    {
+                        LibraryRange = new LibraryRange("A", null, LibraryDependencyTarget.Package)
+                    }),
+                    PackagesToPrune = new Dictionary<string, PrunePackageReference>
+                    {
+                        { "A", new PrunePackageReference("A", VersionRange.Parse("(,1.0.0]")) }
+                    }
                 }
-            }
-          }
-        }";
+            })
+            {
+                Name = "Project1",
+                FilePath = projectPath,
+                RestoreMetadata = new ProjectRestoreMetadata
+                {
+                    ProjectStyle = ProjectStyle.PackageReference,
+                    ProjectPath = projectPath,
+                    ProjectName = "Project1",
+                    ProjectUniqueName = projectPath,
+                    OutputPath = Path.Combine(pathContext.SolutionRoot, "Project1", "obj"),
+                    OriginalTargetFrameworks = ["net10.0"],
+                    ConfigFilePaths = [],
+                    PackagesPath = pathContext.UserPackagesFolder,
+                    Sources = [new PackageSource(pathContext.PackageSource)],
+                    FallbackFolders = [],
+                    CentralPackageVersionsEnabled = true
+                }
+            };
 
-            // Setup project
-            var projectSpec = ProjectTestHelpers.GetPackageSpecWithProjectNameAndSpec("Project1", pathContext.SolutionRoot, rootProject);
             var testLogger = new TestLogger();
 
             // Act - This should not throw a NullReferenceException during restore