| /* |
| Copyright 2014 The Kubernetes Authors. |
| |
| Licensed under the Apache License, Version 2.0 (the "License"); |
| you may not use this file except in compliance with the License. |
| You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| */ |
| |
| package runtime |
| |
| import ( |
| "fmt" |
| "testing" |
| ) |
| |
| func TestHandleCrash(t *testing.T) { |
| defer func() { |
| if x := recover(); x == nil { |
| t.Errorf("Expected a panic to recover from") |
| } |
| }() |
| defer HandleCrash() |
| panic("Test Panic") |
| } |
| |
| func TestCustomHandleCrash(t *testing.T) { |
| old := PanicHandlers |
| defer func() { PanicHandlers = old }() |
| var result interface{} |
| PanicHandlers = []func(interface{}){ |
| func(r interface{}) { |
| result = r |
| }, |
| } |
| func() { |
| defer func() { |
| if x := recover(); x == nil { |
| t.Errorf("Expected a panic to recover from") |
| } |
| }() |
| defer HandleCrash() |
| panic("test") |
| }() |
| if result != "test" { |
| t.Errorf("did not receive custom handler") |
| } |
| } |
| |
| func TestCustomHandleError(t *testing.T) { |
| old := ErrorHandlers |
| defer func() { ErrorHandlers = old }() |
| var result error |
| ErrorHandlers = []func(error){ |
| func(err error) { |
| result = err |
| }, |
| } |
| err := fmt.Errorf("test") |
| HandleError(err) |
| if result != err { |
| t.Errorf("did not receive custom handler") |
| } |
| } |