From e780aa27c8037fadbc4a7c08466b52d0f4ac268a Mon Sep 17 00:00:00 2001
From: xuri <xuri.me@gmail.com>
Date: Thu, 4 Jul 2019 16:15:20 +0800
Subject: [PATCH] Add unit test for GroupSheets and UngroupSheets

---
 sheet_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/sheet_test.go b/sheet_test.go
index a7fd9e9..3a7f579 100644
--- a/sheet_test.go
+++ b/sheet_test.go
@@ -187,3 +187,24 @@ func TestDefinedName(t *testing.T) {
 	assert.Exactly(t, "Sheet1!$A$2:$D$5", f.GetDefinedName()[1].RefersTo)
 	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestDefinedName.xlsx")))
 }
+
+func TestGroupSheets(t *testing.T) {
+	f := excelize.NewFile()
+	sheets := []string{"Sheet2", "Sheet3"}
+	for _, sheet := range sheets {
+		f.NewSheet(sheet)
+	}
+	assert.EqualError(t, f.GroupSheets([]string{"Sheet1", "SheetN"}), "sheet SheetN is not exist")
+	assert.EqualError(t, f.GroupSheets([]string{"Sheet2", "Sheet3"}), "group worksheet must contain an active worksheet")
+	assert.NoError(t, f.GroupSheets([]string{"Sheet1", "Sheet2"}))
+	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestGroupSheets.xlsx")))
+}
+
+func TestUngroupSheets(t *testing.T) {
+	f := excelize.NewFile()
+	sheets := []string{"Sheet2", "Sheet3", "Sheet4", "Sheet5"}
+	for _, sheet := range sheets {
+		f.NewSheet(sheet)
+	}
+	assert.NoError(t, f.UngroupSheets())
+}