Add tests for multiple search/replaces

Author: chriszarate

main_test.go

@@ -11,12 +11,12 @@ import (
 
 var (
 	_, b, _, _ = runtime.Caller(0)
-	basepath   = filepath.Dir(b)
+	basePath   = filepath.Dir(b)
 )
 
-func doMainTest(t *testing.T, from []string, to []string, input string, expected string) {
-	// TODO: Support multiple from-to pairs
-	cmd := exec.Command("go", "run", basepath, from[0], to[0])
+func doMainTest(t *testing.T, input string, expected string, mainArgs []string) {
+	execArgs := append([]string{"run", basePath}, mainArgs...)
+	cmd := exec.Command("go", execArgs...)
 
 	cmd.Stdin = strings.NewReader(input)
 	var out bytes.Buffer
@@ -33,17 +33,54 @@ func doMainTest(t *testing.T, from []string, to []string, input string, expected
 }
 
 func TestSimpleReplaceWithNewlineAtEOF(t *testing.T) {
-	from := []string{"uss-enterprise.com"}
-	to := []string{"ncc-1701-d.space"}
-	input := "Space, the final frontier!\nCheck out: https://uss-enterprise.com/decks/10/sections/forward\n"
+	mainArgs := []string{
+		"http://uss-enterprise.com",
+		"https://ncc-1701-d.space",
+	}
+
+	input := "Space, the final frontier!\nCheck out: http://uss-enterprise.com/decks/10/sections/forward\n"
 	expected := "Space, the final frontier!\nCheck out: https://ncc-1701-d.space/decks/10/sections/forward\n"
-	doMainTest(t, from, to, input, expected)
+	doMainTest(t, input, expected, mainArgs)
 }
 
 func TestSimpleReplaceWithoutNewlineAtEOF(t *testing.T) {
-	from := []string{"uss-enterprise.com"}
-	to := []string{"ncc-1701-d.space"}
-	input := "I tend bar, and I listen.\nhttps://uss-enterprise.com/personnel/guinan"
+	mainArgs := []string{
+		"http://uss-enterprise.com",
+		"https://ncc-1701-d.space",
+	}
+	input := "I tend bar, and I listen.\nhttp://uss-enterprise.com/personnel/guinan"
 	expected := "I tend bar, and I listen.\nhttps://ncc-1701-d.space/personnel/guinan"
-	doMainTest(t, from, to, input, expected)
+	doMainTest(t, input, expected, mainArgs)
+}
+
+func TestMultipleReplaceWithNewlineAtEOF(t *testing.T) {
+	mainArgs := []string{
+		"http://uss-enterprise.com",
+		"https://ncc-1701-d.space",
+
+		"sections",
+		"areas",
+
+		"https",
+		"warp",
+	}
+	input := "Space, the final frontier!\nCheck out: http://uss-enterprise.com/decks/10/sections/forward\n"
+	expected := "Space, the final frontier!\nCheck out: warp://ncc-1701-d.space/decks/10/areas/forward\n"
+	doMainTest(t, input, expected, mainArgs)
 }
+
+func TestMultipleReplaceWithoutNewlineAtEOF(t *testing.T) {
+	mainArgs := []string{
+		"http://uss-enterprise.com",
+		"https://ncc-1701-d.space",
+
+		"sections",
+		"areas",
+
+		"https",
+		"warp",
+	}
+	input := "Space, the final frontier!\nCheck out: http://uss-enterprise.com/decks/10/sections/forward"
+	expected := "Space, the final frontier!\nCheck out: warp://ncc-1701-d.space/decks/10/areas/forward"
+	doMainTest(t, input, expected, mainArgs)
+}