Fix some panics resulting from calling ExpirationManager.Stop directly. (#11547)

This commit is contained in:
Nick Cabatoff 2021-05-06 12:39:01 -04:00 committed by GitHub
parent 57b6e786fc
commit 0926e302c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions

View File

@ -2531,7 +2531,8 @@ func registerOneLease(t *testing.T, ctx context.Context, exp *ExpirationManager)
}
func TestExpiration_MarkIrrevocable(t *testing.T) {
exp := mockExpiration(t)
c, _, _ := TestCoreUnsealed(t)
exp := c.expiration
ctx := namespace.RootContext(nil)
leaseID := registerOneLease(t, ctx, exp)
@ -2570,7 +2571,7 @@ func TestExpiration_MarkIrrevocable(t *testing.T) {
}
// stop and restore to verify that irrevocable leases are properly loaded from storage
err = exp.Stop()
err = c.stopExpiration()
if err != nil {
t.Fatalf("error stopping expiration manager: %v", err)
}
@ -2646,7 +2647,8 @@ func TestExpiration_FetchLeaseTimesIrrevocable(t *testing.T) {
}
func TestExpiration_StopClearsIrrevocableCache(t *testing.T) {
exp := mockExpiration(t)
c, _, _ := TestCoreUnsealed(t)
exp := c.expiration
ctx := namespace.RootContext(nil)
leaseID := registerOneLease(t, ctx, exp)
@ -2656,7 +2658,7 @@ func TestExpiration_StopClearsIrrevocableCache(t *testing.T) {
}
exp.markLeaseIrrevocable(ctx, le, fmt.Errorf("test irrevocable error"))
err = exp.Stop()
err = c.stopExpiration()
if err != nil {
t.Fatalf("error stopping expiration manager: %v", err)
}