mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 15:09:42 +00:00
68a6f5828f
* Binance: Fix subscription failures ignored * Testing: Fix race on shared config singleton * Config: Privatise Global config var We should *either* use a private var *or* use an accessor, but it doesn't make sense to mix paradigms. Since GetConfig() is well established this instead removes the limited uses of direct public access and adds a Setter * Zip: Fix test failure on http mocks
76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
package archive
|
|
|
|
import (
|
|
"archive/zip"
|
|
"errors"
|
|
"io/fs"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUnZip(t *testing.T) {
|
|
tempDir := t.TempDir()
|
|
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) {
|
|
tempDir := t.TempDir()
|
|
outFile := filepath.Join(tempDir, "out.zip")
|
|
err := Zip(filepath.Join("..", "..", "..", "testdata", "configtest.json"), outFile)
|
|
require.NoError(t, err, "Zip should not error")
|
|
o, err := UnZip(outFile, tempDir)
|
|
require.NoError(t, err, "UnZip should not error")
|
|
assert.Len(t, o, 1, "Should extract 1 file")
|
|
|
|
folder := filepath.Join("..", "..", "..", "testdata", "gctscript")
|
|
outFolderZip := filepath.Join(tempDir, "out_folder.zip")
|
|
err = Zip(folder, outFolderZip)
|
|
require.NoError(t, err, "Zip should not error")
|
|
o, err = UnZip(outFolderZip, tempDir)
|
|
require.NoError(t, err, "UnZip should not error")
|
|
var found bool
|
|
for i := range o {
|
|
if filepath.Base(o[i]) == "timer.gct" {
|
|
found = true
|
|
}
|
|
}
|
|
assert.True(t, found, "Should find a gctscript in the zip")
|
|
assert.GreaterOrEqual(t, len(o), 6, "Should extract at least 6 files")
|
|
|
|
folder = filepath.Join("..", "..", "..", "testdata", "invalid_file.json")
|
|
err = Zip(folder, filepath.Join(tempDir, "invalid.zip"))
|
|
assert.ErrorIs(t, err, fs.ErrNotExist, "Zip should error correctly")
|
|
|
|
addFilesToZip = addFilesToZipTestWrapper
|
|
folder = filepath.Join("..", "..", "..", "testdata", "http_mock")
|
|
outFolderZip = filepath.Join(tempDir, "error_zip.zip")
|
|
err = Zip(folder, outFolderZip)
|
|
assert.ErrorContains(t, err, "specific error", "Zip should error correctly")
|
|
}
|
|
|
|
func addFilesToZipTestWrapper(_ *zip.Writer, _ string, _ bool) error {
|
|
return errors.New("specific error")
|
|
}
|