You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
1.9 KiB
Go
94 lines
1.9 KiB
Go
package syncx
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type dummyResource struct {
|
|
age int
|
|
}
|
|
|
|
func (dr *dummyResource) Close() error {
|
|
return errors.New("close")
|
|
}
|
|
|
|
func TestResourceManager_GetResource(t *testing.T) {
|
|
manager := NewResourceManager()
|
|
defer manager.Close()
|
|
|
|
var age int
|
|
for i := 0; i < 10; i++ {
|
|
val, err := manager.GetResource("key", func() (io.Closer, error) {
|
|
age++
|
|
return &dummyResource{
|
|
age: age,
|
|
}, nil
|
|
})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, val.(*dummyResource).age)
|
|
}
|
|
}
|
|
|
|
func TestResourceManager_GetResourceError(t *testing.T) {
|
|
manager := NewResourceManager()
|
|
defer manager.Close()
|
|
|
|
for i := 0; i < 10; i++ {
|
|
_, err := manager.GetResource("key", func() (io.Closer, error) {
|
|
return nil, errors.New("fail")
|
|
})
|
|
assert.NotNil(t, err)
|
|
}
|
|
}
|
|
|
|
func TestResourceManager_Close(t *testing.T) {
|
|
manager := NewResourceManager()
|
|
defer manager.Close()
|
|
|
|
for i := 0; i < 10; i++ {
|
|
_, err := manager.GetResource("key", func() (io.Closer, error) {
|
|
return nil, errors.New("fail")
|
|
})
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
if assert.NoError(t, manager.Close()) {
|
|
assert.Equal(t, 0, len(manager.resources))
|
|
}
|
|
}
|
|
|
|
func TestResourceManager_UseAfterClose(t *testing.T) {
|
|
manager := NewResourceManager()
|
|
defer manager.Close()
|
|
|
|
_, err := manager.GetResource("key", func() (io.Closer, error) {
|
|
return nil, errors.New("fail")
|
|
})
|
|
assert.NotNil(t, err)
|
|
if assert.NoError(t, manager.Close()) {
|
|
_, err = manager.GetResource("key", func() (io.Closer, error) {
|
|
return nil, errors.New("fail")
|
|
})
|
|
assert.NotNil(t, err)
|
|
}
|
|
}
|
|
|
|
func TestResourceManager_Inject(t *testing.T) {
|
|
manager := NewResourceManager()
|
|
defer manager.Close()
|
|
|
|
manager.Inject("key", &dummyResource{
|
|
age: 10,
|
|
})
|
|
|
|
val, err := manager.GetResource("key", func() (io.Closer, error) {
|
|
return nil, nil
|
|
})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 10, val.(*dummyResource).age)
|
|
}
|