Implement benchmark

felixge · Jan 11, 2015 · 0bce954 · 0bce954
1 parent 69b5fe0
commit 0bce954
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/xxd_test.go b/xxd_test.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"bytes"
+	"crypto/rand"
 	"flag"
 	"fmt"
 	"io"
@@ -47,6 +48,19 @@ func TestXXD(t *testing.T) {
 	}
 }
 
+func BenchmarkXXD(b *testing.B) {
+	b.StopTimer()
+	data := make([]byte, b.N)
+	if _, err := io.ReadFull(rand.Reader, data); err != nil {
+		b.Fatal(err)
+	}
+	buf := bytes.NewBuffer(data)
+	b.StartTimer()
+	if err := XXD(buf, ioutil.Discard); err != nil {
+		b.Fatal(err)
+	}
+}
+
 func xxdNative(r io.Reader, w io.Writer) error {
 	xxd := exec.Command("xxd", "-")
 	xxd.Stdin = r