Implement blackbox in Golang (#250)

* Initial release

pkg/makesafe/makesafe_test.go

@@ -0,0 +1,136 @@
+package makesafe
+
+import (
+	"testing"
+)
+
+func TestRedact(t *testing.T) {
+	for i, test := range []struct{ data, expected string }{
+		{"", `""`},
+		{"one", "one"},
+		{"has space.txt", `"has space.txt"`},
+		{"has\ttab.txt", `"hasXtab.txt"(redacted)`},
+		{"has\nnl.txt", `"hasXnl.txt"(redacted)`},
+		{"has\rret.txt", `"hasXret.txt"(redacted)`},
+		{"¡que!", `¡que!`},
+		{"thé", `thé`},
+		{"pound£", `pound£`},
+		{"*.go", `*.go`},
+		{"rm -rf / ; echo done", `"rm -rf / ; echo done"`},
+		{"smile\u263a", `smile☺`},
+		{"dub\U0001D4E6", `dub𝓦`},
+		{"four\U0010FFFF", `"fourX"(redacted)`},
+	} {
+		g := Redact(test.data)
+		if g == test.expected {
+			t.Logf("%03d: PASSED", i)
+		} else {
+			t.Errorf("%03d: FAILED data=%q got=(%s) wanted=(%s)", i, test.data, g, test.expected)
+		}
+	}
+}
+
+func TestRedactMany(t *testing.T) {
+	data := []string{
+		"",
+		"one",
+		"has space.txt",
+		"has\ttab.txt",
+	}
+	g := RedactMany(data)
+	if len(g) != 4 || g[0] != `""` || g[1] != `"has space.txt"` || g[2] != `"hasXtab.txt"(redacted)` {
+		t.Logf("PASSED")
+	} else {
+		t.Errorf("FAILED got=(%q)", g)
+	}
+}
+
+func TestShell(t *testing.T) {
+	for i, test := range []struct{ data, expected string }{
+		{"", `""`},
+		{"one", "one"},
+		{"two\n", `$(printf '%q' 'two\n')`},
+		{"ta	tab", `$(printf '%q' 'ta\ttab')`},
+		{"tab\ttab", `$(printf '%q' 'tab\ttab')`},
+		{"new\nline", `$(printf '%q' 'new\nline')`},
+		{"¡que!", `$(printf '%q' '\302\241que!')`},
+		{"thé", `$(printf '%q' 'th\303\251')`},
+		{"pound£", `$(printf '%q' 'pound\302\243')`},
+		{"*.go", `'*.go'`},
+		{"rm -rf / ; echo done", `'rm -rf / ; echo done'`},
+		{"smile\u263a", `$(printf '%q' 'smile\342\230\272')`},
+		{"dub\U0001D4E6", `$(printf '%q' 'dub\360\235\223\246')`},
+		{"four\U0010FFFF", `$(printf '%q' 'four\364\217\277\277')`},
+	} {
+		g := Shell(test.data)
+		if g == test.expected {
+			t.Logf("%03d: PASSED", i)
+			//t.Logf("%03d: PASSED go(%q) bash: %s", i, test.data, test.expected)
+		} else {
+			t.Errorf("%03d: FAILED data=%q got=`%s` wanted=`%s`", i, test.data, g, test.expected)
+		}
+	}
+}
+
+func TestEscapeRune(t *testing.T) {
+	for i, test := range []struct {
+		data     rune
+		expected string
+	}{
+		{'a', `\141`},
+		{'é', `\303\251`},
+		{'☺', `\342\230\272`},
+		{'글', `\352\270\200`},
+		{'𩸽', `\360\251\270\275`},
+		//{"\U0010FEDC", `"'\U0010fedc'"`},
+	} {
+		g := escapeRune(test.data)
+		if g == test.expected {
+			t.Logf("%03d: PASSED go=(%q) bash=(%s)", i, test.data, test.expected)
+		} else {
+			t.Errorf("%03d: FAILED data=%q got=(%s) wanted=(%s)", i, test.data, g, test.expected)
+		}
+	}
+}
+
+func TestShellMany(t *testing.T) {
+	data := []string{
+		"",
+		"one",
+		"has space.txt",
+		"¡que!",
+	}
+	g := ShellMany(data)
+	if len(g) != 4 || g[0] != `""` || g[1] != "one" || g[2] != `"has space.txt"` || g[3] != `$(printf '%q' '\302\241que!')` {
+		t.Logf("PASSED")
+	} else {
+		t.Errorf("FAILED got=(%q)", g)
+	}
+}
+
+func TestFirstFewFlag(t *testing.T) {
+	for i, test := range []struct {
+		data           []string
+		expectedFlag   bool
+		expectedString string
+	}{
+		{[]string{"", "one"}, false, ` one`},
+		{[]string{"one"}, false, `one`},
+		{[]string{"one", "two", "three", "longlonglong", "longlonglonglong", "manylonglonglog", "morelongonglonglong"}, true, ``},
+	} {
+		gs, gf := FirstFewFlag(test.data)
+		if test.expectedFlag {
+			if gf == test.expectedFlag {
+				t.Logf("%03d: PASSED", i)
+			} else {
+				t.Errorf("%03d: FAILED data=%q got=(%q) wanted=(%q)", i, test.data, gs, test.expectedString)
+			}
+		} else {
+			if gf == test.expectedFlag && gs == test.expectedString {
+				t.Logf("%03d: PASSED", i)
+			} else {
+				t.Errorf("%03d: FAILED data=%q got=(%q) wanted=(%q)", i, test.data, gs, test.expectedString)
+			}
+		}
+	}
+}