Merge pull request #114 from guilhermef/master

More getTree tests

repository/mocks.go

@@ -48,7 +48,7 @@ func (r *MockContentRetriever) GetArchive(repo, ref string, format ArchiveFormat
 	return r.ResultContents, nil
 }
 
-func CreateTestRepository(tmp_path string, repo string, file string, content string) (func(), error) {
+func CreateTestRepository(tmp_path string, repo string, file string, content string, folders ...string) (func(), error) {
 	testPath := path.Join(tmp_path, repo+".git")
 	cleanup := func() {
 		os.RemoveAll(testPath)
@@ -71,7 +71,18 @@ func CreateTestRepository(tmp_path string, repo string, file string, content str
 	if err != nil {
 		return cleanup, err
 	}
-	cmd = exec.Command(gitPath, "add", file)
+	for _, folder := range folders {
+		folderPath := path.Join(testPath, folder)
+		err = os.MkdirAll(folderPath, 0777)
+		if err != nil {
+			return cleanup, err
+		}
+		err = ioutil.WriteFile(path.Join(folderPath, file), []byte(content), 0644)
+		if err != nil {
+			return cleanup, err
+		}
+	}
+	cmd = exec.Command(gitPath, "add", ".")
 	cmd.Dir = testPath
 	err = cmd.Run()
 	if err != nil {

repository/repository_test.go

@@ -546,7 +546,7 @@ func (s *S) TestGetFileContentWhenRefIsInvalid(c *gocheck.C) {
 	}()
 	c.Assert(errCreate, gocheck.IsNil)
 	_, err := GetFileContents(repo, "MuchMissing", file)
-	c.Assert(err.Error(), gocheck.Equals, "Error when trying to obtain file README on ref MuchMissing of repository gandalf-test-repo (exit status 128).")
+	c.Assert(err, gocheck.ErrorMatches, "^Error when trying to obtain file README on ref MuchMissing of repository gandalf-test-repo \\(exit status 128\\)\\.$")
 }
 
 func (s *S) TestGetFileContentWhenFileIsInvalid(c *gocheck.C) {
@@ -562,5 +562,59 @@ func (s *S) TestGetFileContentWhenFileIsInvalid(c *gocheck.C) {
 	}()
 	c.Assert(errCreate, gocheck.IsNil)
 	_, err := GetFileContents(repo, "master", "Such file")
-	c.Assert(err.Error(), gocheck.Equals, "Error when trying to obtain file Such file on ref master of repository gandalf-test-repo (exit status 128).")
+	c.Assert(err, gocheck.ErrorMatches, "^Error when trying to obtain file Such file on ref master of repository gandalf-test-repo \\(exit status 128\\)\\.$")
+}
+
+func (s *S) TestGetTreeIntegration(c *gocheck.C) {
+	oldBare := bare
+	bare = "/tmp"
+	repo := "gandalf-test-repo"
+	file := "README"
+	content := "much WOW"
+	cleanUp, errCreate := CreateTestRepository(bare, repo, file, content, []string{"such", "folder", "much", "magic"}...)
+	defer func() {
+		cleanUp()
+		bare = oldBare
+	}()
+	c.Assert(errCreate, gocheck.IsNil)
+	tree, err := GetTree(repo, "master", "much/README")
+	c.Assert(err, gocheck.IsNil)
+	c.Assert(tree[0]["path"], gocheck.Equals, "much/README")
+	c.Assert(tree[0]["rawPath"], gocheck.Equals, "much/README")
+}
+
+func (s *S) TestGetTreeIntegrationWithEscapedFileName(c *gocheck.C) {
+	oldBare := bare
+	bare = "/tmp"
+	repo := "gandalf-test-repo"
+	file := "such\tREADME"
+	content := "much WOW"
+	cleanUp, errCreate := CreateTestRepository(bare, repo, file, content, []string{"such", "folder", "much", "magic"}...)
+	defer func() {
+		cleanUp()
+		bare = oldBare
+	}()
+	c.Assert(errCreate, gocheck.IsNil)
+	tree, err := GetTree(repo, "master", "much/such\tREADME")
+	c.Assert(err, gocheck.IsNil)
+	c.Assert(tree[0]["path"], gocheck.Equals, "much/such\\tREADME")
+	c.Assert(tree[0]["rawPath"], gocheck.Equals, "\"much/such\\tREADME\"")
+}
+
+func (s *S) TestGetTreeIntegrationWithFileNameWithSpace(c *gocheck.C) {
+	oldBare := bare
+	bare = "/tmp"
+	repo := "gandalf-test-repo"
+	file := "much README"
+	content := "much WOW"
+	cleanUp, errCreate := CreateTestRepository(bare, repo, file, content, []string{"such", "folder", "much", "magic"}...)
+	defer func() {
+		cleanUp()
+		bare = oldBare
+	}()
+	c.Assert(errCreate, gocheck.IsNil)
+	tree, err := GetTree(repo, "master", "much/much README")
+	c.Assert(err, gocheck.IsNil)
+	c.Assert(tree[0]["path"], gocheck.Equals, "much/much README")
+	c.Assert(tree[0]["rawPath"], gocheck.Equals, "much/much README")
 }