From 6c68150e45eb0ad236b0c89695a0c172f96d53f1 Mon Sep 17 00:00:00 2001
From: Alexander Neumann <alexander@bumpern.de>
Date: Wed, 11 Feb 2015 20:01:01 +0100
Subject: [PATCH] Add streaming functions to mockBackend

---
 backend/generic_test.go | 23 +++++++++++++++++------
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/backend/generic_test.go b/backend/generic_test.go
index 2d6aff550..71d868693 100644
--- a/backend/generic_test.go
+++ b/backend/generic_test.go
@@ -2,6 +2,7 @@ package backend_test
 
 import (
 	"fmt"
+	"io"
 	"path/filepath"
 	"reflect"
 	"runtime"
@@ -47,12 +48,14 @@ func str2id(s string) backend.ID {
 }
 
 type mockBackend struct {
-	list   func(backend.Type) (backend.IDs, error)
-	get    func(backend.Type, backend.ID) ([]byte, error)
-	create func(backend.Type, []byte) (backend.ID, error)
-	test   func(backend.Type, backend.ID) (bool, error)
-	remove func(backend.Type, backend.ID) error
-	close  func() error
+	list       func(backend.Type) (backend.IDs, error)
+	get        func(backend.Type, backend.ID) ([]byte, error)
+	getReader  func(backend.Type, backend.ID) (io.ReadCloser, error)
+	create     func(backend.Type, []byte) (backend.ID, error)
+	createFrom func(backend.Type, io.Reader) (backend.ID, error)
+	test       func(backend.Type, backend.ID) (bool, error)
+	remove     func(backend.Type, backend.ID) error
+	close      func() error
 }
 
 func (m mockBackend) List(t backend.Type) (backend.IDs, error) {
@@ -63,10 +66,18 @@ func (m mockBackend) Get(t backend.Type, id backend.ID) ([]byte, error) {
 	return m.get(t, id)
 }
 
+func (m mockBackend) GetReader(t backend.Type, id backend.ID) (io.ReadCloser, error) {
+	return m.getReader(t, id)
+}
+
 func (m mockBackend) Create(t backend.Type, data []byte) (backend.ID, error) {
 	return m.create(t, data)
 }
 
+func (m mockBackend) CreateFrom(t backend.Type, r io.Reader) (backend.ID, error) {
+	return m.createFrom(t, r)
+}
+
 func (m mockBackend) Test(t backend.Type, id backend.ID) (bool, error) {
 	return m.test(t, id)
 }