feat(benchmarks): add protected route benchmarks

Author: shamikkarkhanis

tests/benchmarks/benchmark_test.go

@@ -1,6 +1,9 @@
 package benchmarks
 
 import (
+	"bytes"
+	"encoding/json"
+	"fmt"
 	"io"
 	"net/http"
 	"testing"
@@ -80,3 +83,160 @@ func BenchmarkAuthMicrosoftInitiation(b *testing.B) {
 		resp.Body.Close()
 	}
 }
+
+func BenchmarkGetMe(b *testing.B) {
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		req, _ := http.NewRequest("GET", benchServer.URL+"/v1/auth/me", nil)
+		req.AddCookie(&http.Cookie{Name: "capy_auth", Value: benchJWTToken})
+		resp, err := benchClient.Do(req)
+		if err != nil {
+			b.Fatalf("failed to make request: %v", err)
+		}
+		if resp.StatusCode != http.StatusOK {
+			b.Fatalf("unexpected status code: %d", resp.StatusCode)
+		}
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}
+}
+
+func BenchmarkGetUser(b *testing.B) {
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		req, _ := http.NewRequest("GET", benchServer.URL+"/v1/users/"+benchUserID, nil)
+		req.AddCookie(&http.Cookie{Name: "capy_auth", Value: benchJWTToken})
+		resp, err := benchClient.Do(req)
+		if err != nil {
+			b.Fatalf("failed to make request: %v", err)
+		}
+		if resp.StatusCode != http.StatusOK {
+			b.Fatalf("unexpected status code: %d", resp.StatusCode)
+		}
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}
+}
+
+func BenchmarkListOrganizations(b *testing.B) {
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		req, _ := http.NewRequest("GET", benchServer.URL+"/v1/organizations", nil)
+		req.AddCookie(&http.Cookie{Name: "capy_auth", Value: benchJWTToken})
+		resp, err := benchClient.Do(req)
+		if err != nil {
+			b.Fatalf("failed to make request: %v", err)
+		}
+		if resp.StatusCode != http.StatusOK {
+			b.Fatalf("unexpected status code: %d", resp.StatusCode)
+		}
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}
+}
+
+func BenchmarkCreateOrganization(b *testing.B) {
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		body := map[string]string{
+			"name": fmt.Sprintf("Bench Org %d", i),
+			"slug": fmt.Sprintf("bench-org-%d", i),
+		}
+		jsonBody, _ := json.Marshal(body)
+		req, _ := http.NewRequest("POST", benchServer.URL+"/v1/organizations", bytes.NewBuffer(jsonBody))
+		req.AddCookie(&http.Cookie{Name: "capy_auth", Value: benchJWTToken})
+		req.Header.Set("Content-Type", "application/json")
+		resp, err := benchClient.Do(req)
+		if err != nil {
+			b.Fatalf("failed to make request: %v", err)
+		}
+		if resp.StatusCode != http.StatusCreated {
+			b.Fatalf("unexpected status code: %d", resp.StatusCode)
+		}
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}
+}
+
+func BenchmarkGetOrganization(b *testing.B) {
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		req, _ := http.NewRequest("GET", benchServer.URL+"/v1/organizations/"+benchOrgID, nil)
+		req.AddCookie(&http.Cookie{Name: "capy_auth", Value: benchJWTToken})
+		resp, err := benchClient.Do(req)
+		if err != nil {
+			b.Fatalf("failed to make request: %v", err)
+		}
+		if resp.StatusCode != http.StatusOK {
+			b.Fatalf("unexpected status code: %d", resp.StatusCode)
+		}
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}
+}
+
+func BenchmarkListEvents(b *testing.B) {
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		req, _ := http.NewRequest("GET", benchServer.URL+"/v1/events", nil)
+		req.AddCookie(&http.Cookie{Name: "capy_auth", Value: benchJWTToken})
+		resp, err := benchClient.Do(req)
+		if err != nil {
+			b.Fatalf("failed to make request: %v", err)
+		}
+		if resp.StatusCode != http.StatusOK {
+			b.Fatalf("unexpected status code: %d", resp.StatusCode)
+		}
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}
+}
+
+func BenchmarkCreateEvent(b *testing.B) {
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		body := map[string]interface{}{
+			"location":    "Bench Location",
+			"description": "Bench Description",
+			"org_id":      benchOrgID,
+		}
+		jsonBody, _ := json.Marshal(body)
+		req, _ := http.NewRequest("POST", benchServer.URL+"/v1/events", bytes.NewBuffer(jsonBody))
+		req.AddCookie(&http.Cookie{Name: "capy_auth", Value: benchJWTToken})
+		req.Header.Set("Content-Type", "application/json")
+		resp, err := benchClient.Do(req)
+		if err != nil {
+			b.Fatalf("failed to make request: %v", err)
+		}
+		if resp.StatusCode != http.StatusCreated {
+			b.Fatalf("unexpected status code: %d", resp.StatusCode)
+		}
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}
+}
+
+func BenchmarkGetEvent(b *testing.B) {
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		req, _ := http.NewRequest("GET", benchServer.URL+"/v1/events/"+benchEventID, nil)
+		req.AddCookie(&http.Cookie{Name: "capy_auth", Value: benchJWTToken})
+		resp, err := benchClient.Do(req)
+		if err != nil {
+			b.Fatalf("failed to make request: %v", err)
+		}
+		if resp.StatusCode != http.StatusOK {
+			b.Fatalf("unexpected status code: %d", resp.StatusCode)
+		}
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}
+}