When getting tables to truncate, panic on unknown version (#1198)

This one's related to code review feedback here [1]. When getting a list
of tables that need to be truncated for the `TestSchema` helpers, it was
possible to receive migration versions that didn't exist and the
function would tolerate it, returning the latest known list of tables.
This often works, but even where it does there's some risk of a new
table coming in that doesn't end up getting cleaned up properly.

Here, panic when receiving an unknown version number so that the problem
can be corrected immediately. The panic may seem undesirable, but these
functions are only used in tests, so it should be fine.

[1] riverqueue/riverpro#281 (comment)

Author: brandur

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()