source: add some test cases for File

helpers/general.go

@@ -75,6 +75,9 @@ func GuessType(in string) string {
 // ReaderToBytes takes an io.Reader argument, reads from it
 // and returns bytes.
 func ReaderToBytes(lines io.Reader) []byte {
+	if lines == nil {
+		return []byte{}
+	}
 	b := bp.GetBuffer()
 	defer bp.PutBuffer(b)
 
@@ -87,6 +90,9 @@ func ReaderToBytes(lines io.Reader) []byte {
 
 // ReaderToString is the same as ReaderToBytes, but returns a string.
 func ReaderToString(lines io.Reader) string {
+	if lines == nil {
+		return ""
+	}
 	b := bp.GetBuffer()
 	defer bp.PutBuffer(b)
 	b.ReadFrom(lines)

source/file_test.go

@@ -0,0 +1,25 @@
+package source
+
+import (
+	"github.com/spf13/hugo/helpers"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestFileUniqueID(t *testing.T) {
+	f1 := File{uniqueID: "123"}
+	f2 := NewFile("a")
+
+	assert.Equal(t, "123", f1.UniqueID())
+	assert.Equal(t, "0cc175b9c0f1b6a831c399e269772661", f2.UniqueID())
+}
+
+func TestFileString(t *testing.T) {
+	assert.Equal(t, "abc", NewFileWithContents("a", helpers.StringToReader("abc")).String())
+	assert.Equal(t, "", NewFile("a").String())
+}
+
+func TestFileBytes(t *testing.T) {
+	assert.Equal(t, []byte("abc"), NewFileWithContents("a", helpers.StringToReader("abc")).Bytes())
+	assert.Equal(t, []byte(""), NewFile("a").Bytes())
+}