1 files changed, 164 insertions, 0 deletions
diff --git a/submodule_test.go b/submodule_test.go
new file mode 100644
index 0000000..a933965
--- /dev/null
+++ b/submodule_test.go
@@ -0,0 +1,164 @@
+package git
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+
+	"github.com/src-d/go-git-fixtures"
+
+	. "gopkg.in/check.v1"
+	"srcd.works/go-git.v4/plumbing"
+)
+
+type SubmoduleSuite struct {
+	BaseSuite
+	Worktree *Worktree
+	path     string
+}
+
+var _ = Suite(&SubmoduleSuite{})
+
+func (s *SubmoduleSuite) SetUpTest(c *C) {
+	path := fixtures.ByTag("submodule").One().Worktree().Base()
+
+	dir, err := ioutil.TempDir("", "submodule")
+	c.Assert(err, IsNil)
+
+	r, err := PlainClone(dir, false, &CloneOptions{
+		URL: fmt.Sprintf("file://%s", filepath.Join(path)),
+	})
+
+	c.Assert(err, IsNil)
+
+	s.Repository = r
+	s.Worktree, err = r.Worktree()
+	c.Assert(err, IsNil)
+
+	s.path = dir
+}
+
+func (s *SubmoduleSuite) TearDownTest(c *C) {
+	err := os.RemoveAll(s.path)
+	c.Assert(err, IsNil)
+}
+
+func (s *SubmoduleSuite) TestInit(c *C) {
+	sm, err := s.Worktree.Submodule("basic")
+	c.Assert(err, IsNil)
+
+	err = sm.Init()
+	c.Assert(err, IsNil)
+
+	cfg, err := s.Repository.Config()
+	c.Assert(err, IsNil)
+
+	c.Assert(cfg.Submodules, HasLen, 1)
+	c.Assert(cfg.Submodules["basic"], NotNil)
+}
+
+func (s *SubmoduleSuite) TestUpdate(c *C) {
+	sm, err := s.Worktree.Submodule("basic")
+	c.Assert(err, IsNil)
+
+	err = sm.Update(&SubmoduleUpdateOptions{
+		Init: true,
+	})
+
+	c.Assert(err, IsNil)
+
+	r, err := sm.Repository()
+	c.Assert(err, IsNil)
+
+	ref, err := r.Reference(plumbing.HEAD, true)
+	c.Assert(err, IsNil)
+	c.Assert(ref.Hash().String(), Equals, "6ecf0ef2c2dffb796033e5a02219af86ec6584e5")
+
+}
+
+func (s *SubmoduleSuite) TestUpdateWithoutInit(c *C) {
+	sm, err := s.Worktree.Submodule("basic")
+	c.Assert(err, IsNil)
+
+	err = sm.Update(&SubmoduleUpdateOptions{})
+	c.Assert(err, Equals, ErrSubmoduleNotInitialized)
+}
+
+func (s *SubmoduleSuite) TestUpdateWithNotFetch(c *C) {
+	sm, err := s.Worktree.Submodule("basic")
+	c.Assert(err, IsNil)
+
+	err = sm.Update(&SubmoduleUpdateOptions{
+		Init:    true,
+		NoFetch: true,
+	})
+
+	// Since we are not fetching, the object is not there
+	c.Assert(err, Equals, plumbing.ErrObjectNotFound)
+}
+
+func (s *SubmoduleSuite) TestUpdateWithRecursion(c *C) {
+	sm, err := s.Worktree.Submodule("itself")
+	c.Assert(err, IsNil)
+
+	err = sm.Update(&SubmoduleUpdateOptions{
+		Init:              true,
+		RecurseSubmodules: 2,
+	})
+
+	c.Assert(err, IsNil)
+
+	_, err = s.Worktree.fs.Stat("itself/basic/LICENSE")
+	c.Assert(err, IsNil)
+}
+
+func (s *SubmoduleSuite) TestUpdateWithInitAndUpdate(c *C) {
+	sm, err := s.Worktree.Submodule("basic")
+	c.Assert(err, IsNil)
+
+	err = sm.Update(&SubmoduleUpdateOptions{
+		Init: true,
+	})
+	c.Assert(err, IsNil)
+
+	idx, err := s.Repository.Storer.Index()
+	c.Assert(err, IsNil)
+
+	for i, e := range idx.Entries {
+		if e.Name == "basic" {
+			e.Hash = plumbing.NewHash("b029517f6300c2da0f4b651b8642506cd6aaf45d")
+		}
+
+		idx.Entries[i] = e
+	}
+
+	err = s.Repository.Storer.SetIndex(idx)
+	c.Assert(err, IsNil)
+
+	err = sm.Update(&SubmoduleUpdateOptions{})
+	c.Assert(err, IsNil)
+
+	r, err := sm.Repository()
+	c.Assert(err, IsNil)
+
+	ref, err := r.Reference(plumbing.HEAD, true)
+	c.Assert(err, IsNil)
+	c.Assert(ref.Hash().String(), Equals, "b029517f6300c2da0f4b651b8642506cd6aaf45d")
+
+}
+
+func (s *SubmoduleSuite) TestSubmodulesInit(c *C) {
+	sm, err := s.Worktree.Submodules()
+	c.Assert(err, IsNil)
+
+	err = sm.Init()
+	c.Assert(err, IsNil)
+
+	sm, err = s.Worktree.Submodules()
+	c.Assert(err, IsNil)
+
+	for _, m := range sm {
+		c.Assert(m.initialized, Equals, true)
+	}
+}