HgGetter: GetFile

get_hg.go

@@ -2,9 +2,11 @@ package getter
 
 import (
 	"fmt"
+	"io/ioutil"
 	"net/url"
 	"os"
 	"os/exec"
+	"path/filepath"
 	"runtime"
 
 	urlhelper "github.com/hashicorp/terraform/helper/url"
@@ -55,6 +57,37 @@ func (g *HgGetter) Get(dst string, u *url.URL) error {
 	return g.update(dst, newURL, rev)
 }
 
+// GetFile for Hg doesn't support updating at this time. It will download
+// the file every time.
+func (g *HgGetter) GetFile(dst string, u *url.URL) error {
+	td, err := ioutil.TempDir("", "getter-hg")
+	if err != nil {
+		return err
+	}
+	if err := os.RemoveAll(td); err != nil {
+		return err
+	}
+
+	// Get the filename, and strip the filename from the URL so we can
+	// just get the repository directly.
+	filename := filepath.Base(u.Path)
+	u.Path = filepath.Dir(u.Path)
+
+	// Get the full repository
+	if err := g.Get(td, u); err != nil {
+		return err
+	}
+
+	// Copy the single file
+	u, err = urlhelper.Parse(fmtFileURL(filepath.Join(td, filename)))
+	if err != nil {
+		return err
+	}
+
+	fg := &FileGetter{Copy: true}
+	return fg.GetFile(dst, u)
+}
+
 func (g *HgGetter) clone(dst string, u *url.URL) error {
 	cmd := exec.Command("hg", "clone", "-U", u.String(), dst)
 	return getRunCommand(cmd)

get_hg_test.go

@@ -20,8 +20,6 @@ func TestHgGetter_impl(t *testing.T) {
 }
 
 func TestHgGetter(t *testing.T) {
-	t.Parallel()
-
 	if !testHasHg {
 		t.Log("hg not found, skipping")
 		t.Skip()
@@ -43,8 +41,6 @@ func TestHgGetter(t *testing.T) {
 }
 
 func TestHgGetter_branch(t *testing.T) {
-	t.Parallel()
-
 	if !testHasHg {
 		t.Log("hg not found, skipping")
 		t.Skip()
@@ -79,3 +75,24 @@ func TestHgGetter_branch(t *testing.T) {
 		t.Fatalf("err: %s", err)
 	}
 }
+
+func TestHgGetter_GetFile(t *testing.T) {
+	if !testHasHg {
+		t.Log("hg not found, skipping")
+		t.Skip()
+	}
+
+	g := new(HgGetter)
+	dst := tempFile(t)
+
+	// Download
+	if err := g.GetFile(dst, testModuleURL("basic-hg/foo.txt")); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	// Verify the main file exists
+	if _, err := os.Stat(dst); err != nil {
+		t.Fatalf("err: %s", err)
+	}
+	assertContents(t, dst, "Hello\n")
+}

test-fixtures/basic-hg/.hg/cache/branch2-served

@@ -1,3 +1,3 @@
-c65e998d747ffbb1fe3b1c067a50664bb3fb5da4 1
-dcaed7754d58264cb9a5916215a5442377307bd1 o default
+992604507bcd66370bf91a0c9d526ccd833412bf 2
+992604507bcd66370bf91a0c9d526ccd833412bf o default
 c65e998d747ffbb1fe3b1c067a50664bb3fb5da4 o test-branch

test-fixtures/basic-hg/.hg/cache/rbc-names-v1

@@ -0,0 +1 @@
+default

test-fixtures/basic-hg/.hg/last-message.txt

@@ -1,2 +1,2 @@
-Branch
+add file
 

test-fixtures/basic-hg/.hg/store/fncache

@@ -1,2 +1,3 @@
 data/main.tf.i
+data/foo.txt.i
 data/main_branch.tf.i

test-fixtures/basic-hg/.hg/store/undo.backup.fncache

@@ -0,0 +1,2 @@
+data/main.tf.i
+data/main_branch.tf.i

test-fixtures/basic-hg/.hg/undo.branch

@@ -1 +1 @@
-test-branch
+default

test-fixtures/basic-hg/.hg/undo.desc

@@ -1,2 +1,2 @@
-1
+2
 commit

test-fixtures/basic-hg/foo.txt

@@ -0,0 +1 @@
+Hello