Test for Q2331

febinjoy · Jan 2, 2024 · 9d978a7 · 9d978a7
1 parent 8178d35
commit 9d978a7
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 3 deletions.
diff --git a/LeetCodeSolutions/Questions/Easy/Q2331.cs b/LeetCodeSolutions/Questions/Easy/Q2331.cs
@@ -17,6 +17,7 @@ private class TreeNode
             public TreeNode left;
             public TreeNode right;
 #pragma warning disable CS8625 // Cannot convert null literal to non-nullable reference type.
+
             public TreeNode(int val = 0, TreeNode left = null, TreeNode right = null)
 #pragma warning restore CS8625 // Cannot convert null literal to non-nullable reference type.
             {
@@ -38,7 +39,7 @@ public IResult Execute()
         {
             TreeNode node = new TreeNode(2, new TreeNode(1), new TreeNode(3, new TreeNode(0), new TreeNode(1)));
             bool output = EvaluateTree(node);
-            if (output == true)
+            if (output)
                 return new SuccessResult();
             else
                 return new ErrorResult("Expected True. But was False.");
@@ -56,5 +57,4 @@ private bool EvaluateTree(TreeNode root)
             return true;
         }
     }
-}
-
+}
diff --git a/LeetCodeSolutionsTest/QuestionServiceTests.cs b/LeetCodeSolutionsTest/QuestionServiceTests.cs
@@ -1,4 +1,6 @@
 using LeetCodeSolutions;
+using LeetCodeSolutions.Classes;
+using LeetCodeSolutions.Enums;
 using LeetCodeSolutions.Exceptions;
 using LeetCodeSolutions.Interfaces;
 using LeetCodeSolutions.Questions.Easy;
@@ -11,12 +13,18 @@ public class QuestionServiceTests
     {
         private QuestionService questionService;
 
+        #region Setup
+
         [SetUp]
         public void Setup()
         {
             questionService = new QuestionService();
         }
 
+        #endregion Setup
+
+        #region Exception Tests
+
         [Test]
         public void ExecuteQuestion_UnknownQuestionNumber_ThrowsQuestionNotFoundException()
         {
@@ -41,5 +49,25 @@ public void ExecuteQuestion_NotImplementedQuestion_ThrowsNotImplementedException
             // Act + Assert
             Assert.Throws<NotImplementedException>(() => questionService.ExecuteQuestion(questionNumber));
         }
+
+        #endregion Exception Tests
+
+        #region Easy Questions
+
+        [Test]
+        public void ExecuteQuestion_Q2331_ReturnsSuccessResult()
+        {
+            // Arrange
+            int questionNumber = 2331;
+
+            // Act
+            IResult result = questionService.ExecuteQuestion(questionNumber);
+
+            // Assert
+            Assert.IsInstanceOf<SuccessResult>(result);
+            Assert.That(result.Status, Is.EqualTo(ResultStatus.Passed));
+        }
+
+        #endregion Easy Questions
     }
 }