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]>
130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package v1_14
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hanzoai/git/models/migrations/migrationtest"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_RemoveInvalidLabels(t *testing.T) {
|
|
// Models used by the migration
|
|
type Comment struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
Type int `xorm:"INDEX"`
|
|
IssueID int64 `xorm:"INDEX"`
|
|
LabelID int64
|
|
ShouldRemain bool // <- Flag for testing the migration
|
|
}
|
|
|
|
type Issue struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
RepoID int64 `xorm:"INDEX UNIQUE(repo_index)"`
|
|
Index int64 `xorm:"UNIQUE(repo_index)"` // Index in one repository.
|
|
}
|
|
|
|
type Repository struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
OwnerID int64 `xorm:"UNIQUE(s) index"`
|
|
LowerName string `xorm:"UNIQUE(s) INDEX NOT NULL"`
|
|
}
|
|
|
|
type Label struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
RepoID int64 `xorm:"INDEX"`
|
|
OrgID int64 `xorm:"INDEX"`
|
|
}
|
|
|
|
type IssueLabel struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
IssueID int64 `xorm:"UNIQUE(s)"`
|
|
LabelID int64 `xorm:"UNIQUE(s)"`
|
|
ShouldRemain bool // <- Flag for testing the migration
|
|
}
|
|
|
|
// load and prepare the test database
|
|
x, deferable := migrationtest.PrepareTestEnv(t, 0, new(Comment), new(Issue), new(Repository), new(IssueLabel), new(Label))
|
|
if x == nil || t.Failed() {
|
|
defer deferable()
|
|
return
|
|
}
|
|
defer deferable()
|
|
|
|
var issueLabels []*IssueLabel
|
|
ilPreMigration := map[int64]*IssueLabel{}
|
|
ilPostMigration := map[int64]*IssueLabel{}
|
|
|
|
var comments []*Comment
|
|
comPreMigration := map[int64]*Comment{}
|
|
comPostMigration := map[int64]*Comment{}
|
|
|
|
// Get pre migration values
|
|
if err := x.Find(&issueLabels); err != nil {
|
|
t.Errorf("Unable to find issueLabels: %v", err)
|
|
return
|
|
}
|
|
for _, issueLabel := range issueLabels {
|
|
ilPreMigration[issueLabel.ID] = issueLabel
|
|
}
|
|
if err := x.Find(&comments); err != nil {
|
|
t.Errorf("Unable to find comments: %v", err)
|
|
return
|
|
}
|
|
for _, comment := range comments {
|
|
comPreMigration[comment.ID] = comment
|
|
}
|
|
|
|
// Run the migration
|
|
if err := RemoveInvalidLabels(x); err != nil {
|
|
t.Errorf("unable to RemoveInvalidLabels: %v", err)
|
|
}
|
|
|
|
// Get the post migration values
|
|
issueLabels = issueLabels[:0]
|
|
if err := x.Find(&issueLabels); err != nil {
|
|
t.Errorf("Unable to find issueLabels: %v", err)
|
|
return
|
|
}
|
|
for _, issueLabel := range issueLabels {
|
|
ilPostMigration[issueLabel.ID] = issueLabel
|
|
}
|
|
comments = comments[:0]
|
|
if err := x.Find(&comments); err != nil {
|
|
t.Errorf("Unable to find comments: %v", err)
|
|
return
|
|
}
|
|
for _, comment := range comments {
|
|
comPostMigration[comment.ID] = comment
|
|
}
|
|
|
|
// Finally test results of the migration
|
|
for id, comment := range comPreMigration {
|
|
post, ok := comPostMigration[id]
|
|
if ok {
|
|
if !comment.ShouldRemain {
|
|
t.Errorf("Comment[%d] remained but should have been deleted", id)
|
|
}
|
|
assert.Equal(t, comment, post)
|
|
} else if comment.ShouldRemain {
|
|
t.Errorf("Comment[%d] was deleted but should have remained", id)
|
|
}
|
|
}
|
|
|
|
for id, il := range ilPreMigration {
|
|
post, ok := ilPostMigration[id]
|
|
if ok {
|
|
if !il.ShouldRemain {
|
|
t.Errorf("IssueLabel[%d] remained but should have been deleted", id)
|
|
}
|
|
assert.Equal(t, il, post)
|
|
} else if il.ShouldRemain {
|
|
t.Errorf("IssueLabel[%d] was deleted but should have remained", id)
|
|
}
|
|
}
|
|
}
|