Add more options to highlight

Fixes #1021

1 files changed, 42 insertions, 0 deletions

diff --git a/helpers/pygments_test.go b/helpers/pygments_test.go
new file mode 100644
index 000000000..be0c1a7dc
--- /dev/null
+++ b/helpers/pygments_test.go
@@ -0,0 +1,42 @@
+package helpers
+
+import (
+	"github.com/spf13/viper"
+	"testing"
+)
+
+func TestParsePygmentsArgs(t *testing.T) {
+	for i, this := range []struct {
+		in                 string
+		pygmentsStyle      string
+		pygmentsUseClasses bool
+		expect1            interface{}
+	}{
+		{"", "foo", true, "style=foo,noclasses=false,encoding=utf8"},
+		{"style=boo,noclasses=true", "foo", true, "encoding=utf8,noclasses=true,style=boo"},
+		{"Style=boo, noClasses=true", "foo", true, "encoding=utf8,noclasses=true,style=boo"},
+		{"noclasses=true", "foo", true, "encoding=utf8,noclasses=true,style=foo"},
+		{"style=boo", "foo", true, "encoding=utf8,noclasses=false,style=boo"},
+		{"boo=invalid", "foo", false, false},
+		{"style", "foo", false, false},
+	} {
+		viper.Set("PygmentsStyle", this.pygmentsStyle)
+		viper.Set("PygmentsUseClasses", this.pygmentsUseClasses)
+
+		result1, err := parsePygmentsOpts(this.in)
+		if b, ok := this.expect1.(bool); ok && !b {
+			if err == nil {
+				t.Errorf("[%d] parsePygmentArgs didn't return an expected error", i)
+			}
+		} else {
+			if err != nil {
+				t.Errorf("[%d] parsePygmentArgs failed: %s", i, err)
+				continue
+			}
+			if result1 != this.expect1 {
+				t.Errorf("[%d] parsePygmentArgs got %v but expected %v", i, result1, this.expect1)
+			}
+
+		}
+	}
+}