Merge pull request #36 from mcuadros/temporal

helper: new temporal helper

Author: mcuadros

helper/temporal/temporal.go

@@ -0,0 +1,30 @@
+package temporal
+
+import (
+	"gopkg.in/src-d/go-billy.v3"
+	"gopkg.in/src-d/go-billy.v3/util"
+)
+
+// Temporal is a helper that implements billy.TempFile over any filesystem.
+type Temporal struct {
+	billy.Filesystem
+	defaultDir string
+}
+
+// New creates a new filesystem wrapping up 'fs' the intercepts the calls to
+// the TempFile method. The param defaultDir is used as default directory were
+// the tempfiles are created.
+func New(fs billy.Filesystem, defaultDir string) billy.Filesystem {
+	return &Temporal{
+		Filesystem: fs,
+		defaultDir: defaultDir,
+	}
+}
+
+func (h *Temporal) TempFile(dir, prefix string) (billy.File, error) {
+	if dir == "" {
+		dir = h.defaultDir
+	}
+
+	return util.TempFile(h.Filesystem, dir, prefix)
+}

helper/temporal/temporal_test.go

@@ -0,0 +1,32 @@
+package temporal
+
+import (
+	"strings"
+	"testing"
+
+	"gopkg.in/src-d/go-billy.v3/memfs"
+	"gopkg.in/src-d/go-billy.v3/test"
+
+	. "gopkg.in/check.v1"
+)
+
+func Test(t *testing.T) { TestingT(t) }
+
+var _ = Suite(&TemporalSuite{})
+
+type TemporalSuite struct {
+	test.TempFileSuite
+}
+
+func (s *TemporalSuite) SetUpTest(c *C) {
+	s.FS = New(memfs.New(), "foo")
+}
+
+func (s *TemporalSuite) TestTempFileDefaultPath(c *C) {
+	fs := New(memfs.New(), "foo")
+	f, err := fs.TempFile("", "bar")
+	c.Assert(err, IsNil)
+	c.Assert(f.Close(), IsNil)
+
+	c.Assert(strings.HasPrefix(f.Name(), fs.Join("foo", "bar")), Equals, true)
+}

test/tempfile.go

@@ -20,7 +20,8 @@ func (s *TempFileSuite) TestTempFile(c *C) {
 	f, err := s.FS.TempFile("", "bar")
 	c.Assert(err, IsNil)
 	c.Assert(f.Close(), IsNil)
-	c.Assert(strings.HasPrefix(f.Name(), "bar"), Equals, true)
+
+	c.Assert(strings.Index(f.Name(), "bar"), Not(Equals), -1)
 }
 
 func (s *TempFileSuite) TestTempFileWithPath(c *C) {
@@ -36,7 +37,7 @@ func (s *TempFileSuite) TestTempFileFullWithPath(c *C) {
 	c.Assert(err, IsNil)
 	c.Assert(f.Close(), IsNil)
 
-	c.Assert(strings.HasPrefix(f.Name(), s.FS.Join("foo", "bar")), Equals, true)
+	c.Assert(strings.Index(f.Name(), s.FS.Join("foo", "bar")), Not(Equals), -1)
 }
 
 func (s *TempFileSuite) TestRemoveTempFile(c *C) {