From 458e61d6abfb1b6fe514f2843adfcc5247422d1f Mon Sep 17 00:00:00 2001
From: EuAndreh <eu@euandre.org>
Date: Wed, 25 Sep 2024 16:04:26 -0300
Subject: src/gobang.go: Add Timed()

---
 tests/gobang.go | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

(limited to 'tests/gobang.go')

diff --git a/tests/gobang.go b/tests/gobang.go
index e2dd61b..14fe829 100644
--- a/tests/gobang.go
+++ b/tests/gobang.go
@@ -383,6 +383,55 @@ func test_metric() {
 	})
 }
 
+func test_Timed() {
+	TestStart("Timed()")
+
+	Testing("we can use a noop thunk", func() {
+		savedLogger := slog.Default()
+		savedFlag   := emitMetric
+
+		s := new(strings.Builder)
+		SetLoggerOutput(s)
+		emitMetric = true
+
+		Timed("timer-label", func() any { return nil }, "key-1", "value-1")
+
+		var data map[string]interface{}
+		err := json.Unmarshal([]byte(s.String()), &data)
+		ErrorIf(err)
+
+		AssertEqual(data["type"].(string),  "timer")
+		AssertEqual(data["label"].(string), "timer-label")
+		AssertEqual(data["key-1"].(string), "value-1")
+
+		var ok bool
+		_, ok = data["start"].(string)
+		AssertEqual(ok, true)
+		_, ok = data["end"].(string)
+		AssertEqual(ok, true)
+		_, ok = data["duration"].(float64)
+		AssertEqual(ok, true)
+
+		slog.SetDefault(savedLogger)
+		emitMetric = savedFlag
+	})
+
+	Testing("it propagates the return value", func() {
+		savedLogger := slog.Default()
+		savedFlag   := emitMetric
+
+		s := new(strings.Builder)
+		SetLoggerOutput(s)
+		emitMetric = true
+
+		val := Timed("timer-label", func() any { return 123 }, "key-1", "value-1")
+		AssertEqual(val, 123)
+
+		slog.SetDefault(savedLogger)
+		emitMetric = savedFlag
+	})
+}
+
 func test_MakeCounter() {
 	TestStart("MakeCounter()")
 
@@ -875,6 +924,7 @@ func MainTest() {
 	test_Warning()
 	test_Error()
 	test_metric()
+	test_Timed()
 	test_MakeCounter()
 	test_MakeGauge()
 	test_ErrorIf()
-- 
cgit v1.2.3