diff options
| -rw-r--r-- | database_test.go | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/database_test.go b/database_test.go index d285c95..26e4e3d 100644 --- a/database_test.go +++ b/database_test.go @@ -68,6 +68,31 @@ func TestMigrate(t *testing.T) { } } +func TestUserVersion(t *testing.T) { + db, _ := emptyDB(t) + + expectedVersion := 42 + err := Transaction(db, func(tx *sql.Tx) error { + _, err := tx.Exec(fmt.Sprintf("PRAGMA user_version = %d;", expectedVersion)) + if err != nil { + return err + } + return nil + }) + if err != nil { + t.Fatalf("Failed to set user expectedVersion: %v", err) + } + + version, err := UserVersion(db) + if err != nil { + t.Fatalf("Could not execute function to test: %v", err) + } + + if version != expectedVersion { + t.Fatalf("Expected user_version %d but got %d.", expectedVersion, version) + } +} + func TestMigrateToLatest(t *testing.T) { db, _ := emptyDB(t) |