Add lunar replace command

cmd/replace.go

@@ -0,0 +1,43 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/leonvogt/lunar/internal"
+	"github.com/spf13/cobra"
+)
+
+var (
+	replaceCmd = &cobra.Command{
+		Use:   "replace",
+		Short: "Replaces a snapshot",
+		Run: func(_ *cobra.Command, args []string) {
+			replaceSnapshot(args)
+		},
+	}
+)
+
+func replaceSnapshot(args []string) {
+	if !internal.DoesConfigExist() {
+		fmt.Println("There seems to be no configuration file. Please run 'lunar init' first.")
+		return
+	}
+
+	if len(args) != 1 {
+		fmt.Println("Please provide a snapshot name.")
+		return
+	}
+
+	snapshotName := args[0]
+	config, _ := internal.ReadConfig()
+	snapshotDatabaseName := internal.SnapshotDatabaseName(config.DatabaseName, snapshotName)
+
+	fmt.Println("Removing snapshot ", snapshotName)
+	internal.TerminateAllCurrentConnections(snapshotDatabaseName)
+	internal.DropDatabase(snapshotDatabaseName)
+
+	fmt.Println("Creating snapshot for database", config.DatabaseName, "with name", snapshotName)
+	internal.TerminateAllCurrentConnections(snapshotDatabaseName)
+	internal.TerminateAllCurrentConnections(config.DatabaseName)
+	internal.CreateSnapshot(config.DatabaseName, snapshotDatabaseName)
+}

cmd/root.go

@@ -32,4 +32,5 @@ func init() {
 	rootCmd.AddCommand(listCmd)
 	rootCmd.AddCommand(restoreCmd)
 	rootCmd.AddCommand(removeCmd)
+	rootCmd.AddCommand(replaceCmd)
 }

tests/replace_test.go

@@ -0,0 +1,33 @@
+package tests
+
+import (
+	"os/exec"
+	"testing"
+
+	"github.com/leonvogt/lunar/internal"
+)
+
+func TestReplace(t *testing.T) {
+	SetupTestDatabase()
+
+	// Create a snapshot
+	command := "go run ../main.go snapshot production"
+	err := exec.Command("sh", "-c", command).Run()
+	if err != nil {
+		t.Errorf("Error: %v", err)
+	}
+
+	if !DoesDatabaseExists("lunar_snapshot__lunar_test__production") {
+		t.Errorf("Expected database `lunar_snapshot__lunar_test__production` to exist - but it does not")
+	}
+
+	// Replace the snapshot
+	command = "go run ../main.go replace production"
+	err = exec.Command("sh", "-c", command).Run()
+	if err != nil {
+		t.Errorf("Error: %v", err)
+	}
+
+	// Cleanup
+	internal.DropDatabase("lunar_snapshot__lunar_test__production")
+}