1 files changed, 52 insertions, 0 deletions

diff --git a/vendor/github.com/hashicorp/go-getter/get_mock.go b/vendor/github.com/hashicorp/go-getter/get_mock.go
new file mode 100644
index 0000000..882e694
--- /dev/null
+++ b/vendor/github.com/hashicorp/go-getter/get_mock.go
@@ -0,0 +1,52 @@
+package getter
+
+import (
+        "net/url"
+)
+
+// MockGetter is an implementation of Getter that can be used for tests.
+type MockGetter struct {
+        // Proxy, if set, will be called after recording the calls below.
+        // If it isn't set, then the *Err values will be returned.
+        Proxy Getter
+
+        GetCalled bool
+        GetDst    string
+        GetURL    *url.URL
+        GetErr    error
+
+        GetFileCalled bool
+        GetFileDst    string
+        GetFileURL    *url.URL
+        GetFileErr    error
+}
+
+func (g *MockGetter) Get(dst string, u *url.URL) error {
+        g.GetCalled = true
+        g.GetDst = dst
+        g.GetURL = u
+
+        if g.Proxy != nil {
+                return g.Proxy.Get(dst, u)
+        }
+
+        return g.GetErr
+}
+
+func (g *MockGetter) GetFile(dst string, u *url.URL) error {
+        g.GetFileCalled = true
+        g.GetFileDst = dst
+        g.GetFileURL = u
+
+        if g.Proxy != nil {
+                return g.Proxy.GetFile(dst, u)
+        }
+        return g.GetFileErr
+}
+
+func (g *MockGetter) ClientMode(u *url.URL) (ClientMode, error) {
+        if l := len(u.Path); l > 0 && u.Path[l-1:] == "/" {
+                return ClientModeDir, nil
+        }
+        return ClientModeFile, nil
+}