From 57bcc65f28e1162532f222e7837f10646cab24bf Mon Sep 17 00:00:00 2001
From: Jordan Krage <jmank88@gmail.com>
Date: Sat, 23 Sep 2017 08:22:16 -0500
Subject: [PATCH] gps: vcs: dedupe git version list

---
 internal/gps/vcs_source.go      | 13 +++++--
 internal/gps/vcs_source_test.go | 67 +++++++++++++++++++++++++++++++++
 2 files changed, 76 insertions(+), 4 deletions(-)

diff --git a/internal/gps/vcs_source.go b/internal/gps/vcs_source.go
index d71941ee1c..6fc9054660 100644
--- a/internal/gps/vcs_source.go
+++ b/internal/gps/vcs_source.go
@@ -215,7 +215,7 @@ func (s *gitSource) listVersions(ctx context.Context) (vlist []PairedVersion, er
 	var headrev Revision
 	var onedef, multidef, defmaster bool
 
-	smap := make(map[string]bool)
+	smap := make(map[string]int)
 	uniq := 0
 	vlist = make([]PairedVersion, len(all))
 	for _, pair := range all {
@@ -250,7 +250,12 @@ func (s *gitSource) listVersions(ctx context.Context) (vlist []PairedVersion, er
 				// If the suffix is there, then we *know* this is the rev of
 				// the underlying commit object that we actually want
 				vstr = strings.TrimSuffix(vstr, "^{}")
-			} else if smap[vstr] {
+				if i, ok := smap[vstr]; ok {
+					v = NewVersion(vstr).Pair(Revision(pair[:40]))
+					vlist[i] = v
+					continue
+				}
+			} else if _, ok := smap[vstr]; ok {
 				// Already saw the deref'd version of this tag, if one
 				// exists, so skip this.
 				continue
@@ -258,8 +263,8 @@ func (s *gitSource) listVersions(ctx context.Context) (vlist []PairedVersion, er
 				// version first. Which should be impossible, but this
 				// covers us in case of weirdness, anyway.
 			}
-			v = NewVersion(vstr).Pair(Revision(pair[:40])).(PairedVersion)
-			smap[vstr] = true
+			v = NewVersion(vstr).Pair(Revision(pair[:40]))
+			smap[vstr] = uniq
 			vlist[uniq] = v
 			uniq++
 		}
diff --git a/internal/gps/vcs_source_test.go b/internal/gps/vcs_source_test.go
index d797f970b8..d48e34412d 100644
--- a/internal/gps/vcs_source_test.go
+++ b/internal/gps/vcs_source_test.go
@@ -580,6 +580,73 @@ func TestGitSourceListVersionsNoHEAD(t *testing.T) {
 	}
 }
 
+func TestGitSourceListVersionsNoDupes(t *testing.T) {
+	t.Parallel()
+
+	// This test is slowish, skip it on -short
+	if testing.Short() {
+		t.Skip("Skipping git source version fetching test in short mode")
+	}
+	requiresBins(t, "git")
+
+	cpath, err := ioutil.TempDir("", "smcache")
+	if err != nil {
+		t.Errorf("Failed to create temp dir: %s", err)
+	}
+	defer func() {
+		if err := os.RemoveAll(cpath); err != nil {
+			t.Errorf("removeAll failed: %s", err)
+		}
+	}()
+
+	n := "github.com/carolynvs/deptest-importers"
+	un := "https://" + n
+	u, err := url.Parse(un)
+	if err != nil {
+		t.Fatalf("Error parsing URL %s: %s", un, err)
+	}
+	mb := maybeGitSource{
+		url: u,
+	}
+
+	ctx := context.Background()
+	superv := newSupervisor(ctx)
+	src, state, err := mb.try(ctx, cpath, newMemoryCache(), superv)
+	if err != nil {
+		t.Fatalf("Unexpected error while setting up gitSource for test repo: %s", err)
+	}
+
+	wantstate := sourceIsSetUp | sourceExistsUpstream | sourceHasLatestVersionList
+	if state != wantstate {
+		t.Errorf("Expected return state to be %v, got %v", wantstate, state)
+	}
+
+	err = src.initLocal(ctx)
+	if err != nil {
+		t.Fatalf("Error on cloning git repo: %s", err)
+	}
+
+	pvlist, err := src.listVersions(ctx)
+	if err != nil {
+		t.Fatalf("Unexpected error getting version pairs from git repo: %s", err)
+	}
+
+	for i := range pvlist {
+		pv1 := pvlist[i]
+		uv1 := pv1.Unpair()
+		for j := range pvlist {
+			if i == j {
+				continue
+			}
+			pv2 := pvlist[j]
+			uv2 := pv2.Unpair()
+			if uv1 == uv2 {
+				t.Errorf("duplicate version pair mapping from %#v to both %q and %q", uv1, pv1.Revision(), pv2.Revision())
+			}
+		}
+	}
+}
+
 func Test_bzrSource_exportRevisionTo_removeVcsFiles(t *testing.T) {
 	t.Parallel()