From e7c9d4b7714712a2d4e21838ca9875cf725dc6bc Mon Sep 17 00:00:00 2001
From: Paul Bellamy <paul.a.bellamy@gmail.com>
Date: Mon, 18 Jan 2016 17:40:51 +0000
Subject: [PATCH] adding a benchmark for the various render topologies

---
 render/benchmark_internal_test.go | 66 +++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 render/benchmark_internal_test.go

diff --git a/render/benchmark_internal_test.go b/render/benchmark_internal_test.go
new file mode 100644
index 000000000..03dc0a345
--- /dev/null
+++ b/render/benchmark_internal_test.go
@@ -0,0 +1,66 @@
+package render_test
+
+import (
+	"testing"
+
+	"github.com/weaveworks/scope/render"
+	"github.com/weaveworks/scope/test/fixture"
+)
+
+func BenchmarkEndpointRender(b *testing.B) { benchmarkRender(b, render.EndpointRenderer) }
+func BenchmarkEndpointStats(b *testing.B)  { benchmarkStats(b, render.EndpointRenderer) }
+func BenchmarkProcessRender(b *testing.B)  { benchmarkRender(b, render.ProcessRenderer) }
+func BenchmarkProcessStats(b *testing.B)   { benchmarkStats(b, render.ProcessRenderer) }
+func BenchmarkProcessWithContainerNameRender(b *testing.B) {
+	benchmarkRender(b, render.ProcessWithContainerNameRenderer)
+}
+func BenchmarkProcessWithContainerNameStats(b *testing.B) {
+	benchmarkStats(b, render.ProcessWithContainerNameRenderer)
+}
+func BenchmarkProcessNameRender(b *testing.B) { benchmarkRender(b, render.ProcessNameRenderer) }
+func BenchmarkProcessNameStats(b *testing.B)  { benchmarkStats(b, render.ProcessNameRenderer) }
+func BenchmarkContainerRender(b *testing.B)   { benchmarkRender(b, render.ContainerRenderer) }
+func BenchmarkContainerStats(b *testing.B)    { benchmarkStats(b, render.ContainerRenderer) }
+func BenchmarkContainerWithImageNameRender(b *testing.B) {
+	benchmarkRender(b, render.ContainerWithImageNameRenderer)
+}
+func BenchmarkContainerWithImageNameStats(b *testing.B) {
+	benchmarkStats(b, render.ContainerWithImageNameRenderer)
+}
+func BenchmarkContainerImageRender(b *testing.B) { benchmarkRender(b, render.ContainerImageRenderer) }
+func BenchmarkContainerImageStats(b *testing.B)  { benchmarkStats(b, render.ContainerImageRenderer) }
+func BenchmarkContainerHostnameRender(b *testing.B) {
+	benchmarkRender(b, render.ContainerHostnameRenderer)
+}
+func BenchmarkContainerHostnameStats(b *testing.B) {
+	benchmarkStats(b, render.ContainerHostnameRenderer)
+}
+func BenchmarkHostRender(b *testing.B)       { benchmarkRender(b, render.HostRenderer) }
+func BenchmarkHostStats(b *testing.B)        { benchmarkStats(b, render.HostRenderer) }
+func BenchmarkPodRender(b *testing.B)        { benchmarkRender(b, render.PodRenderer) }
+func BenchmarkPodStats(b *testing.B)         { benchmarkStats(b, render.PodRenderer) }
+func BenchmarkPodServiceRender(b *testing.B) { benchmarkRender(b, render.PodServiceRenderer) }
+func BenchmarkPodServiceStats(b *testing.B)  { benchmarkStats(b, render.PodServiceRenderer) }
+
+func benchmarkRender(b *testing.B, r render.Renderer) {
+	var result map[string]render.RenderableNode
+	b.ReportAllocs()
+	b.ResetTimer()
+
+	for i := 0; i < b.N; i++ {
+		result = r.Render(fixture.Report)
+		if len(result) == 0 {
+			b.Errorf("Rendered topology contained no nodes")
+		}
+	}
+}
+
+func benchmarkStats(b *testing.B, r render.Renderer) {
+	b.ReportAllocs()
+	b.ResetTimer()
+
+	for i := 0; i < b.N; i++ {
+		// No way to tell if this was successful :(
+		r.Stats(fixture.Report)
+	}
+}