Merge pull request #2 from rrrkren/eric/blockstring

block string

argument.go

@@ -60,6 +60,10 @@ func ArgumentString(name string, value string) Argument {
 	return Argument{name, argString(value)}
 }
 
+func ArgumentBlockString(name string, value string) Argument {
+	return Argument{name, argBlockString(value)}
+}
+
 func ArgumentEnum(name string, value string) Argument {
 	return Argument{name, argEnum(value)}
 }
@@ -130,6 +134,20 @@ func (v argString) stringChan() <-chan string {
 	return tokenChan
 }
 
+
+// argBlockString represents a block string value.
+type argBlockString string
+
+func (v argBlockString) stringChan() <- chan string {
+	tokenChan := make(chan string)
+	go func() {
+		tokenChan <- fmt.Sprintf(`"""%s"""`, v)
+		close(tokenChan)
+	}()
+	return tokenChan
+}
+
+
 // argEnum represents a enum value.
 type argEnum string
 

argument_test.go

@@ -59,6 +59,13 @@ func TestArgumentString(t *testing.T) {
 	assert.Equal(t, Argument{"blocked", argString("")}, a)
 }
 
+func TestArgumentBlockString(t *testing.T) {
+	a := ArgumentBlockString("blocked", `a`)
+	assert.Equal(t, Argument{"blocked", argBlockString("a")}, a)
+	a = ArgumentBlockString("blocked", ``)
+	assert.Equal(t, Argument{"blocked", argBlockString("")}, a)
+}
+
 func TestArgumentStringSlice(t *testing.T) {
 	a := ArgumentStringSlice("blocked", "a", "b", "", " ", "d")
 	assert.Equal(t, Argument{"blocked", argStringSlice([]string{"a", "b", "", " ", "d"})}, a)
@@ -87,6 +94,16 @@ func Test_argEnum(t *testing.T) {
 	assert.Equal(t, 1, i)
 }
 
+func Test_argBlockString(t *testing.T) {
+	b := argBlockString(`blockstring`)
+	i := 0
+	for str := range b.stringChan() {
+		assert.Equal(t, "\"\"\"blockstring\"\"\"", str)
+		i++
+	}
+	assert.Equal(t, 1, i)
+}
+
 func Test_argEnumSlice(t *testing.T) {
 	es := argEnumSlice([]string{"ENUM_VALUE", "ENUM_SLICE"})
 	c := es.stringChan()