tmp

Author: rsteube

context_test.go

@@ -1,7 +1,6 @@
 package carapace
 
 import (
-	"fmt"
 	"os"
 	"path/filepath"
 	"strings"
@@ -97,36 +96,56 @@ func TestEnv(t *testing.T) {
 
 func TestEnvsubst(t *testing.T) {
 	c := Context{}
-
-	if s, err := c.Envsubst("start${example}end"); s != "startend" || err != nil {
-		t.Fail()
-	}
-
-	if s, err := c.Envsubst("start${!example}end"); s != "start${example}end" || err != nil {
-		fmt.Println(s)
-		t.Fail()
-	}
-
-	if s, err := c.Envsubst("start${example:-default}end"); s != "startdefaultend" || err != nil {
-		t.Fail()
-	}
-
-	if s, err := c.Envsubst("start${!example:-default}end"); s != "start${example:-default}end" || err != nil {
-		t.Fail()
-	}
-
-	c.Setenv("example", "value")
-	if s, err := c.Envsubst("start${example}end"); s != "startvalueend" || err != nil {
-		t.Fail()
-	}
-	if s, err := c.Envsubst("start${!example}end"); s != "start${example}end" || err != nil {
-		t.Fail()
-	}
-
-	if s, err := c.Envsubst("start${example:-default}end"); s != "startvalueend" || err != nil {
-		t.Fail()
-	}
-	if s, err := c.Envsubst("start${!example:-default}end"); s != "start${example:-default}end" || err != nil {
-		t.Fail()
-	}
+	c.Setenv("REPLACE", "me")
+	for s, expected := range map[string]string{
+		"BEFORE${REPLACE}AFTER":  "BEFOREmeAFTER",
+		"BEFORE${!REPLACE}AFTER": "BEFORE${REPLACE}AFTER",
+
+		"BEFORE${REPLACE:-default}AFTER":  "BEFOREmeAFTER",
+		"BEFORE${!REPLACE:-default}AFTER": "BEFORE${REPLACE:-default}AFTER",
+		"BEFORE${UNSET:-default}AFTER":    "BEFOREdefaultAFTER",
+		"BEFORE${!UNSET:-default}AFTER":   "BEFORE${UNSET:-default}AFTER",
+	} {
+		t.Run(s, func(t *testing.T) {
+			actual, err := c.Envsubst(s)
+			if err != nil {
+				t.Fatal(err)
+			}
+			if actual != expected {
+				t.Fatalf("invalid replacement\nexpected: %#v\nactual  : %#v", expected, actual)
+			}
+		})
+	}
+
+	// if s, err := c.Envsubst("start${example}end"); s != "startend" || err != nil {
+	// 	t.Fail()
+	// }
+
+	// if s, err := c.Envsubst("start${!example}end"); s != "start${example}end" || err != nil {
+	// 	fmt.Println(s)
+	// 	t.Fail()
+	// }
+
+	// if s, err := c.Envsubst("start${example:-default}end"); s != "startdefaultend" || err != nil {
+	// 	t.Fail()
+	// }
+
+	// if s, err := c.Envsubst("start${!example:-default}end"); s != "start${example:-default}end" || err != nil {
+	// 	t.Fail()
+	// }
+
+	// c.Setenv("example", "value")
+	// if s, err := c.Envsubst("start${example}end"); s != "startvalueend" || err != nil {
+	// 	t.Fail()
+	// }
+	// if s, err := c.Envsubst("start${!example}end"); s != "start${example}end" || err != nil {
+	// 	t.Fail()
+	// }
+
+	// if s, err := c.Envsubst("start${example:-default}end"); s != "startvalueend" || err != nil {
+	// 	t.Fail()
+	// }
+	// if s, err := c.Envsubst("start${!example:-default}end"); s != "start${example:-default}end" || err != nil {
+	// 	t.Fail()
+	// }
 }