74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
|
package fileUtil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTar(t *testing.T) {
|
||
|
|
path := GetCurrentPath()
|
||
|
|
fmt.Printf("CurrPath:%s\n", path)
|
||
|
|
|
||
|
|
fileName1 := fmt.Sprintf("%s/%s", path, "test1.txt")
|
||
|
|
fileName2 := fmt.Sprintf("%s/%s", path, "test2.txt")
|
||
|
|
|
||
|
|
if err := WriteFile(path, "test1.txt", true, "first line"); err != nil {
|
||
|
|
t.Errorf("there should be no error, but now it is:%s", err)
|
||
|
|
}
|
||
|
|
if err := WriteFile(path, "test2.txt", true, "first line"); err != nil {
|
||
|
|
t.Errorf("there should be no error, but now it is:%s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
sourceList := make([]string, 0, 2)
|
||
|
|
sourceList = append(sourceList, fileName1)
|
||
|
|
sourceList = append(sourceList, fileName2)
|
||
|
|
target := fmt.Sprintf("%s/%s", path, "test.tar")
|
||
|
|
if err := Tar(sourceList, target); err != nil {
|
||
|
|
t.Errorf("There should be no error, but now it has:%s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if fileList, err := GetFileList(path); err != nil {
|
||
|
|
t.Errorf("There should be no error, but now it has:%s", err)
|
||
|
|
} else {
|
||
|
|
for _, item := range fileList {
|
||
|
|
fmt.Printf("item:%s\n", item)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
DeleteFile(fileName1)
|
||
|
|
DeleteFile(fileName2)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUntar(t *testing.T) {
|
||
|
|
path := GetCurrentPath()
|
||
|
|
fmt.Printf("CurrPath:%s\n", path)
|
||
|
|
|
||
|
|
source := fmt.Sprintf("%s/%s", path, "test.tar")
|
||
|
|
// target := path
|
||
|
|
target := ""
|
||
|
|
if err := Untar(source, target); err != nil {
|
||
|
|
t.Errorf("There should be no error, but now it has:%s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if fileList, err := GetFileList(path); err != nil {
|
||
|
|
t.Errorf("There should be no error, but now it has:%s", err)
|
||
|
|
} else {
|
||
|
|
for _, item := range fileList {
|
||
|
|
fmt.Printf("item:%s\n", item)
|
||
|
|
|
||
|
|
if strings.HasSuffix(item, "txt") {
|
||
|
|
if content, err := ReadFileContent(item); err != nil {
|
||
|
|
t.Errorf("There should be no error, but now it has:%s", err)
|
||
|
|
} else {
|
||
|
|
fmt.Printf("content:%s\n", content)
|
||
|
|
}
|
||
|
|
|
||
|
|
DeleteFile(item)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
DeleteFile(source)
|
||
|
|
}
|
||
|
|
}
|