From c945b012db4ee1c04b26cfb1c171b6c3236cc510 Mon Sep 17 00:00:00 2001 From: kenjiuno Date: Wed, 2 Aug 2023 16:42:49 +0900 Subject: [PATCH] Adding unit tests: TestRemoveSheetMethod and TestRemoveSheetAtMethod --- .../ooxml/XSSF/UserModel/TestXSSFWorkbook.cs | 51 +++++++++++++++++++ 1 file changed, 51 insertions(+) diff --git a/testcases/ooxml/XSSF/UserModel/TestXSSFWorkbook.cs b/testcases/ooxml/XSSF/UserModel/TestXSSFWorkbook.cs index 69c4d59f7..ee1fc94f5 100644 --- a/testcases/ooxml/XSSF/UserModel/TestXSSFWorkbook.cs +++ b/testcases/ooxml/XSSF/UserModel/TestXSSFWorkbook.cs @@ -1196,5 +1196,56 @@ public void TestRemoveSheet() wb.Close(); } + [Test] + public void TestRemoveSheetMethod() + { + byte[] contents; + { + XSSFWorkbook wb = new XSSFWorkbook(); + var sheet1 = wb.CreateSheet("Sheet1"); + var sheet2 = wb.CreateSheet("Sheet2"); + Assert.True(wb.Remove(sheet2)); + Assert.AreEqual(1, wb.NumberOfSheets); + Assert.AreEqual("Sheet1", wb.GetSheetName(0)); + Assert.AreEqual(sheet1, wb.GetSheet("Sheet1")); + var stream = new MemoryStream(); + wb.Write(stream); + contents = stream.ToArray(); + wb.Close(); + } + + { + XSSFWorkbook wb = new XSSFWorkbook(new MemoryStream(contents, false)); + Assert.AreEqual(1, wb.NumberOfSheets); + Assert.AreEqual("Sheet1", wb.GetSheetName(0)); + wb.Close(); + } + } + + [Test] + public void TestRemoveSheetAtMethod() + { + byte[] contents; + { + XSSFWorkbook wb = new XSSFWorkbook(); + var sheet1 = wb.CreateSheet("Sheet1"); + var sheet2 = wb.CreateSheet("Sheet2"); + wb.RemoveSheetAt(1); + Assert.AreEqual(1, wb.NumberOfSheets); + Assert.AreEqual("Sheet1", wb.GetSheetName(0)); + Assert.AreEqual(sheet1, wb.GetSheet("Sheet1")); + var stream = new MemoryStream(); + wb.Write(stream); + contents = stream.ToArray(); + wb.Close(); + } + + { + XSSFWorkbook wb = new XSSFWorkbook(new MemoryStream(contents, false)); + Assert.AreEqual(1, wb.NumberOfSheets); + Assert.AreEqual("Sheet1", wb.GetSheetName(0)); + wb.Close(); + } + } } }