Fix flag parsing with go test, and switch to -short to gate long tests

1 files changed, 19 insertions, 0 deletions

diff --git a/init_test.go b/init_test.go
new file mode 100644
index 0000000..37e510b
--- /dev/null
+++ b/init_test.go
@@ -0,0 +1,19 @@
+// Copyright 2019 Nick White.
+// Use of this source code is governed by the GPLv3
+// license that can be found in the LICENSE file.
+
+package preproc
+
+import (
+	"flag"
+	"os"
+	"testing"
+)
+
+var update = flag.Bool("update", false, "update golden files")
+
+// TestMain is needed to ensure flags are parsed
+func TestMain(m *testing.M) {
+	flag.Parse()
+	os.Exit(m.Run())
+}