From 2a8ea07bd8ce753e53aa51e9283dd1fd796b618a Mon Sep 17 00:00:00 2001
From: Christophe de Vienne <christophe.devienne@orus.io>
Date: Tue, 7 May 2024 13:39:46 +0200
Subject: [PATCH] hg-cache: add a e2e test

---
 test/e2e/hg_test.go | 135 +++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 132 insertions(+), 3 deletions(-)

diff --git a/test/e2e/hg_test.go b/test/e2e/hg_test.go
index 0340bd73..2d5b1bad 100644
--- a/test/e2e/hg_test.go
+++ b/test/e2e/hg_test.go
@@ -1,15 +1,50 @@
 package e2e
 
 import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
 	"os"
 	"os/exec"
+	"path"
+	"path/filepath"
 	"strings"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
+type HgAssetInfo struct {
+	InitialChangeset string `json:"initial-changeset"`
+	ExtraBundle      string `json:"extra-bundle"`
+	ExtraChangeset   string `json:"extra-changeset"`
+}
+
+type hgVendirOutput struct {
+	Tables string
+	Blocks []string
+	Lines  []string
+}
+
+func loadHgAssetInfo(t *testing.T, filename string) HgAssetInfo {
+	t.Helper()
+	f, err := os.Open(filename)
+	if err != nil {
+		t.Fatalf("could not open file %s: %s", filename, err)
+	}
+	var info HgAssetInfo
+	if err := json.NewDecoder(f).Decode(&info); err != nil {
+		t.Fatalf("could not parse file %s: %s", filename, err)
+	}
+	return info
+}
+
 func TestHgCache(t *testing.T) {
 	env := BuildEnv(t)
 	logger := Logger{}
+	vendir := Vendir{t, env.BinaryPath, logger}
 
 	_ = Vendir{t, env.BinaryPath, logger}
 
@@ -24,7 +59,10 @@ func TestHgCache(t *testing.T) {
 		t.Fatalf("Unpacking hg-repos asset: %s (output: '%s')", err, out)
 	}
 
-	yamlConfig := strings.NewReader(`
+	info := loadHgAssetInfo(t, path.Join(hgSrcPath, "info.json"))
+
+	yamlConfig := func(ref string) io.Reader {
+		return strings.NewReader(fmt.Sprintf(`
 apiVersion: vendir.k14s.io/v1alpha1
 kind: Config
 directories:
@@ -32,7 +70,98 @@ directories:
   contents:
   - path: test
     hg:
-      url: "%s"
+      url: "%s/repo"
       ref: "%s"
-	`)
+`, hgSrcPath, ref))
+	}
+
+	cachePath, err := os.MkdirTemp("", "vendir-e2e-hg-cache-vendir-cache")
+	require.NoError(t, err)
+	defer os.RemoveAll(cachePath)
+
+	dstPath, err := os.MkdirTemp("", "vendir-e2e-hg-cache-dst")
+	require.NoError(t, err)
+	defer os.RemoveAll(dstPath)
+
+	var stdout bytes.Buffer
+	stdoutDec := json.NewDecoder(&stdout)
+	//defer os.RemoveAll(dstPath)
+	logger.Section("initial clone", func() {
+		stdout.Truncate(0)
+		vendir.RunWithOpts(
+			[]string{"sync", "-f", "-", "--json"},
+			RunOpts{
+				Dir:          dstPath,
+				StdinReader:  yamlConfig(info.InitialChangeset),
+				StdoutWriter: &stdout,
+				Env:          []string{"VENDIR_CACHE_DIR=" + cachePath},
+			})
+
+		var out hgVendirOutput
+		require.NoError(t, stdoutDec.Decode(&out))
+
+		assert.Contains(t, out.Lines[1], "init")
+		assert.Contains(t, out.Lines[3], "pull")
+
+		b, err := os.ReadFile(filepath.Join(dstPath, "vendor", "test", "file1.txt"))
+		assert.NoError(t, err)
+		assert.Equal(t, "content1\n", string(b))
+	})
+
+	logger.Section("sync from cache only", func() {
+		stdout.Truncate(0)
+		vendir.RunWithOpts(
+			[]string{"sync", "-f", "-", "--json"},
+			RunOpts{
+				Dir:          dstPath,
+				StdinReader:  yamlConfig(info.InitialChangeset),
+				StdoutWriter: &stdout,
+				Env:          []string{"VENDIR_CACHE_DIR=" + cachePath},
+			})
+
+		var out hgVendirOutput
+		require.NoError(t, stdoutDec.Decode(&out))
+
+		for _, line := range out.Lines {
+			assert.NotContains(t, line, "init")
+			assert.NotContains(t, line, "pull")
+		}
+
+		b, err := os.ReadFile(filepath.Join(dstPath, "vendor", "test", "file1.txt"))
+		assert.NoError(t, err)
+		assert.Equal(t, "content1\n", string(b))
+	})
+
+	out, err = exec.Command(
+		"hg", "unbundle",
+		"--repository", filepath.Join(hgSrcPath, "/repo"),
+		filepath.Join(hgSrcPath, info.ExtraBundle),
+	).CombinedOutput()
+	if err != nil {
+		t.Fatalf("Unpacking hg-repos asset: %s (output: '%s')", err, out)
+	}
+
+	logger.Section("sync from cache + pull", func() {
+		stdout.Truncate(0)
+		vendir.RunWithOpts(
+			[]string{"sync", "-f", "-", "--json"},
+			RunOpts{
+				Dir:          dstPath,
+				StdinReader:  yamlConfig(info.ExtraChangeset),
+				StdoutWriter: &stdout,
+				Env:          []string{"VENDIR_CACHE_DIR=" + cachePath},
+			})
+
+		var out hgVendirOutput
+		require.NoError(t, stdoutDec.Decode(&out))
+
+		assert.Contains(t, out.Lines[3], "pull")
+		for _, line := range out.Lines {
+			assert.NotContains(t, line, "init")
+		}
+
+		b, err := os.ReadFile(filepath.Join(dstPath, "vendor", "test", "file1.txt"))
+		assert.NoError(t, err)
+		assert.Equal(t, "content2\n", string(b))
+	})
 }