From 0ff9ef2b44c53e557c78bde0fd9c29847e5f0e23 Mon Sep 17 00:00:00 2001
From: Máximo Cuadros <mcuadros@gmail.com>
Date: Mon, 7 Nov 2016 20:29:58 +0100
Subject: global storage interface refactor  (#112)

* core: ObjectStorage, ReferenceStorage renamed to ObjectStorer and
ReferenceStorer

* rebase

* general, changes request by @alcortes

* general, changes request by @alcortes
---
 remote_test.go | 28 +++++++++++++++++++++-------
 1 file changed, 21 insertions(+), 7 deletions(-)

(limited to 'remote_test.go')

diff --git a/remote_test.go b/remote_test.go
index 37539af..04167cd 100644
--- a/remote_test.go
+++ b/remote_test.go
@@ -1,6 +1,7 @@
 package git
 
 import (
+	"io"
 	"io/ioutil"
 	"os"
 
@@ -80,7 +81,7 @@ func (s *RemoteSuite) TestFetch(c *C) {
 	})
 
 	c.Assert(err, IsNil)
-	c.Assert(sto.ObjectStorage().(*memory.ObjectStorage).Objects, HasLen, 31)
+	c.Assert(sto.Objects, HasLen, 31)
 
 	expectedRefs := []*core.Reference{
 		core.NewReferenceFromStrings("refs/remotes/origin/master", "6ecf0ef2c2dffb796033e5a02219af86ec6584e5"),
@@ -89,22 +90,33 @@ func (s *RemoteSuite) TestFetch(c *C) {
 	}
 
 	for _, exp := range expectedRefs {
-		r, _ := sto.ReferenceStorage().Get(exp.Name())
+		r, _ := sto.Reference(exp.Name())
 		c.Assert(exp.String(), Equals, r.String())
 	}
 }
 
-func (s *RemoteSuite) TestFetchObjectStorageWriter(c *C) {
+type mockPackfileWriter struct {
+	Storer
+	PackfileWriterCalled bool
+}
+
+func (m *mockPackfileWriter) PackfileWriter() (io.WriteCloser, error) {
+	m.PackfileWriterCalled = true
+	return m.Storer.(core.PackfileWriter).PackfileWriter()
+}
+
+func (s *RemoteSuite) TestFetchWithPackfileWriter(c *C) {
 	dir, err := ioutil.TempDir("", "fetch")
 	c.Assert(err, IsNil)
 
 	defer os.RemoveAll(dir) // clean up
 
-	var sto Storage
-	sto, err = filesystem.NewStorage(osfs.New(dir))
+	fss, err := filesystem.NewStorage(osfs.New(dir))
 	c.Assert(err, IsNil)
 
-	r := newRemote(sto, &config.RemoteConfig{Name: "foo", URL: RepositoryFixture})
+	mock := &mockPackfileWriter{Storer: fss}
+
+	r := newRemote(mock, &config.RemoteConfig{Name: "foo", URL: RepositoryFixture})
 	r.upSrv = &MockGitUploadPackService{}
 
 	c.Assert(r.Connect(), IsNil)
@@ -116,14 +128,16 @@ func (s *RemoteSuite) TestFetchObjectStorageWriter(c *C) {
 	c.Assert(err, IsNil)
 
 	var count int
-	iter, err := sto.ObjectStorage().Iter(core.AnyObject)
+	iter, err := mock.IterObjects(core.AnyObject)
 	c.Assert(err, IsNil)
 
 	iter.ForEach(func(core.Object) error {
 		count++
 		return nil
 	})
+
 	c.Assert(count, Equals, 31)
+	c.Assert(mock.PackfileWriterCalled, Equals, true)
 }
 
 func (s *RemoteSuite) TestFetchNoErrAlreadyUpToDate(c *C) {
-- 
cgit