mirror of
https://github.com/nabbar/golib.git
synced 2025-09-26 20:01:15 +08:00
71c18b96a5
Package Archive: - complete rework of package - split package between compression (stream compression only) and archive (catalogue/dictionary/tape archive type) - rework compression to be stream only working without any temporary file or any like - rework archive like TAR, ZIP... to expose stream instead of writing directly file/path - rework archive like TAR, ZIP... to allow adding file from stream only and allowing to parsing local path to add it into archive - rework detection of compression / archive - rework extractAll function to not using useless local temporary file - adding test can be used as example and perform testing of most of package code - using const custom type for compression and archive allow them to be parsed, marshalled or unmarshalled from text or json and more - apply following change into other module that use it
119 lines
3.1 KiB
Go
119 lines
3.1 KiB
Go
/*
|
|
* MIT License
|
|
*
|
|
* Copyright (c) 2020 Nicolas JUHEL
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*
|
|
*/
|
|
|
|
package archive_test
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"os"
|
|
|
|
libarc "github.com/nabbar/golib/archive"
|
|
arccmp "github.com/nabbar/golib/archive/compress"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("archive/compress/bzip", func() {
|
|
Context("Write/Read a bzip compressed file", func() {
|
|
It("Create a bzip compressed file must succeed", func() {
|
|
var (
|
|
hdf *os.File
|
|
buf *bufio.Writer
|
|
wrt io.WriteCloser
|
|
)
|
|
|
|
defer func() {
|
|
if hdf != nil {
|
|
_ = hdf.Close()
|
|
}
|
|
}()
|
|
|
|
arc[arccmp.Bzip2.String()] = "lorem_ipsum" + arccmp.Bzip2.Extension()
|
|
|
|
hdf, err = os.Create(arc[arccmp.Bzip2.String()])
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(hdf).ToNot(BeNil())
|
|
|
|
wrt, err = arccmp.Bzip2.Writer(hdf)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(wrt).ToNot(BeNil())
|
|
|
|
buf = bufio.NewWriter(wrt)
|
|
_, err = buf.WriteString(loremIpsum)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = buf.Flush()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = hdf.Sync()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = wrt.Close()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = hdf.Close()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
})
|
|
|
|
It("Detect and Extract a bzip compressed file must succeed", func() {
|
|
var (
|
|
hdf *os.File
|
|
alg arccmp.Algorithm
|
|
rdr io.ReadCloser
|
|
buf *bufio.Reader
|
|
)
|
|
|
|
defer func() {
|
|
if hdf != nil {
|
|
_ = hdf.Close()
|
|
}
|
|
}()
|
|
|
|
hdf, err = os.Open(arc[arccmp.Bzip2.String()])
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(hdf).ToNot(BeNil())
|
|
|
|
alg, rdr, err = libarc.DetectCompression(hdf)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(rdr).ToNot(BeNil())
|
|
Expect(alg).To(Equal(arccmp.Bzip2))
|
|
|
|
buf = bufio.NewReader(rdr)
|
|
_, err = io.Copy(io.Discard, buf)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = buf.UnreadByte()
|
|
Expect(err).To(HaveOccurred())
|
|
|
|
err = rdr.Close()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = hdf.Close()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
})
|
|
})
|
|
})
|