Add main tests

1 files changed, 40 insertions, 0 deletions

diff --git a/main/main_test.go b/main/main_test.go
new file mode 100644
index 0000000..a7a7795
--- /dev/null
+++ b/main/main_test.go
@@ -0,0 +1,40 @@
+package main
+
+import (
+	"strings"
+	"testing"
+)
+
+func TestMain(t *testing.T) {
+	type test struct {
+		program string
+		quiet bool
+		input string
+		expected string
+	}
+
+	tests := []test {
+		{
+			program: `s/#(~(people)~$_@(1$_#(~(first_name)~$_.|(..$_){-0})-|(..$_){-0})-|(..$_){-0})-/p`,
+			quiet: true,
+			input: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`,
+			expected: `"Tom"`,
+		},
+	}
+
+	for i, test := range tests {
+		t.Logf("Running test: %d", i)
+
+		var output strings.Builder
+		run(config {
+			quiet: test.quiet,
+			program: test.program,
+			in: strings.NewReader(test.input),
+			out: &output,
+		})
+		
+		if output.String() != test.expected {
+			t.Errorf("Ran '%s' and expected %s but got %s", test.program, test.expected, output.String())
+		}
+	}
+}