test: unit test for load reference util

aws · Jul 1, 2022 · 861281d · 861281d
1 parent 8a1eba8
commit 861281d
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/tst/CTA.Rules.Test/UtilsTest.cs b/tst/CTA.Rules.Test/UtilsTest.cs
@@ -5,7 +5,10 @@
 using System.IO;
 using System.Linq;
 using System.Threading.Tasks;
+using Codelyzer.Analysis;
+using Codelyzer.Analysis.Model;
 using CTA.Rules.Common.Helpers;
+using CTA.Rules.PortCore;
 
 namespace CTA.Rules.Test
 {
@@ -121,5 +124,34 @@ public void Test_Is_VisualBasic_Project()
             Assert.IsTrue(VisualBasicUtils.IsVisualBasicProject("C://user/john/repos/test.vbproj"));
             Assert.IsFalse(VisualBasicUtils.IsVisualBasicProject("vbprojproject.cs"));
         }
+
+        [Test]
+        public void TestLoadReferencesFromProject()
+        {
+            var analyzeResult = new AnalyzerResult
+            {
+                ProjectResult = new ProjectWorkspace("dummy.csproj")
+                {
+                    SourceFileResults = new UstList<RootUstNode>()
+                    {
+                        new RootUstNode()
+                        {
+                            Children = new UstList<UstNode>
+                            {
+                                new ImportsStatement { Identifier = "Newtonsoft.Json"},
+                                new UsingDirective {Identifier = "BouncyCastle.NetCore"}
+                            },
+                            References = new UstList<Reference>()
+                            {
+                                new Reference() {Namespace = "BouncyCastle.NetCore"}
+                            }
+                        }
+                    }
+                }
+            };
+
+            var references = PortCoreUtils.GetReferencesForProject(analyzeResult);
+            Assert.IsTrue(references.Count == 2);
+        }
     }
 }