TsDB2/migrate/value_test.go

package migrate

import (
	"TsDB2/DB/core/types"
	"math"
	"testing"
)

// TestRoundFloatToIntBoundary 钉住上界的写法。
//
// float64(math.MaxInt64) 存不下 2^63-1, 会被舍入成 2^63 —— 于是 "f > MaxInt64"
// 这个看着像校验的守卫, 恰好放过了唯一越界的那一个值, 而 int64(2^63) 是越界转换
// (结果未定义, amd64 上回绕成 MinInt64): 一个超范围的值静默变成反数存进去。
// 下界不受影响: MinInt64 = -2^63 在 float64 里可以精确表示。
func TestRoundFloatToIntBoundary(t *testing.T) {
	if got, err := roundFloatToInt(0); err != nil || got != 0 {
		t.Errorf("0 应当原样通过, 实际 (%d, %v)", got, err)
	}
	if _, err := roundFloatToInt(float64(math.MaxInt32)); err != nil {
		t.Errorf("MaxInt32 远在范围内, 不该报错: %v", err)
	}
	// 下界是可以精确表示的那个值, 不该被误伤
	if got, err := roundFloatToInt(math.MinInt64); err != nil || got != math.MinInt64 {
		t.Errorf("MinInt64 应当原样通过, 实际 (%d, %v)", got, err)
	}

	// 注意别写 math.MinInt64 - 1 当越界样例: 那个数学值在 float64 里根本没有对应的
	// 格点, 会被舍入回 -2^63 这个合法值。上界那边正好相反 —— MaxInt64 舍入到 2^63,
	// 恰好越过边界, 所以 >= 才是对的。
	for _, f := range []float64{
		math.MaxInt64, // 2^63: 越界, 但 float64(MaxInt64) 也正好等于它
		math.MaxInt64 * 2,
		math.Pow(2, 63) * 4,
		math.Inf(1),
		math.Inf(-1),
		math.NaN(),
	} {
		if got, err := roundFloatToInt(f); err == nil {
			t.Errorf("%v 超出 int64, 应当报错, 实际返回 %d", f, got)
		}
	}
}

// TestCoerceForSQL8ByteByDialect uint64 绑定的 Go 类型必须跟目标列的【声明】一致,
// 而两个方言的声明本就不同 (见 sqlTypeName):
//
//   - MySQL 声明 BIGINT UNSIGNED → 绑 uint64。以前一律绑 int64(u), 高位为 1 的值
//     (包括 VRTX 从 int64 负值搬过来的位模式) 在严格模式下会被判成
//     "Out of range value" 直接报错, 非严格模式则被静默改写。
//   - SQLite 只有有符号的 INTEGER → 绑 int64, 位模式原样存。
//
// 共同的保证是【位模式不变】; 两种绑定在各自的列里都必须无损。
func TestCoerceForSQL8ByteByDialect(t *testing.T) {
	col := Column{Name: "u", Type: types.FieldType_Uint64}

	cases := []struct {
		name       string
		in         any
		wantMySQL  any
		wantSQLite any
	}{
		{"uint64 上界", uint64(math.MaxUint64), uint64(math.MaxUint64), int64(-1)},
		{"int64 负数 (位模式)", int64(-1), uint64(math.MaxUint64), int64(-1)},
		{"普通值", int64(42), uint64(42), int64(42)},
		{"2^63 (无符号档才装得下)", uint64(math.MaxInt64) + 1, uint64(math.MaxInt64) + 1, int64(math.MinInt64)},
	}
	for _, c := range cases {
		gotMySQL, err := coerceForSQL(col, c.in, true)
		if err != nil {
			t.Errorf("%s: MySQL 侧不该报错: %v", c.name, err)
		} else if gotMySQL != c.wantMySQL {
			t.Errorf("%s: MySQL 侧 = %v (%T), 期望 %v (%T)",
				c.name, gotMySQL, gotMySQL, c.wantMySQL, c.wantMySQL)
		}

		gotSQLite, err := coerceForSQL(col, c.in, false)
		if err != nil {
			t.Errorf("%s: SQLite 侧不该报错: %v", c.name, err)
		} else if gotSQLite != c.wantSQLite {
			t.Errorf("%s: SQLite 侧 = %v (%T), 期望 %v (%T)",
				c.name, gotSQLite, gotSQLite, c.wantSQLite, c.wantSQLite)
		}
	}

	// 时间列走的还是"还原成时间文本"那条路, 与方言无关 —— 别被上面的方言分支带走。
	timeCol := Column{Name: "t", Type: types.FieldType_Uint64, TimeKind: "datetime"}
	for _, mysql := range []bool{true, false} {
		v, err := coerceForSQL(timeCol, int64(0), mysql)
		if err != nil {
			t.Fatalf("时间列 (mysql=%v) 不该报错: %v", mysql, err)
		}
		if _, ok := v.(string); !ok {
			t.Errorf("时间列 (mysql=%v) 应当还原成时间文本, 实际 %T", mysql, v)
		}
	}
}

