mirror of
https://github.com/hoernschen/dendrite.git
synced 2025-04-04 11:03:39 +00:00
109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package macaroon_test
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/base64"
|
|
"testing"
|
|
|
|
"gopkg.in/macaroon.v2"
|
|
)
|
|
|
|
func randomBytes(n int) []byte {
|
|
b := make([]byte, n)
|
|
_, err := rand.Read(b)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return b
|
|
}
|
|
|
|
func BenchmarkNew(b *testing.B) {
|
|
rootKey := randomBytes(24)
|
|
id := []byte(base64.StdEncoding.EncodeToString(randomBytes(100)))
|
|
loc := base64.StdEncoding.EncodeToString(randomBytes(40))
|
|
b.ResetTimer()
|
|
for i := b.N - 1; i >= 0; i-- {
|
|
MustNew(rootKey, id, loc, macaroon.LatestVersion)
|
|
}
|
|
}
|
|
|
|
func BenchmarkAddCaveat(b *testing.B) {
|
|
rootKey := randomBytes(24)
|
|
id := []byte(base64.StdEncoding.EncodeToString(randomBytes(100)))
|
|
loc := base64.StdEncoding.EncodeToString(randomBytes(40))
|
|
b.ResetTimer()
|
|
for i := b.N - 1; i >= 0; i-- {
|
|
b.StopTimer()
|
|
m := MustNew(rootKey, id, loc, macaroon.LatestVersion)
|
|
b.StartTimer()
|
|
m.AddFirstPartyCaveat([]byte("some caveat stuff"))
|
|
}
|
|
}
|
|
|
|
func benchmarkVerify(b *testing.B, mspecs []macaroonSpec) {
|
|
rootKey, macaroons := makeMacaroons(mspecs)
|
|
check := func(string) error {
|
|
return nil
|
|
}
|
|
b.ResetTimer()
|
|
for i := b.N - 1; i >= 0; i-- {
|
|
err := macaroons[0].Verify(rootKey, check, macaroons[1:])
|
|
if err != nil {
|
|
b.Fatalf("verification failed: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkVerifyLarge(b *testing.B) {
|
|
benchmarkVerify(b, multilevelThirdPartyCaveatMacaroons)
|
|
}
|
|
|
|
func BenchmarkVerifySmall(b *testing.B) {
|
|
benchmarkVerify(b, []macaroonSpec{{
|
|
rootKey: "root-key",
|
|
id: "root-id",
|
|
caveats: []caveat{{
|
|
condition: "wonderful",
|
|
}},
|
|
}})
|
|
}
|
|
|
|
func BenchmarkMarshalJSON(b *testing.B) {
|
|
rootKey := randomBytes(24)
|
|
id := []byte(base64.StdEncoding.EncodeToString(randomBytes(100)))
|
|
loc := base64.StdEncoding.EncodeToString(randomBytes(40))
|
|
m := MustNew(rootKey, id, loc, macaroon.LatestVersion)
|
|
b.ResetTimer()
|
|
for i := b.N - 1; i >= 0; i-- {
|
|
_, err := m.MarshalJSON()
|
|
if err != nil {
|
|
b.Fatalf("cannot marshal JSON: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func MustNew(rootKey, id []byte, loc string, vers macaroon.Version) *macaroon.Macaroon {
|
|
m, err := macaroon.New(rootKey, id, loc, vers)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return m
|
|
}
|
|
|
|
func BenchmarkUnmarshalJSON(b *testing.B) {
|
|
rootKey := randomBytes(24)
|
|
id := []byte(base64.StdEncoding.EncodeToString(randomBytes(100)))
|
|
loc := base64.StdEncoding.EncodeToString(randomBytes(40))
|
|
m := MustNew(rootKey, id, loc, macaroon.LatestVersion)
|
|
data, err := m.MarshalJSON()
|
|
if err != nil {
|
|
b.Fatalf("cannot marshal JSON: %v", err)
|
|
}
|
|
for i := b.N - 1; i >= 0; i-- {
|
|
var m macaroon.Macaroon
|
|
err := m.UnmarshalJSON(data)
|
|
if err != nil {
|
|
b.Fatalf("cannot unmarshal JSON: %v", err)
|
|
}
|
|
}
|
|
}
|