add tests for MustParseBytes func

nleof · Jul 25, 2016 · 11d8984 · 11d8984
1 parent 51cd20d
commit 11d8984
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/goyesql_test.go b/goyesql_test.go
@@ -22,6 +22,24 @@ func TestMustParseFileNoPanic(t *testing.T) {
 	MustParseFile("tests/samples/valid.sql")
 }
 
+func TestMustParseBytesPanic(t *testing.T) {
+	defer func() {
+		if r := recover(); r == nil {
+			t.Errorf("MustParseBytes should panic if an error occurs, got '%s'", r)
+		}
+	}()
+	MustParseBytes([]byte("I won't work"))
+}
+
+func TestMustParseBytesNoPanic(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			t.Errorf("MustParseBytes should not panic if an error occurs, got '%s'", r)
+		}
+	}()
+	MustParseBytes([]byte("-- name: byte-me\nSELECT * FROM bytes;"))
+}
+
 func BenchmarkMustParseFile(b *testing.B) {
 	for i := 0; i < b.N; i++ {
 		MustParseFile("tests/samples/valid.sql")