When getting tables to truncate, panic on unknown version

riverdriver/river_driver_interface.go

@@ -15,6 +15,7 @@ package riverdriver
 import (
 	"context"
 	"errors"
+	"fmt"
 	"io/fs"
 	"time"
 
@@ -865,15 +866,17 @@ type TableTruncateParams struct {
 //
 // API is not stable. DO NOT USE.
 func MigrationLineMainTruncateTables(version int) []string {
+	// 0 value must be handled and should always point to latest migration version
 	switch version {
 	case 1:
 		return nil // don't truncate `river_migrate`
 	case 2, 3:
 		return []string{"river_job", "river_leader"}
 	case 4:
 		return []string{"river_job", "river_leader", "river_queue"}
+	case 0, 5, 6:
+		return []string{"river_job", "river_leader", "river_queue", "river_client", "river_client_queue"}
 	}
 
-	// 0 (zero value), 5, 6
-	return []string{"river_job", "river_leader", "river_queue", "river_client", "river_client_queue"}
+	panic(fmt.Sprintf("unrecognized migration version: %d", version))
 }

riverdriver/river_driver_interface_test.go

@@ -250,6 +250,19 @@ func TestJobSetStateSnoozed(t *testing.T) { //nolint:dupl
 	})
 }
 
+func TestMigrationLineMainTruncateTables(t *testing.T) {
+	t.Parallel()
+
+	t.Run("ZeroValueDoesNotPanic", func(t *testing.T) {
+		t.Parallel()
+
+		require.NotPanics(t, func() {
+			tables := MigrationLineMainTruncateTables(0)
+			require.NotEmpty(t, tables)
+		})
+	})
+}
+
 func TestJobSetStateSnoozedAvailable(t *testing.T) { //nolint:dupl
 	t.Parallel()