Test utils.IsDeadlock()

Author: Al2Klimov

utils/utils_test.go

@@ -4,6 +4,8 @@ import (
 	"context"
 	"encoding/hex"
 	"fmt"
+	"github.com/go-sql-driver/mysql"
+	"github.com/lib/pq"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"testing"
@@ -99,6 +101,40 @@ func TestChecksum(t *testing.T) {
 	}
 }
 
+func TestIsDeadlock(t *testing.T) {
+	msg := "Unsuccessful attempt of confusing the tested code."
+	code := [5]byte{0, 23, 42, 77, 255}
+
+	subtests := []struct {
+		name   string
+		input  error
+		output bool
+	}{
+		{"nil", nil, false},
+		{"deadline", context.DeadlineExceeded, false},
+		{"mysql1204", &mysql.MySQLError{Number: 1204}, false},
+		{"mysql1205", &mysql.MySQLError{Number: 1205}, true},
+		{"mysql1205_with_crap", &mysql.MySQLError{Number: 1205, SQLState: code, Message: msg}, true},
+		{"mysql1206", &mysql.MySQLError{Number: 1206}, false},
+		{"mysql1212", &mysql.MySQLError{Number: 1212}, false},
+		{"mysql1213", &mysql.MySQLError{Number: 1213}, true},
+		{"mysql1213_with_crap", &mysql.MySQLError{Number: 1213, SQLState: code, Message: msg}, true},
+		{"mysql1214", &mysql.MySQLError{Number: 1214}, false},
+		{"postgres40000", &pq.Error{Code: "40000"}, false},
+		{"postgres40001", &pq.Error{Code: "40001"}, true},
+		{"postgres40001_with_crap", &pq.Error{Code: "40001", Message: msg}, true},
+		{"postgres40002", &pq.Error{Code: "40002"}, false},
+		{"postgres40P01", &pq.Error{Code: "40P01"}, true},
+		{"postgres40P01_with_crap", &pq.Error{Code: "40P01", Message: msg}, true},
+	}
+
+	for _, st := range subtests {
+		t.Run(st.name, func(t *testing.T) {
+			require.Equal(t, st.output, IsDeadlock(st.input))
+		})
+	}
+}
+
 func TestChanFromSlice(t *testing.T) {
 	t.Run("Nil", func(t *testing.T) {
 		ch := ChanFromSlice[int](nil)