[SNOW-856228] Improve assert library (#947)

Author: sfc-gh-knozderko

assert_test.go

@@ -5,6 +5,7 @@ package gosnowflake
 import (
 	"fmt"
 	"reflect"
+	"runtime"
 	"strings"
 	"testing"
 )
@@ -35,16 +36,20 @@ func assertHasPrefixE(t *testing.T, actual string, expectedPrefix string, descri
 
 func fatalOnNonEmpty(t *testing.T, errMsg string) {
 	if errMsg != "" {
-		t.Fatal(errMsg)
+		t.Fatal(formatErrorMessage(errMsg))
 	}
 }
 
 func errorOnNonEmpty(t *testing.T, errMsg string) {
 	if errMsg != "" {
-		t.Error(errMsg)
+		t.Error(formatErrorMessage(errMsg))
 	}
 }
 
+func formatErrorMessage(errMsg string) string {
+	return fmt.Sprintf("%s. Thrown from %s", errMsg, thrownFrom())
+}
+
 func validateNil(actual any, descriptions ...string) string {
 	if isNil(actual) {
 		return ""
@@ -96,3 +101,16 @@ func isNil(value any) bool {
 	val := reflect.ValueOf(value)
 	return val.Kind() == reflect.Pointer && val.IsNil()
 }
+
+func thrownFrom() string {
+	buf := make([]byte, 1024)
+	size := runtime.Stack(buf, false)
+	stack := string(buf[0:size])
+	lines := strings.Split(stack, "\n\t")
+	for i, line := range lines {
+		if i > 0 && !strings.Contains(line, "assert_test.go") {
+			return line
+		}
+	}
+	return stack
+}