scottk/vault
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/vault.git synced 2025-11-01 19:17:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Exit ScanView if context has been cancelled (#7419)

Browse Source
This commit is contained in:
Michael Gaffney
2019-09-04 09:18:19 -04:00
committed by GitHub
parent 6c54c27ac0
commit 33fd94e56f
2 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
sdk/logical/storage_test.go
View File

@@ -34,6 +34,24 @@ func TestScanView(t *testing.T) {
}
}
func TestScanView_CancelContext(t *testing.T) {
s := prepKeyStorage(t)
ctx, cancelCtx := context.WithCancel(context.Background())
var i int
err := ScanView(ctx, s, func(path string) {
cancelCtx()
i++
})
if err == nil {
t.Error("Want context cancel err, got none")
}
if i != 1 {
t.Errorf("Want i==1, got %d", i)
}
}
func TestCollectKeys(t *testing.T) {
s := prepKeyStorage(t)