mirror of
https://github.com/coder/coder.git
synced 2025-07-13 21:36:50 +00:00
feat: Add reset-password command (#1380)
* allow non-destructively checking if database needs to be migrated * feat: Add reset-password command * fix linter errors * clean up reset-password usage prompt * Add confirmation to reset-password command * Ping database before checking migration, to improve error message
This commit is contained in:
@ -4,8 +4,11 @@ package database_test
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/golang-migrate/migrate/v4/source"
|
||||
"github.com/golang-migrate/migrate/v4/source/stub"
|
||||
"github.com/stretchr/testify/require"
|
||||
"go.uber.org/goleak"
|
||||
|
||||
@ -75,3 +78,54 @@ func testSQLDB(t testing.TB) *sql.DB {
|
||||
|
||||
return db
|
||||
}
|
||||
|
||||
// paralleltest linter doesn't correctly handle table-driven tests (https://github.com/kunwardeep/paralleltest/issues/8)
|
||||
// nolint:paralleltest
|
||||
func TestCheckLatestVersion(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
type test struct {
|
||||
currentVersion uint
|
||||
existingVersions []uint
|
||||
expectedResult string
|
||||
}
|
||||
|
||||
tests := []test{
|
||||
// successful cases
|
||||
{1, []uint{1}, ""},
|
||||
{3, []uint{1, 2, 3}, ""},
|
||||
{3, []uint{1, 3}, ""},
|
||||
|
||||
// failure cases
|
||||
{1, []uint{1, 2}, "current version is 1, but later version 2 exists"},
|
||||
{2, []uint{1, 2, 3}, "current version is 2, but later version 3 exists"},
|
||||
{4, []uint{1, 2, 3}, "get previous migration: prev for version 4 : file does not exist"},
|
||||
{4, []uint{1, 2, 3, 5}, "get previous migration: prev for version 4 : file does not exist"},
|
||||
}
|
||||
|
||||
for i, tc := range tests {
|
||||
i, tc := i, tc
|
||||
t.Run(fmt.Sprintf("entry %d", i), func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
driver, _ := stub.WithInstance(nil, &stub.Config{})
|
||||
stub, ok := driver.(*stub.Stub)
|
||||
require.True(t, ok)
|
||||
for _, version := range tc.existingVersions {
|
||||
stub.Migrations.Append(&source.Migration{
|
||||
Version: version,
|
||||
Identifier: "",
|
||||
Direction: source.Up,
|
||||
Raw: "",
|
||||
})
|
||||
}
|
||||
|
||||
err := database.CheckLatestVersion(driver, tc.currentVersion)
|
||||
var errMessage string
|
||||
if err != nil {
|
||||
errMessage = err.Error()
|
||||
}
|
||||
require.Equal(t, tc.expectedResult, errMessage)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user