snapshot of https://github.com/golang-migrate/migrate.git (master, 2026-09-15; blobs > 131072B omitted)

database/util_test.go

@@ -0,0 +1,108 @@
+package database
+
+import (
+	"errors"
+	"sync/atomic"
+	"testing"
+)
+
+func TestGenerateAdvisoryLockId(t *testing.T) {
+	testcases := []struct {
+		dbname     string
+		additional []string
+		expectedID string // empty string signifies that an error is expected
+	}{
+		{
+			dbname:     "database_name",
+			expectedID: "1764327054",
+		},
+		{
+			dbname:     "database_name",
+			additional: []string{"schema_name_1"},
+			expectedID: "2453313553",
+		},
+		{
+			dbname:     "database_name",
+			additional: []string{"schema_name_2"},
+			expectedID: "235207038",
+		},
+		{
+			dbname:     "database_name",
+			additional: []string{"schema_name_1", "schema_name_2"},
+			expectedID: "3743845847",
+		},
+	}
+
+	for _, tc := range testcases {
+		t.Run(tc.dbname, func(t *testing.T) {
+			if id, err := GenerateAdvisoryLockId(tc.dbname, tc.additional...); err == nil {
+				if id != tc.expectedID {
+					t.Error("Generated incorrect ID:", id, "!=", tc.expectedID)
+				}
+			} else {
+				if tc.expectedID != "" {
+					t.Error("Got unexpected error:", err)
+				}
+			}
+		})
+	}
+}
+
+func TestCasRestoreOnErr(t *testing.T) {
+	casErr := errors.New("test lock CAS failure")
+	fErr := errors.New("test callback error")
+
+	testcases := []struct {
+		name        string
+		lock        bool
+		from        bool
+		to          bool
+		expectLock  bool
+		fErr        error
+		expectError error
+	}{
+		{
+			name:        "Test positive CAS lock",
+			lock:        false,
+			from:        false,
+			to:          true,
+			expectLock:  true,
+			fErr:        nil,
+			expectError: nil,
+		},
+		{
+			name:        "Test negative CAS lock",
+			lock:        true,
+			from:        false,
+			to:          true,
+			expectLock:  true,
+			fErr:        nil,
+			expectError: casErr,
+		},
+		{
+			name:        "Test negative with callback lock",
+			lock:        false,
+			from:        false,
+			to:          true,
+			expectLock:  false,
+			fErr:        fErr,
+			expectError: fErr,
+		},
+	}
+
+	for _, tc := range testcases {
+		t.Run(tc.name, func(t *testing.T) {
+			var lock atomic.Bool
+			lock.Store(tc.lock)
+			if err := CasRestoreOnErr(&lock, tc.from, tc.to, casErr, func() error {
+				return tc.fErr
+			}); err != tc.expectError {
+				t.Error("Incorrect error value returned")
+			}
+
+			if lock.Load() != tc.expectLock {
+				t.Error("Incorrect state of lock")
+			}
+		})
+	}
+}