43 lines
880 B
Go
43 lines
880 B
Go
// Copyright 2014 The Gogs Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package captcha
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/gogits/gogs/modules/base"
|
|
)
|
|
|
|
type byteCounter struct {
|
|
n int64
|
|
}
|
|
|
|
func (bc *byteCounter) Write(b []byte) (int, error) {
|
|
bc.n += int64(len(b))
|
|
return len(b), nil
|
|
}
|
|
|
|
func BenchmarkNewImage(b *testing.B) {
|
|
b.StopTimer()
|
|
d := base.RandomCreateBytes(challengeNums, defaultChars...)
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
NewImage(d, stdWidth, stdHeight)
|
|
}
|
|
}
|
|
|
|
func BenchmarkImageWriteTo(b *testing.B) {
|
|
b.StopTimer()
|
|
d := base.RandomCreateBytes(challengeNums, defaultChars...)
|
|
b.StartTimer()
|
|
counter := &byteCounter{}
|
|
for i := 0; i < b.N; i++ {
|
|
img := NewImage(d, stdWidth, stdHeight)
|
|
img.WriteTo(counter)
|
|
b.SetBytes(counter.n)
|
|
counter.n = 0
|
|
}
|
|
}
|