mirror of
https://github.com/astaxie/beego.git
synced 2024-12-24 02:20:50 +00:00
39 lines
715 B
Go
39 lines
715 B
Go
|
package captcha
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
"github.com/astaxie/beego/utils"
|
||
|
)
|
||
|
|
||
|
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 := utils.RandomCreateBytes(challengeNums, defaultChars...)
|
||
|
b.StartTimer()
|
||
|
for i := 0; i < b.N; i++ {
|
||
|
NewImage(d, stdWidth, stdHeight)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func BenchmarkImageWriteTo(b *testing.B) {
|
||
|
b.StopTimer()
|
||
|
d := utils.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
|
||
|
}
|
||
|
}
|