From a39833ec3cf2e35c30c7a78cda9c0b4954de4270 Mon Sep 17 00:00:00 2001
From: Artem Koloskov <artem@ironsoftware.com>
Date: Tue, 28 Nov 2023 16:45:29 +0700
Subject: [PATCH] Upstream feature: add tests for ISheet.RemoveValidationData
 method

---
 .../main/HSSF/UserModel/TestDataValidation.cs | 26 +++++++++++++++--
 .../XSSF/UserModel/TestXSSFDataValidation.cs  | 28 +++++++++++++++++--
 2 files changed, 50 insertions(+), 4 deletions(-)

diff --git a/testcases/main/HSSF/UserModel/TestDataValidation.cs b/testcases/main/HSSF/UserModel/TestDataValidation.cs
index 428f97078..3c5aa1088 100644
--- a/testcases/main/HSSF/UserModel/TestDataValidation.cs
+++ b/testcases/main/HSSF/UserModel/TestDataValidation.cs
@@ -464,6 +464,28 @@ public void TestGetDataValidationsFormula()
             Assert.AreEqual(double.NaN, c.Value2);
         }
 
-    }
+        [Test]
+        public void TestRemoveValidationData()
+        {
+            
+            HSSFWorkbook wb = new HSSFWorkbook();
+            HSSFSheet sheet = wb.CreateSheet() as HSSFSheet;
+            List<IDataValidation> list = sheet.GetDataValidations();
+            Assert.AreEqual(0, list.Count);
 
-}
\ No newline at end of file
+            IDataValidationHelper dataValidationHelper = sheet.GetDataValidationHelper();
+            IDataValidationConstraint constraint = dataValidationHelper.CreateCustomConstraint("A2:A3");
+            CellRangeAddressList AddressList = new CellRangeAddressList(0, 0, 0, 0);
+            IDataValidation validation = dataValidationHelper.CreateValidation(constraint, AddressList);
+            sheet.AddValidationData(validation);
+
+            list = sheet.GetDataValidations();
+            Assert.AreEqual(1, list.Count);
+
+            sheet.RemoveValidationData(validation);
+
+            list = sheet.GetDataValidations();
+            Assert.AreEqual(0, list.Count);
+        }
+    }
+}
diff --git a/testcases/ooxml/XSSF/UserModel/TestXSSFDataValidation.cs b/testcases/ooxml/XSSF/UserModel/TestXSSFDataValidation.cs
index 4ab0d20b5..b7d112834 100644
--- a/testcases/ooxml/XSSF/UserModel/TestXSSFDataValidation.cs
+++ b/testcases/ooxml/XSSF/UserModel/TestXSSFDataValidation.cs
@@ -372,6 +372,32 @@ public void TestCreateMultipleRegionsValidation()
             }
         }
 
+        [Test]
+        public void TestRemoveValidationData()
+        {
+            XSSFWorkbook wb = new XSSFWorkbook();
+            try {
+                XSSFSheet sheet = wb.CreateSheet() as XSSFSheet;
+                IDataValidationHelper dataValidationHelper = sheet.GetDataValidationHelper();
+                IDataValidationConstraint constraint = dataValidationHelper.CreateExplicitListConstraint(new string[] { "A" });
+                CellRangeAddressList cellRangeAddressList = new CellRangeAddressList();
+                cellRangeAddressList.AddCellRangeAddress(0, 0, 0, 0);
+                cellRangeAddressList.AddCellRangeAddress(0, 1, 0, 1);
+                cellRangeAddressList.AddCellRangeAddress(0, 2, 0, 2);
+                XSSFDataValidation dataValidation = dataValidationHelper.CreateValidation(constraint, cellRangeAddressList) as XSSFDataValidation;
+                sheet.AddValidationData(dataValidation);
+
+                Assert.AreEqual(1, sheet.GetDataValidations().Count);
+
+                sheet.RemoveValidationData(dataValidation);
+
+                Assert.AreEqual(0, sheet.GetDataValidations().Count);
+            }
+            finally {
+                wb.Close();
+            }
+        }
+
         private XSSFDataValidation CreateValidation(XSSFSheet sheet)
         {
             //create the cell that will have the validation applied
@@ -387,5 +413,3 @@ private XSSFDataValidation CreateValidation(XSSFSheet sheet)
 
     }
 }
-
-