initial 2

Author: tonyredondo

instrumentation/testing/init.go

@@ -57,11 +57,11 @@ func Init(m *testing.M) {
 			var err error
 			runPatch, err = mpatch.PatchMethodByReflect(tRunMethod, func(t *testing.T, name string, f func(t *testing.T)) bool {
 				logOnError(runPatch.Unpatch())
-				defer func() {
-					logOnError(runPatch.Patch())
-				}()
+				defer func() { logOnError(runPatch.Patch()) }()
 				pc, _, _, _ := runtime.Caller(1)
 				return t.Run(name, func(childT *testing.T) {
+					logOnError(runPatch.Patch())
+					defer func() { logOnError(runPatch.Unpatch()) }()
 					addAutoInstrumentedTest(childT)
 					childTest := StartTestFromCaller(childT, pc)
 					defer childTest.end()

instrumentation/testing/logger.go

@@ -1,6 +1,7 @@
 package testing
 
 import (
+	"fmt"
 	"reflect"
 	"sync"
 	"testing"
@@ -160,6 +161,7 @@ func patch(methodName string, methodBody func(test *Test, argsValues []reflect.V
 func logOnError(err error) {
 	if err != nil {
 		instrumentation.Logger().Println(err)
+		fmt.Println(err)
 	}
 }