Add benchmarks for API tests

duda · Sep 19, 2014 · 463a16a · 463a16a
1 parent 54e0af8
commit 463a16a
Show file tree

Hide file tree

Showing 2 changed files with 75 additions and 0 deletions.
diff --git a/hack/benchmark-go.sh b/hack/benchmark-go.sh
@@ -0,0 +1,19 @@
+#!/bin/bash
+
+# Copyright 2014 Google Inc. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -e
+
+KUBE_COVER=" " KUBE_RACE=" " hack/test-go.sh "" -test.run="^X" -benchtime=1s -bench=. -benchmem
diff --git a/pkg/api/serialization_test.go b/pkg/api/serialization_test.go
@@ -17,7 +17,9 @@ limitations under the License.
 package api_test
 
 import (
+	"encoding/json"
 	"flag"
+	"math/rand"
 	"reflect"
 	"testing"
 
@@ -203,3 +205,57 @@ func TestBadJSONRejection(t *testing.T) {
 		t.Errorf("Kind is set but doesn't match the object type: %s", badJSONKindMismatch)
 	}*/
 }
+
+const benchmarkSeed = 100
+
+func BenchmarkEncode(b *testing.B) {
+	pod := api.Pod{}
+	apiObjectFuzzer.RandSource(rand.NewSource(benchmarkSeed))
+	apiObjectFuzzer.Fuzz(&pod)
+	for i := 0; i < b.N; i++ {
+		latest.Codec.Encode(&pod)
+	}
+}
+
+// BenchmarkEncodeJSON provides a baseline for regular JSON encode performance
+func BenchmarkEncodeJSON(b *testing.B) {
+	pod := api.Pod{}
+	apiObjectFuzzer.RandSource(rand.NewSource(benchmarkSeed))
+	apiObjectFuzzer.Fuzz(&pod)
+	for i := 0; i < b.N; i++ {
+		json.Marshal(&pod)
+	}
+}
+
+func BenchmarkDecode(b *testing.B) {
+	pod := api.Pod{}
+	apiObjectFuzzer.RandSource(rand.NewSource(benchmarkSeed))
+	apiObjectFuzzer.Fuzz(&pod)
+	data, _ := latest.Codec.Encode(&pod)
+	for i := 0; i < b.N; i++ {
+		latest.Codec.Decode(data)
+	}
+}
+
+func BenchmarkDecodeInto(b *testing.B) {
+	pod := api.Pod{}
+	apiObjectFuzzer.RandSource(rand.NewSource(benchmarkSeed))
+	apiObjectFuzzer.Fuzz(&pod)
+	data, _ := latest.Codec.Encode(&pod)
+	for i := 0; i < b.N; i++ {
+		obj := api.Pod{}
+		latest.Codec.DecodeInto(data, &obj)
+	}
+}
+
+// BenchmarkDecodeJSON provides a baseline for regular JSON decode performance
+func BenchmarkDecodeJSON(b *testing.B) {
+	pod := api.Pod{}
+	apiObjectFuzzer.RandSource(rand.NewSource(benchmarkSeed))
+	apiObjectFuzzer.Fuzz(&pod)
+	data, _ := latest.Codec.Encode(&pod)
+	for i := 0; i < b.N; i++ {
+		obj := api.Pod{}
+		json.Unmarshal(data, &obj)
+	}
+}