6e4b7cd59e
- Implement new process handlers for creating, reading, updating, copying, and deleting sheets in Excel files. - Enhance the Open function to automatically create directories for file paths if they do not exist. - Update README.md to include detailed documentation and examples for new sheet operations, improving user guidance.
123 lines
2.9 KiB
Go
123 lines
2.9 KiB
Go
package excel
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSheetOperations(t *testing.T) {
|
|
// Get test files and open the test file
|
|
files := testFiles(t)
|
|
handler, err := Open(files["test-01"], true) // Open in writable mode
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer Close(handler)
|
|
|
|
excel, err := Get(handler)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Test CreateSheet
|
|
t.Run("CreateSheet", func(t *testing.T) {
|
|
// Create a new sheet
|
|
idx, err := excel.CreateSheet("TestSheet1")
|
|
assert.NoError(t, err)
|
|
assert.Greater(t, idx, 0)
|
|
|
|
// Try to create a sheet with the same name (should fail)
|
|
_, err = excel.CreateSheet("TestSheet1")
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
// Test ReadSheet
|
|
t.Run("ReadSheet", func(t *testing.T) {
|
|
// Create test data
|
|
testData := [][]interface{}{
|
|
{"Header1", "Header2"},
|
|
{1, "Data1"},
|
|
{2, "Data2"},
|
|
}
|
|
|
|
// Write test data
|
|
err := excel.WriteAll("TestSheet1", "A1", testData)
|
|
assert.NoError(t, err)
|
|
|
|
// Read the data back
|
|
data, err := excel.ReadSheet("TestSheet1")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, len(testData), len(data))
|
|
|
|
// Try to read non-existent sheet
|
|
_, err = excel.ReadSheet("NonExistentSheet")
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
// Test UpdateSheet
|
|
t.Run("UpdateSheet", func(t *testing.T) {
|
|
newData := [][]interface{}{
|
|
{"NewHeader1", "NewHeader2"},
|
|
{3, "NewData1"},
|
|
{4, "NewData2"},
|
|
}
|
|
|
|
// Update existing sheet
|
|
err := excel.UpdateSheet("TestSheet1", newData)
|
|
assert.NoError(t, err)
|
|
|
|
// Read back and verify
|
|
data, err := excel.ReadSheet("TestSheet1")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, len(newData), len(data))
|
|
|
|
// Update non-existent sheet (should create new)
|
|
err = excel.UpdateSheet("NewSheet", newData)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
// Test ListSheets
|
|
t.Run("ListSheets", func(t *testing.T) {
|
|
sheets := excel.ListSheets()
|
|
assert.Contains(t, sheets, "TestSheet1")
|
|
assert.Contains(t, sheets, "NewSheet")
|
|
})
|
|
|
|
// Test CopySheet
|
|
t.Run("CopySheet", func(t *testing.T) {
|
|
// Copy existing sheet
|
|
err := excel.CopySheet("TestSheet1", "CopiedSheet")
|
|
assert.NoError(t, err)
|
|
|
|
// Verify the copy
|
|
originalData, err := excel.ReadSheet("TestSheet1")
|
|
assert.NoError(t, err)
|
|
copiedData, err := excel.ReadSheet("CopiedSheet")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, originalData, copiedData)
|
|
|
|
// Try to copy to existing sheet name (should fail)
|
|
err = excel.CopySheet("TestSheet1", "CopiedSheet")
|
|
assert.Error(t, err)
|
|
|
|
// Try to copy non-existent sheet (should fail)
|
|
err = excel.CopySheet("NonExistentSheet", "NewSheet2")
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
// Test DeleteSheet
|
|
t.Run("DeleteSheet", func(t *testing.T) {
|
|
// Delete existing sheet
|
|
err := excel.DeleteSheet("CopiedSheet")
|
|
assert.NoError(t, err)
|
|
|
|
// Verify sheet is deleted
|
|
sheets := excel.ListSheets()
|
|
assert.NotContains(t, sheets, "CopiedSheet")
|
|
|
|
// Try to delete non-existent sheet
|
|
err = excel.DeleteSheet("NonExistentSheet")
|
|
assert.Error(t, err)
|
|
})
|
|
}
|