Merge pull request #23 from qor/fix-same

Fix same

filesystem/filesystem.go

@@ -1,6 +1,7 @@
 package filesystem
 
 import (
+	"bytes"
 	"fmt"
 	"io"
 	"os"
@@ -54,13 +55,18 @@ func (fileSystem FileSystem) Put(path string, reader io.Reader) (*oss.Object, er
 		return nil, err
 	}
 
+	if seeker, ok := reader.(io.ReadSeeker); ok {
+		seeker.Seek(0, 0)
+	}
+	buf := bytes.NewBuffer([]byte{})
+	if _, err = io.Copy(buf, reader); err != nil {
+		return nil, err
+	}
+
 	dst, err := os.Create(fullpath)
 
 	if err == nil {
-		if seeker, ok := reader.(io.ReadSeeker); ok {
-			seeker.Seek(0, 0)
-		}
-		_, err = io.Copy(dst, reader)
+		_, err = io.Copy(dst, buf)
 	}
 
 	return &oss.Object{Path: path, Name: filepath.Base(path), StorageInterface: fileSystem}, err

tests/tests.go

@@ -33,6 +33,17 @@ func TestAll(storage oss.StorageInterface, t *testing.T) {
 		t.Errorf("No error should happen when opem sample file, but got %v", err)
 	}
 
+	// Put file again
+	if file, err := storage.GetStream(fileName); err == nil {
+		if object, err := storage.Put(fileName, file); err != nil {
+			t.Errorf("No error should happen when save sample file, but got %v", err)
+		} else if object.Path == "" || object.StorageInterface == nil {
+			t.Errorf("returned object should necessary information")
+		}
+	} else {
+		t.Errorf("No error should happen when opem sample file, but got %v", err)
+	}
+
 	if file, err := os.Open(sampleFile); err == nil {
 		if object, err := storage.Put(fileName2, file); err != nil {
 			t.Errorf("No error should happen when save sample file, but got %v", err)