sjwt/benchmark_test.go at master · brianvoe/sjwt · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package sjwt

import (
	"fmt"
	"testing"
)

type benchCase struct {
	name           string
	generateSecret []byte
	verifySecret   []byte
	buildClaims    func() *Claims
	mutateToken    func(string) string
	skipVerify     bool
}

func makeBenchCases() []benchCase {
	return []benchCase{
		{
			name:           "small",
			generateSecret: []byte("benchmark-small-secret-01234567890123"),
			verifySecret:   []byte("benchmark-small-secret-01234567890123"),
			buildClaims: func() *Claims {
				c := New()
				c.Set("user", "123")
				c.Set("role", "admin")
				c.Set("active", true)
				return c
			},
		},
		{
			name:           "medium",
			generateSecret: []byte("benchmark-medium-secret-012345678901"),
			verifySecret:   []byte("benchmark-medium-secret-012345678901"),
			buildClaims: func() *Claims {
				c := New()
				for i := 0; i < 32; i++ {
					c.Set(keyForIndex(i), i)
				}
				return c
			},
		},
		{
			name:           "large",
			generateSecret: []byte("benchmark-large-secret-01234567890123"),
			verifySecret:   []byte("benchmark-large-secret-01234567890123"),
			buildClaims: func() *Claims {
				c := New()
				for i := 0; i < 200; i++ {
					c.Set(keyForIndex(i), i)
				}
				return c
			},
		},
	}
}

func keyForIndex(i int) string {
	return keyPrefix(i) + "_" + fmt.Sprintf("%03d", i)
}

func keyPrefix(i int) string {
	switch {
	case i < 10:
		return "claim_small"
	case i < 100:
		return "claim_medium"
	default:
		return "claim_large"
	}
}

func BenchmarkFullUsage(b *testing.B) {
	cases := makeBenchCases()
	for _, bc := range cases {
		bc := bc
		b.Run(bc.name, func(b *testing.B) {
			claims := bc.buildClaims()
			token, err := claims.Generate(bc.generateSecret)
			if err != nil {
				b.Fatalf("generate failed: %v", err)
			}
			if bc.mutateToken != nil {
				token = bc.mutateToken(token)
			}

			b.ResetTimer()
			for i := 0; i < b.N; i++ {
				claims := bc.buildClaims()
				token, err := claims.Generate(bc.generateSecret)
				if err != nil {
					b.Fatalf("generate failed: %v", err)
				}

				parsed, err := Parse(token)
				if err != nil {
					b.Fatalf("parse failed: %v", err)
				}

				if !bc.skipVerify && !Verify(token, bc.verifySecret) {
					b.Fatalf("verify failed")
				}

				if err := parsed.Validate(); err != nil {
					b.Fatalf("validate failed: %v", err)
				}
			}
		})
	}
}