From 76246841fb1684953c899f620088e426b29f49ff Mon Sep 17 00:00:00 2001
From: Louis <lspgn@users.noreply.github.com>
Date: Sat, 4 Nov 2023 16:04:47 -0700
Subject: [PATCH] decoders: add test/benchmark (#232)

---
 decoders/netflow/templates_test.go | 39 ++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 decoders/netflow/templates_test.go

diff --git a/decoders/netflow/templates_test.go b/decoders/netflow/templates_test.go
new file mode 100644
index 0000000..4ee8f5e
--- /dev/null
+++ b/decoders/netflow/templates_test.go
@@ -0,0 +1,39 @@
+package netflow
+
+import (
+	"testing"
+)
+
+func benchTemplatesAdd(ts NetFlowTemplateSystem, obs uint32, N int, b *testing.B) {
+	for n := 0; n <= N; n++ {
+		ts.AddTemplate(10, obs, uint16(n), n)
+	}
+}
+
+func BenchmarkTemplatesAdd(b *testing.B) {
+	ts := CreateTemplateSystem()
+	b.Log("Creating", b.N, "templates")
+	benchTemplatesAdd(ts, uint32(b.N)%0xffff+1, b.N, b)
+}
+
+func BenchmarkTemplatesAddGet(b *testing.B) {
+	ts := CreateTemplateSystem()
+	templates := 1000
+	b.Log("Adding", templates, "templates")
+	benchTemplatesAdd(ts, 1, templates, b)
+	b.Log("Getting", b.N, "templates")
+
+	for n := 0; n <= b.N; n++ {
+		data, err := ts.GetTemplate(10, 1, uint16(n%templates))
+		if err != nil {
+			b.Fatal(err)
+		}
+		dataC, ok := data.(int)
+		if !ok {
+			b.Fatal("template not an integer")
+		}
+		if dataC != n%templates {
+			b.Fatal("different values", dataC, "!=", n%templates)
+		}
+	}
+}