forked from Shiloh/githaven
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
|
// Copyright 2024 The Gitea Authors. All rights reserved.
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
package v1_22 //nolint
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"code.gitea.io/gitea/models/migrations/base"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_UpdateBadgeColName(t *testing.T) {
|
||
|
|
type Badge struct {
|
||
|
|
ID int64 `xorm:"pk autoincr"`
|
||
|
|
Description string
|
||
|
|
ImageURL string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Prepare and load the testing database
|
||
|
|
x, deferable := base.PrepareTestEnv(t, 0, new(BadgeUnique), new(Badge))
|
||
|
|
defer deferable()
|
||
|
|
if x == nil || t.Failed() {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
oldBadges := []Badge{
|
||
|
|
{ID: 1, Description: "Test Badge 1", ImageURL: "https://example.com/badge1.png"},
|
||
|
|
{ID: 2, Description: "Test Badge 2", ImageURL: "https://example.com/badge2.png"},
|
||
|
|
{ID: 3, Description: "Test Badge 3", ImageURL: "https://example.com/badge3.png"},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, badge := range oldBadges {
|
||
|
|
_, err := x.Insert(&badge)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := UseSlugInsteadOfIDForBadges(x); err != nil {
|
||
|
|
assert.NoError(t, err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
got := []BadgeUnique{}
|
||
|
|
if err := x.Table("badge").Asc("id").Find(&got); !assert.NoError(t, err) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
for i, e := range oldBadges {
|
||
|
|
got := got[i]
|
||
|
|
assert.Equal(t, e.ID, got.ID)
|
||
|
|
assert.Equal(t, fmt.Sprintf("%d", e.ID), got.Slug)
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO: check if badges have been updated
|
||
|
|
}
|