Merge pull request #48 from keybase/strib/gh-truncate

fs: add `Truncate` method to `File` interface

Author: mcuadros

fs.go

@@ -140,4 +140,6 @@ type File interface {
 	Lock() error
 	// Unlock unlocks the file.
 	Unlock() error
+	// Truncate the file.
+	Truncate(size int64) error
 }

memfs/memory.go

@@ -270,6 +270,16 @@ func (f *file) Close() error {
 	return nil
 }
 
+func (f *file) Truncate(size int64) error {
+	if size < int64(len(f.content.bytes)) {
+		f.content.bytes = f.content.bytes[:size]
+	} else if more := int(size)-len(f.content.bytes); more > 0 {
+		f.content.bytes = append(f.content.bytes, make([]byte, more)...)
+	}
+
+	return nil
+}
+
 func (f *file) Duplicate(filename string, mode os.FileMode, flag int) billy.File {
 	new := &file{
 		name:    filename,

test/basic.go

@@ -562,3 +562,22 @@ func (s *BasicSuite) TestWriteFile(c *C) {
 
 	c.Assert(f.Close(), IsNil)
 }
+
+func (s *BasicSuite) TestTruncate(c *C) {
+	f, err := s.FS.Create("foo")
+	c.Assert(err, IsNil)
+
+	for _, sz := range []int64{4, 7, 2, 30, 0, 1} {
+		err = f.Truncate(sz)
+		c.Assert(err, IsNil)
+
+		bs, err := ioutil.ReadAll(f)
+		c.Assert(err, IsNil)
+		c.Assert(len(bs), Equals, int(sz))
+
+		_, err = f.Seek(0, io.SeekStart)
+		c.Assert(err, IsNil)
+	}
+
+	c.Assert(f.Close(), IsNil)
+}

test/mock.go

@@ -130,3 +130,7 @@ func (*FileMock) Lock() error {
 func (*FileMock) Unlock() error {
 	return nil
 }
+
+func (*FileMock) Truncate(size int64) error {
+	return nil
+}