add tests

newrelic · Mar 3, 2024 · 153e555 · 153e555
1 parent a39543d
commit 153e555
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/v3/integrations/nrpkgerrors/nrkpgerrors_test.go b/v3/integrations/nrpkgerrors/nrkpgerrors_test.go
@@ -57,6 +57,7 @@ func TestWrappedStackTrace(t *testing.T) {
 		{Error: theta(basicError{}), ExpectTopFrame: ""},
 		{Error: basicNRError(basicError{}), ExpectTopFrame: ""},
 		{Error: withAttributes(basicError{}), ExpectTopFrame: "", ExpectAttributes: map[string]interface{}{"testAttribute": 1, "foo": 2}},
+		{Error: nil, ExpectTopFrame: ""},
 	}
 
 	for idx, tc := range testcases {
@@ -117,6 +118,7 @@ func TestWrappedErrorClass(t *testing.T) {
 		{Error: alpha(basicError{}), ExpectClass: "nrpkgerrors.basicError"},
 		{Error: wrapWithClass(basicError{}, "zip"), ExpectClass: "zip"},
 		{Error: alpha(wrapWithClass(basicError{}, "zip")), ExpectClass: "nrpkgerrors.basicError"},
+		{Error: nil, ExpectClass: "*errors.fundamental"},
 	}
 
 	for idx, tc := range testcases {

diff --git a/v3/integrations/nrpkgerrors/nrpkgerrors.go b/v3/integrations/nrpkgerrors/nrpkgerrors.go
@@ -84,7 +84,11 @@ var (
 // type.
 func Wrap(e error) error {
 	if e == nil {
-		return errNilError
+		return newrelic.Error{
+			Message: errNilError.Error(),
+			Class:   errorClass(errNilError),
+			Stack:   stackTrace(errNilError),
+		}
 	}
 
 	attributes := make(map[string]interface{})