// TestCoerceForSQLUnsignedSmall 2 / 4 字节无符号档出库时必须给【无符号数值】。
//
// 与 8 字节档同一个道理 (见 TestCoerceForSQL8ByteByDialect), 但 16/32 位不需要按方言分岔:
// 目标列在 MySQL 侧声明成 SMALLINT UNSIGNED / INT UNSIGNED, SQLite 侧是 INTEGER ——
// 0..65535 与 0..4294967295 两边都整个装得下, 给无符号值两个方言都对。
// 反过来把位模式当 int64 原样送出去 (0x8278 读出来是 -32136), MySQL 会判 Out of range:
// 严格模式直接报错, 非严格模式截成 0 —— 静默改数据。
func TestCoerceForSQLUnsignedSmall(t *testing.T) {
	cases := []struct {
		name string
		ft   types.FieldType
		in   any
		want int64
	}{
		{"uint16 上半区位模式", types.FieldType_Uint16, int16(-32136), 33400}, // 0x8278
		{"uint16 普通值", types.FieldType_Uint16, int16(3306), 3306},
		{"uint16 上界位模式", types.FieldType_Uint16, int16(-1), 65535},
		{"uint32 上半区位模式", types.FieldType_Uint32, int32(-1294967296), 3000000000},
		{"uint32 上界位模式", types.FieldType_Uint32, int32(-1), 4294967295},
	}
	for _, c := range cases {
		col := Column{Name: "u", Type: c.ft}
		for _, mysql := range []bool{true, false} {
			got, err := coerceForSQL(col, c.in, mysql)
			if err != nil {
				t.Errorf("%s (mysql=%v): 不该报错: %v", c.name, mysql, err)
				continue
			}
			if got != c.want {
				t.Errorf("%s (mysql=%v): = %v (%T), 期望 %d", c.name, mysql, got, got, c.want)
			}
		}
	}
}

// TestCoerceForVRTXUnsignedSmall 2 / 4 字节无符号档"装得下"的判据是【位模式】, 不是
// int16 / int32 的正数区。
//
// 真实踩到: MySQL 的 SMALLINT UNSIGNED 存了端口号 33400, 迁进 VRTX 的 uint16 列时被判
// "值 33400 超出 int16" —— 可 33400 只占 16 位 (0x8278), 位模式完全装得下。按有符号区间判
// 等于给无符号列凭空加了一条"上界只到 32767"的约束, 把 32768..65535 这半个无符号区整片
// 判成脏数据; 而 -32136 只是同一个位模式的另一种读法, 两个方向都得收。
func TestCoerceForVRTXUnsignedSmall(t *testing.T) {
	const col = "port"
	cases := []struct {
		name string
		ft   types.FieldType
		in   any
		want any
	}{
		// 无符号上半区: 合法, 落成负的位模式载体 (读取侧 DecodeValue 也回 int16/int32)
		{"uint16 上半区", types.FieldType_Uint16, int64(33400), int16(-32136)},
		{"uint16 上界", types.FieldType_Uint16, int64(math.MaxUint16), int16(-1)},
		{"uint16 负位模式 (回读 / SQLite INTEGER)", types.FieldType_Uint16, int16(-1), int16(-1)},
		{"uint16 普通值", types.FieldType_Uint16, int64(3306), int16(3306)},
		{"uint32 上半区", types.FieldType_Uint32, int64(3000000000), int32(-1294967296)},
		{"uint32 上界", types.FieldType_Uint32, int64(math.MaxUint32), int32(-1)},
		{"uint32 普通值", types.FieldType_Uint32, int64(42), int32(42)},
	}
	for _, c := range cases {
		got, err := coerceForVRTX(c.ft, 0, c.in, col, true, false)
		if err != nil {
			t.Errorf("%s: 不该报错: %v", c.name, err)
			continue
		}
		if got != c.want {
			t.Errorf("%s: = %v (%T), 期望 %v (%T)", c.name, got, got, c.want, c.want)
		}
	}

	// 放不下位模式才是真放不下 —— 这条闸门不能因为放宽上界就整个拆掉。
	for _, c := range []struct {
		name string
		ft   types.FieldType
		in   any
	}{
		{"uint16 超出 16 位", types.FieldType_Uint16, int64(math.MaxUint16) + 1},
		{"uint16 低于 -32768", types.FieldType_Uint16, int64(math.MinInt16) - 1},
		{"uint32 超出 32 位", types.FieldType_Uint32, int64(math.MaxUint32) + 1},
	} {
		if got, err := coerceForVRTX(c.ft, 0, c.in, col, true, false); err == nil {
			t.Errorf("%s: 应当报错, 实际返回 %v", c.name, got)
		}
	}
}