mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
9a07b4d089
2978 files. The fork is 2026-07-15 (rebrand: gitea.dev -> github.com/hanzoai/git), so 2026 is the year, and "Copyright 2026 Hanzo AI, Inc." was already the form used by the handful of files that had ours. Shape is now: // Copyright 2026 Hanzo AI, Inc. All rights reserved. // Copyright 2015 The Gogs Authors. All rights reserved. // Copyright 2019 The Gitea Authors. All rights reserved. // SPDX-License-Identifier: MIT Ours on top, theirs retained verbatim. Retained because MIT requires it, and separately because they are true — those authors did write that code. Not renamed to "The Git Authors": Git is a different project under a different license, and putting its name here would attribute this code to people who have nothing to do with it. Adding our line is the fix; overwriting theirs would have been both a licence violation and a false statement. gofmt still reports import-order drift in some of these files. It predates this change (verified against a stash) and is left alone rather than burying a one-line-per-file legal change under 3000 files of unrelated reordering. Build clean. Co-authored-by: Hanzo Dev <[email protected]>
137 lines
4.1 KiB
Go
137 lines
4.1 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2022 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package db_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/git/models/db"
|
|
"github.com/hanzoai/git/models/unittest"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestInTransaction(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
assert.False(t, db.InTransaction(t.Context()))
|
|
assert.NoError(t, db.WithTx(t.Context(), func(ctx context.Context) error {
|
|
assert.True(t, db.InTransaction(ctx))
|
|
return nil
|
|
}))
|
|
|
|
ctx, committer, err := db.TxContext(t.Context())
|
|
assert.NoError(t, err)
|
|
defer committer.Close()
|
|
assert.True(t, db.InTransaction(ctx))
|
|
assert.NoError(t, db.WithTx(ctx, func(ctx context.Context) error {
|
|
assert.True(t, db.InTransaction(ctx))
|
|
return nil
|
|
}))
|
|
}
|
|
|
|
func TestTxContext(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
{ // create new transaction
|
|
ctx, committer, err := db.TxContext(t.Context())
|
|
assert.NoError(t, err)
|
|
assert.True(t, db.InTransaction(ctx))
|
|
assert.NoError(t, committer.Commit())
|
|
}
|
|
|
|
{ // reuse the transaction created by TxContext and commit it
|
|
ctx, committer, err := db.TxContext(t.Context())
|
|
engine := db.GetEngine(ctx)
|
|
assert.NoError(t, err)
|
|
assert.True(t, db.InTransaction(ctx))
|
|
{
|
|
ctx, committer, err := db.TxContext(ctx)
|
|
assert.NoError(t, err)
|
|
assert.True(t, db.InTransaction(ctx))
|
|
assert.Equal(t, engine, db.GetEngine(ctx))
|
|
assert.NoError(t, committer.Commit())
|
|
}
|
|
assert.NoError(t, committer.Commit())
|
|
}
|
|
|
|
{ // reuse the transaction created by TxContext and close it
|
|
ctx, committer, err := db.TxContext(t.Context())
|
|
engine := db.GetEngine(ctx)
|
|
assert.NoError(t, err)
|
|
assert.True(t, db.InTransaction(ctx))
|
|
{
|
|
ctx, committer, err := db.TxContext(ctx)
|
|
assert.NoError(t, err)
|
|
assert.True(t, db.InTransaction(ctx))
|
|
assert.Equal(t, engine, db.GetEngine(ctx))
|
|
assert.NoError(t, committer.Close())
|
|
}
|
|
assert.NoError(t, committer.Close())
|
|
}
|
|
|
|
{ // reuse the transaction created by WithTx
|
|
assert.NoError(t, db.WithTx(t.Context(), func(ctx context.Context) error {
|
|
assert.True(t, db.InTransaction(ctx))
|
|
{
|
|
ctx, committer, err := db.TxContext(ctx)
|
|
assert.NoError(t, err)
|
|
assert.True(t, db.InTransaction(ctx))
|
|
assert.NoError(t, committer.Commit())
|
|
}
|
|
return nil
|
|
}))
|
|
}
|
|
}
|
|
|
|
func TestContextSafety(t *testing.T) {
|
|
type TestModel1 struct {
|
|
ID int64
|
|
}
|
|
type TestModel2 struct {
|
|
ID int64
|
|
}
|
|
assert.NoError(t, unittest.GetXORMEngine().Sync(&TestModel1{}, &TestModel2{}))
|
|
assert.NoError(t, db.TruncateBeans(t.Context(), &TestModel1{}, &TestModel2{}))
|
|
testCount := 10
|
|
for i := 1; i <= testCount; i++ {
|
|
assert.NoError(t, db.Insert(t.Context(), &TestModel1{ID: int64(i)}))
|
|
assert.NoError(t, db.Insert(t.Context(), &TestModel2{ID: int64(-i)}))
|
|
}
|
|
|
|
t.Run("Show-XORM-Bug", func(t *testing.T) {
|
|
actualCount := 0
|
|
// here: db.GetEngine(t.Context()) is a new *Session created from *Engine
|
|
_ = db.WithTx(t.Context(), func(ctx context.Context) error {
|
|
_ = db.GetEngine(ctx).Iterate(&TestModel1{}, func(i int, bean any) error {
|
|
// here: db.GetEngine(ctx) is always the unclosed "Iterate" *Session with autoResetStatement=false,
|
|
// and the internal states (including "cond" and others) are always there and not be reset in this callback.
|
|
m1 := bean.(*TestModel1)
|
|
assert.EqualValues(t, i+1, m1.ID)
|
|
|
|
// here: XORM bug, it fails because the SQL becomes "WHERE id=-1", "WHERE id=-1 AND id=-2", "WHERE id=-1 AND id=-2 AND id=-3" ...
|
|
// and it conflicts with the "Iterate"'s internal states.
|
|
// has, err := db.GetEngine(ctx).Get(&TestModel2{ID: -m1.ID})
|
|
|
|
actualCount++
|
|
return nil
|
|
})
|
|
return nil
|
|
})
|
|
assert.Equal(t, testCount, actualCount)
|
|
})
|
|
|
|
t.Run("DenyBadUsage", func(t *testing.T) {
|
|
assert.PanicsWithError(t, "using session context in an iterator would cause corrupted results", func() {
|
|
_ = db.WithTx(t.Context(), func(ctx context.Context) error {
|
|
return db.GetEngine(ctx).Iterate(&TestModel1{}, func(i int, bean any) error {
|
|
_ = db.GetEngine(ctx)
|
|
return nil
|
|
})
|
|
})
|
|
})
|
|
})
|
|
}
|