From d5c5922a42fce75be38650b1443296cb59585e33 Mon Sep 17 00:00:00 2001
From: Gabor Javorszky <gabor@suborbital.dev>
Date: Mon, 20 Mar 2023 20:33:44 +0000
Subject: [PATCH] Add benchmark for cached authorizer

---
 e2core/auth/authorizer_test.go | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/e2core/auth/authorizer_test.go b/e2core/auth/authorizer_test.go
index d3f0e8bd..e02d405a 100644
--- a/e2core/auth/authorizer_test.go
+++ b/e2core/auth/authorizer_test.go
@@ -2,6 +2,7 @@ package auth
 
 import (
 	"encoding/json"
+	"fmt"
 	"net/http"
 	"net/http/httptest"
 	"strings"
@@ -350,3 +351,34 @@ func TestAuthorizerCache_ExpiringEntry(t *testing.T) {
 		assert.True(t, tc.assertOpts(t, opts))
 	}
 }
+func BenchmarkCachedAuthorizer(b *testing.B) {
+	opts := int32(0)
+
+	svr := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		atomic.AddInt32(&opts, 1)
+		w.WriteHeader(http.StatusOK)
+		_ = json.NewEncoder(w).Encode(&TenantInfo{
+			AuthorizedParty: fmt.Sprintf("tester-%d", opts),
+			Environment:     fmt.Sprintf("env-%d", opts),
+			ID:              fmt.Sprintf("123-%d", opts),
+			Name:            fmt.Sprintf("functionname-%d", opts),
+		})
+	}))
+
+	authzCache := newAuthorizationCache(common.SystemTime(), 10*time.Minute)
+
+	authorizer := &AuthzClient{
+		httpClient: svr.Client(),
+		location:   svr.URL + "/api/v2/tenant/",
+		cache:      authzCache,
+	}
+
+	for i := 0; i < b.N; i++ {
+		sfx := b.N / 1000
+		_, _ = authorizer.Authorize(
+			NewAccessToken(fmt.Sprintf("sometoken-%d", sfx)),
+			fmt.Sprintf("ident-%d", sfx),
+			fmt.Sprintf("namespace-%d", sfx),
+			fmt.Sprintf("fnName-%d", sfx))
+	}
+}