2022-10-24 16:59:24 +09:00
|
|
|
// Copyright 2022 The Gitea Authors. All rights reserved.
|
|
|
|
// Use of this source code is governed by a MIT-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
2022-11-02 17:54:36 +09:00
|
|
|
package v1_18 //nolint
|
2022-10-24 16:59:24 +09:00
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
2022-11-02 17:54:36 +09:00
|
|
|
"code.gitea.io/gitea/models/migrations/base"
|
|
|
|
|
2022-10-24 16:59:24 +09:00
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
)
|
|
|
|
|
2022-11-02 17:54:36 +09:00
|
|
|
func Test_AddConfidentialClientColumnToOAuth2ApplicationTable(t *testing.T) {
|
2022-10-24 16:59:24 +09:00
|
|
|
// premigration
|
|
|
|
type OAuth2Application struct {
|
|
|
|
ID int64
|
|
|
|
}
|
|
|
|
|
|
|
|
// Prepare and load the testing database
|
2022-11-02 17:54:36 +09:00
|
|
|
x, deferable := base.PrepareTestEnv(t, 0, new(OAuth2Application))
|
2022-10-24 16:59:24 +09:00
|
|
|
defer deferable()
|
|
|
|
if x == nil || t.Failed() {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2022-11-02 17:54:36 +09:00
|
|
|
if err := AddConfidentialClientColumnToOAuth2ApplicationTable(x); err != nil {
|
2022-10-24 16:59:24 +09:00
|
|
|
assert.NoError(t, err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// postmigration
|
|
|
|
type ExpectedOAuth2Application struct {
|
|
|
|
ID int64
|
|
|
|
ConfidentialClient bool
|
|
|
|
}
|
|
|
|
|
|
|
|
got := []ExpectedOAuth2Application{}
|
|
|
|
if err := x.Table("o_auth2_application").Select("id, confidential_client").Find(&got); !assert.NoError(t, err) {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
assert.NotEmpty(t, got)
|
|
|
|
for _, e := range got {
|
|
|
|
assert.True(t, e.ConfidentialClient)
|
|
|
|
}
|
|
|
|
}
|