gocommon/pkg/common/excel_test.go

package common

import (
	"fmt"
	"os"
	"testing"
)

func TestNewWriter360Excel(t *testing.T) {
	sheet := [][]string{
		{"大小", "...", "2020-01-02 10:11:46"},
		{"a bc", "</.<", "xxxxx"},
		{"b zcv ", "...", "/..   .."},
	}
	excel := NewWriter360Excel()
	err := excel.WriteSheet("中文1", []string{"中文1", "中文2", "中文3"}, func(r Row) (hasNext bool) {
		if len(sheet) == 0 {
			return false
		}
		for _, elem := range sheet[r.RowIndex()] {
			r.Write(elem)
		}
		return r.RowIndex() < len(sheet)-1
	})
	if err != nil {
		t.Fatal(err)
	}
	err = excel.WriteSheet("中文2", []string{"h1", "h2", "h3"}, func(r Row) (hasNext bool) {
		if len(sheet) == 0 {
			return false
		}
		for _, elem := range sheet[r.RowIndex()] {
			r.Write(elem)
		}
		return r.RowIndex() < len(sheet)-1
	})
	if err != nil {
		t.Fatal(err)
	}
	file, err := os.OpenFile("/data/test/new.xlsx", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
	if err != nil {
		t.Fatal(err)
	}
	defer file.Close()
	err = excel.FlushWrite(file)
	if err != nil {
		t.Fatal(err)
	}
}

func TestNewReader360Excel(t *testing.T) {
	file, err := os.OpenFile("/data/test/new.xlsx", os.O_RDONLY, 0)
	if err != nil {
		t.Fatal(err)
	}
	defer file.Close()
	excel, err := NewReader360Excel(file)
	if err != nil {
		t.Fatal(err)
	}
	{
		strings, err := excel.ReadSheet("中文1")
		if err != nil {
			t.Fatal(err)
		}
		for _, row := range strings {
			for _, col := range row {
				fmt.Printf("%v\t", col)
			}
			fmt.Println()
		}
	}
	{
		strings, err := excel.ReadSheet("中文2")
		if err != nil {
			t.Fatal(err)
		}
		for _, row := range strings {
			for _, col := range row {
				fmt.Printf("%v\t", col)
			}
			fmt.Println()
		}
	}
}