mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-22 15:10:13 +00:00
12159e36fa
* Added Zip archive support * extended test coverage * extended test coverage for unzip and zip * cleaned up Zip() and moved walk to non-exported method increased test coverage * underscore unused params * remove unneeded var * check against correct error and return after * Errorf * z -> f * fixed wording on tests * handle fine handler close error
106 lines
2.6 KiB
Go
106 lines
2.6 KiB
Go
package archive
|
|
|
|
import (
|
|
"archive/zip"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
tempDir string
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
var err error
|
|
tempDir, err = ioutil.TempDir("", "gct-temp")
|
|
if err != nil {
|
|
fmt.Printf("failed to create tempDir: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
t := m.Run()
|
|
err = os.RemoveAll(tempDir)
|
|
if err != nil {
|
|
fmt.Printf("Failed to remove tempDir %v", err)
|
|
}
|
|
os.Exit(t)
|
|
}
|
|
|
|
func TestUnZip(t *testing.T) {
|
|
zipFile := filepath.Join("..", "..", "..", "testdata", "testdata.zip")
|
|
files, err := UnZip(zipFile, tempDir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(files) != 2 {
|
|
t.Fatalf("expected 2 files to be extracted received: %v ", len(files))
|
|
}
|
|
|
|
zipFile = filepath.Join("..", "..", "..", "testdata", "zip-slip.zip")
|
|
_, err = UnZip(zipFile, tempDir)
|
|
if err == nil {
|
|
t.Fatal("Zip() expected to error due to ZipSlip detection but extracted successfully")
|
|
}
|
|
|
|
zipFile = filepath.Join("..", "..", "..", "testdata", "configtest.json")
|
|
_, err = UnZip(zipFile, tempDir)
|
|
if err == nil {
|
|
t.Fatal("Zip() expected to error due to invalid zipfile")
|
|
}
|
|
}
|
|
|
|
func TestZip(t *testing.T) {
|
|
singleFile := filepath.Join("..", "..", "..", "testdata", "configtest.json")
|
|
outFile := filepath.Join(tempDir, "out.zip")
|
|
err := Zip(singleFile, outFile)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
o, err := UnZip(outFile, tempDir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(o) != 1 {
|
|
t.Fatalf("expected 1 files to be extracted received: %v ", len(o))
|
|
}
|
|
|
|
folder := filepath.Join("..", "..", "..", "testdata", "http_mock")
|
|
outFolderZip := filepath.Join(tempDir, "out_folder.zip")
|
|
err = Zip(folder, outFolderZip)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
o, err = UnZip(outFolderZip, tempDir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if filepath.Base(o[0]) != "binance.json" || filepath.Base(o[4]) != "localbitcoins.json" {
|
|
t.Fatal("unexpected archive result received")
|
|
}
|
|
if len(o) != 6 {
|
|
t.Fatalf("expected 2 files to be extracted received: %v ", len(o))
|
|
}
|
|
|
|
folder = filepath.Join("..", "..", "..", "testdata", "invalid_file.json")
|
|
outFolderZip = filepath.Join(tempDir, "invalid.zip")
|
|
err = Zip(folder, outFolderZip)
|
|
if err == nil {
|
|
t.Fatal("expected IsNotExistError on invalid file")
|
|
}
|
|
|
|
addFilesToZip = addFilesToZipTestWrapper
|
|
folder = filepath.Join("..", "..", "..", "testdata", "http_mock")
|
|
outFolderZip = filepath.Join(tempDir, "error_zip.zip")
|
|
err = Zip(folder, outFolderZip)
|
|
if err == nil {
|
|
t.Fatal("expected Zip() to fail due to invalid addFilesToZipTestWrapper()")
|
|
}
|
|
}
|
|
|
|
func addFilesToZipTestWrapper(_ *zip.Writer, _ string, _ bool) error {
|
|
return errors.New("error")
|
|
}
|