You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.0 KiB
65 lines
1.0 KiB
package group
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGroupGet(t *testing.T) {
|
|
count := 0
|
|
g := NewGroup(func() interface{} {
|
|
count++
|
|
return count
|
|
})
|
|
v := g.Get("/x/internal/dummy/user")
|
|
assert.Equal(t, 1, v.(int))
|
|
|
|
v = g.Get("/x/internal/dummy/avatar")
|
|
assert.Equal(t, 2, v.(int))
|
|
|
|
v = g.Get("/x/internal/dummy/user")
|
|
assert.Equal(t, 1, v.(int))
|
|
assert.Equal(t, 2, count)
|
|
}
|
|
|
|
func TestGroupReset(t *testing.T) {
|
|
g := NewGroup(func() interface{} {
|
|
return 1
|
|
})
|
|
g.Get("/x/internal/dummy/user")
|
|
call := false
|
|
g.Reset(func() interface{} {
|
|
call = true
|
|
return 1
|
|
})
|
|
|
|
length := 0
|
|
for range g.objs {
|
|
length++
|
|
}
|
|
|
|
assert.Equal(t, 0, length)
|
|
|
|
g.Get("/x/internal/dummy/user")
|
|
assert.Equal(t, true, call)
|
|
}
|
|
|
|
func TestGroupClear(t *testing.T) {
|
|
g := NewGroup(func() interface{} {
|
|
return 1
|
|
})
|
|
g.Get("/x/internal/dummy/user")
|
|
length := 0
|
|
for range g.objs {
|
|
length++
|
|
}
|
|
assert.Equal(t, 1, length)
|
|
|
|
g.Clear()
|
|
length = 0
|
|
for range g.objs {
|
|
length++
|
|
}
|
|
assert.Equal(t, 0, length)
|
|
}
|
|
|