Merge pull request #109 from netlify/not-save-in-hashed

Save sha without original sha when to save in Hashed

Author: bcomnes

go/porcelain/deploy.go

@@ -121,8 +121,13 @@ func newDeployFiles() *deployFiles {
 func (d *deployFiles) Add(p string, f *FileBundle) {
 	d.Files[p] = f
 	d.Sums[p] = f.Sum
-	list, _ := d.Hashed[f.Sum]
-	d.Hashed[f.Sum] = append(list, f)
+	// Remove ":original_sha" part when to save in Hashed (asset management)
+	sum := f.Sum
+	if strings.Contains(sum, ":") {
+		sum = strings.Split(sum, ":")[0]
+	}
+	list, _ := d.Hashed[sum]
+	d.Hashed[sum] = append(list, f)
 }
 
 func (n *Netlify) overCommitted(d *deployFiles) bool {

go/porcelain/deploy_test.go

@@ -24,3 +24,30 @@ func TestGetAssetManagementSha(t *testing.T) {
 		}
 	}
 }
+
+func TestAddWithAssetManagement(t *testing.T) {
+	files := newDeployFiles()
+	tests := []struct {
+		rel string
+		sum string
+	}{
+		{"foo.jpg", "sum1"},
+		{"bar.jpg", "sum2"},
+		{"baz.jpg", "sum3:originalsha"},
+	}
+
+	for _, test := range tests {
+		file := &FileBundle{}
+		file.Sum = test.sum
+		files.Add(test.rel, file)
+	}
+
+	out := files.Hashed["sum3"]
+	if len(out) != 1 {
+		t.Fatalf("expected `%d`, got `%d`", 1, len(out))
+	}
+	out2 := files.Sums["baz.jpg"]
+	if out2 != "sum3:originalsha" {
+		t.Fatalf("expected `%v`, got `%v`", "sum3:originalsha", out2)
+	}
+}