1
0
Fork 0

Remove unneeded files

master
Philip O'Toole 1 year ago
parent 99a6d3c0ba
commit bb3fdf9847

@ -1,28 +0,0 @@
package upload
import (
"os"
)
// AutoDeleteFile is a wrapper around os.File that deletes the file when it is
// closed.
type AutoDeleteFile struct {
*os.File
}
// Close implements the io.Closer interface
func (f *AutoDeleteFile) Close() error {
if err := f.File.Close(); err != nil {
return err
}
return os.Remove(f.Name())
}
// NewAutoDeleteFile takes a filename and wraps it in an AutoDeleteFile
func NewAutoDeleteFile(path string) (*AutoDeleteFile, error) {
f, err := os.Open(path)
if err != nil {
return nil, err
}
return &AutoDeleteFile{f}, nil
}

@ -1,57 +0,0 @@
package upload
import (
"os"
"testing"
)
func Test_NewAutoDeleteTempFile(t *testing.T) {
adFile, err := NewAutoDeleteFile(mustCreateTempFilename())
if err != nil {
t.Fatalf("NewAutoDeleteFile() failed: %v", err)
}
defer adFile.Close()
if _, err := os.Stat(adFile.Name()); os.IsNotExist(err) {
t.Fatalf("Expected file to exist: %s", adFile.Name())
}
}
func Test_AutoDeleteFile_Name(t *testing.T) {
name := mustCreateTempFilename()
adFile, err := NewAutoDeleteFile(name)
if err != nil {
t.Fatalf("NewAutoDeleteFile() failed: %v", err)
}
defer adFile.Close()
if adFile.Name() != name {
t.Fatalf("Expected Name() to return %s, got %s", name, adFile.Name())
}
}
func Test_AutoDeleteFile_Close(t *testing.T) {
adFile, err := NewAutoDeleteFile(mustCreateTempFilename())
if err != nil {
t.Fatalf("NewAutoDeleteFile() failed: %v", err)
}
filename := adFile.Name()
err = adFile.Close()
if err != nil {
t.Fatalf("Close() failed: %v", err)
}
if _, err := os.Stat(filename); !os.IsNotExist(err) {
t.Fatalf("Expected file to be deleted after Close(): %s", filename)
}
}
func mustCreateTempFilename() string {
f, err := os.CreateTemp("", "autodeletefile_test")
if err != nil {
panic(err)
}
f.Close()
return f.Name()
}
Loading…
Cancel
Save