test: add more tests (#1106)

* style: format code

* test: add more tests

* fix: staticcheck errors
This commit is contained in:
Kevin Wan
2021-10-01 10:03:56 +08:00
committed by GitHub
parent d1bfb5ef61
commit d1e702e8a3
3 changed files with 54 additions and 6 deletions

View File

@@ -47,14 +47,28 @@ func TestResourceManager_GetResourceError(t *testing.T) {
func TestResourceManager_Close(t *testing.T) {
manager := NewResourceManager()
for i := 0; i < 10; i++ {
_, err := manager.GetResource("key", func() (io.Closer, error) {
return nil, errors.New("fail")
})
assert.NotNil(t, err)
}
err := manager.Close()
assert.NoError(t, err)
assert.Equal(t, 0, len(manager.resources))
if assert.NoError(t, manager.Close()) {
assert.Equal(t, 0, len(manager.resources))
}
}
func TestResourceManager_UseAfterClose(t *testing.T) {
manager := NewResourceManager()
_, 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)
}
}