mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
81 lines
2.8 KiB
Go
81 lines
2.8 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package v1_27
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hanzoai/git/models/db"
|
|
"github.com/hanzoai/git/models/migrations/migrationtest"
|
|
"github.com/hanzoai/git/modules/setting"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type externalLoginUserBeforeDateTimeMigration struct {
|
|
ExternalID string `xorm:"pk NOT NULL"`
|
|
LoginSourceID int64 `xorm:"pk NOT NULL"`
|
|
ExpiresAt time.Time // sync creates DATETIME2; downgraded to legacy DATETIME via raw SQL below
|
|
}
|
|
|
|
func (externalLoginUserBeforeDateTimeMigration) TableName() string {
|
|
return "external_login_user"
|
|
}
|
|
|
|
type lfsLockBeforeDateTimeMigration struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
Created time.Time `xorm:"created"`
|
|
}
|
|
|
|
func (lfsLockBeforeDateTimeMigration) TableName() string {
|
|
return "lfs_lock"
|
|
}
|
|
|
|
func Test_FixLegacyMSSQLDateTimeColumns(t *testing.T) {
|
|
if !setting.Database.Type.IsMSSQL() {
|
|
t.Skip("Only MSSQL needs to convert the legacy locale-dependent DATETIME columns")
|
|
}
|
|
|
|
x, deferrable := migrationtest.PrepareTestEnv(t, 0,
|
|
new(externalLoginUserBeforeDateTimeMigration),
|
|
new(lfsLockBeforeDateTimeMigration),
|
|
)
|
|
defer deferrable()
|
|
|
|
// Force the legacy DATETIME column type that old upstream versions created.
|
|
_, err := x.Exec("ALTER TABLE [external_login_user] ALTER COLUMN [expires_at] DATETIME")
|
|
require.NoError(t, err)
|
|
_, err = x.Exec("ALTER TABLE [lfs_lock] ALTER COLUMN [created] DATETIME")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "datetime", mssqlColumnType(t, x, "external_login_user", "expires_at"))
|
|
require.Equal(t, "datetime", mssqlColumnType(t, x, "lfs_lock", "created"))
|
|
|
|
require.NoError(t, FixLegacyMSSQLDateTimeColumns(x))
|
|
require.NoError(t, FixLegacyMSSQLDateTimeColumns(x)) // idempotent
|
|
|
|
require.Equal(t, "datetime2", mssqlColumnType(t, x, "external_login_user", "expires_at"))
|
|
require.Equal(t, "datetime2", mssqlColumnType(t, x, "lfs_lock", "created"))
|
|
|
|
// Inserting an ISO-formatted datetime must succeed even under a non-English
|
|
// locale, which is the failure the legacy DATETIME columns produced. The
|
|
// SET LANGUAGE and INSERT run in one Exec so they share a single connection.
|
|
_, err = x.Exec("SET LANGUAGE German; " +
|
|
"INSERT INTO [external_login_user] ([external_id], [login_source_id], [expires_at]) " +
|
|
"VALUES ('ext-id', 1, '2026-06-25 11:58:39')")
|
|
require.NoError(t, err)
|
|
_, err = x.Exec("SET LANGUAGE German; " +
|
|
"INSERT INTO [lfs_lock] ([created]) VALUES ('2026-06-25 11:58:39')")
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func mssqlColumnType(t *testing.T, x db.EngineMigration, table, column string) string {
|
|
t.Helper()
|
|
var dataType string
|
|
has, err := x.SQL("SELECT DATA_TYPE FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = ? AND COLUMN_NAME = ?", table, column).Get(&dataType)
|
|
require.NoError(t, err)
|
|
require.True(t, has)
|
|
return dataType
|
|
}
|