mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
35 lines
785 B
Go
35 lines
785 B
Go
package tempfile
|
|
|
|
import (
|
|
"github.com/docker/docker/pkg/testutil/assert"
|
|
"io/ioutil"
|
|
"os"
|
|
)
|
|
|
|
// TempFile is a temporary file that can be used with unit tests. TempFile
|
|
// reduces the boilerplate setup required in each test case by handling
|
|
// setup errors.
|
|
type TempFile struct {
|
|
File *os.File
|
|
}
|
|
|
|
// NewTempFile returns a new temp file with contents
|
|
func NewTempFile(t assert.TestingT, prefix string, content string) *TempFile {
|
|
file, err := ioutil.TempFile("", prefix+"-")
|
|
assert.NilError(t, err)
|
|
|
|
_, err = file.Write([]byte(content))
|
|
assert.NilError(t, err)
|
|
file.Close()
|
|
return &TempFile{File: file}
|
|
}
|
|
|
|
// Name returns the filename
|
|
func (f *TempFile) Name() string {
|
|
return f.File.Name()
|
|
}
|
|
|
|
// Remove removes the file
|
|
func (f *TempFile) Remove() {
|
|
os.Remove(f.Name())
|
|
}
|