1 files changed, 65 insertions, 0 deletions
diff --git a/syntax/quote_test.go b/syntax/quote_test.go
new file mode 100644
index 0000000..be7498b
--- /dev/null
+++ b/syntax/quote_test.go
@@ -0,0 +1,65 @@
+// Copyright 2017 The Bazel Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package syntax
+
+import (
+	"strings"
+	"testing"
+)
+
+var quoteTests = []struct {
+	q   string // quoted
+	s   string // unquoted (actual string)
+	std bool   // q is standard form for s
+}{
+	{`""`, "", true},
+	{`''`, "", false},
+	{`"hello"`, `hello`, true},
+	{`'hello'`, `hello`, false},
+	{`"quote\"here"`, `quote"here`, true},
+	{`'quote"here'`, `quote"here`, false},
+	{`"quote'here"`, `quote'here`, true},
+	{`'quote\'here'`, `quote'here`, false},
+
+	{`"\a\b\f\n\r\t\v\x00\x7f"`, "\a\b\f\n\r\t\v\000\x7F", true},
+	{`"\a\b\f\n\r\t\v\x00\x7f"`, "\a\b\f\n\r\t\v\000\x7F", false},
+	{`"\a\b\f\n\r\t\v\x00\x7f"`, "\a\b\f\n\r\t\v\000\x7F", false},
+	{`"\a\b\f\n\r\t\v\x00\x7f\"'\\\x03"`, "\a\b\f\n\r\t\v\x00\x7F\"'\\\x03", true},
+	{`"\a\b\f\n\r\t\v\x00\x7f\"'\\\x03"`, "\a\b\f\n\r\t\v\x00\x7F\"'\\\x03", false},
+	{`"\a\b\f\n\r\t\v\x00\x7f\"'\\\x03"`, "\a\b\f\n\r\t\v\x00\x7F\"'\\\x03", false},
+	{`"\a\b\f\n\r\t\v\x00\x7f\"\\\x03"`, "\a\b\f\n\r\t\v\x00\x7F\"\\\x03", false},
+	{
+		`"cat $(SRCS) | grep '\\s*ip_block:' | sed -e 's/\\s*ip_block: \"\\([^ ]*\\)\"/    \x27\\1\x27,/g' >> $@; "`,
+		"cat $(SRCS) | grep '\\s*ip_block:' | sed -e 's/\\s*ip_block: \"\\([^ ]*\\)\"/    '\\1',/g' >> $@; ",
+		false,
+	},
+	{
+		`"cat $(SRCS) | grep '\\s*ip_block:' | sed -e 's/\\s*ip_block: \"\\([^ ]*\\)\"/    '\\1',/g' >> $@; "`,
+		"cat $(SRCS) | grep '\\s*ip_block:' | sed -e 's/\\s*ip_block: \"\\([^ ]*\\)\"/    '\\1',/g' >> $@; ",
+		true,
+	},
+}
+
+func TestQuote(t *testing.T) {
+	for _, tt := range quoteTests {
+		if !tt.std {
+			continue
+		}
+		q := Quote(tt.s, false)
+		if q != tt.q {
+			t.Errorf("quote(%#q) = %s, want %s", tt.s, q, tt.q)
+		}
+	}
+}
+
+func TestUnquote(t *testing.T) {
+	for _, tt := range quoteTests {
+		s, triple, _, err := unquote(tt.q)
+		wantTriple := strings.HasPrefix(tt.q, `"""`) || strings.HasPrefix(tt.q, `'''`)
+		if s != tt.s || triple != wantTriple || err != nil {
+			t.Errorf("unquote(%s) = %#q, %v, %v want %#q, %v, nil", tt.q, s, triple, err, tt.s, wantTriple)
+		}
+	}
+}