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]>
129 lines
3.3 KiB
Go
129 lines
3.3 KiB
Go
// Copyright 2026 Hanzo AI, Inc. 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)
|
|
}
|
|
}
|
|
}
|