Upstream feature: add tests for ISheet.RemoveValidationData method

antony-liu · Feb 9, 2024 · a39833e · a39833e
1 parent fefe574
commit a39833e
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 4 deletions.
diff --git 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);
 
-}
+            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
@@ -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)
 
     }
 }
-
-