2019-08-24 12:24:45 +03:00
// Copyright 2019 The Gitea Authors. All rights reserved.
2016-08-12 12:56:50 +03:00
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
2022-01-02 16:12:35 +03:00
package auth
2016-08-12 12:56:50 +03:00
import (
2021-08-28 06:25:27 +03:00
"strings"
2016-08-12 12:56:50 +03:00
"testing"
2021-09-19 14:49:59 +03:00
"code.gitea.io/gitea/models/db"
2021-11-12 17:36:47 +03:00
"code.gitea.io/gitea/models/unittest"
2021-09-19 14:49:59 +03:00
"code.gitea.io/gitea/modules/json"
2019-08-24 12:24:45 +03:00
2017-02-08 09:29:07 +03:00
"github.com/stretchr/testify/assert"
2021-09-19 14:49:59 +03:00
"xorm.io/xorm/schemas"
2016-08-12 12:56:50 +03:00
)
2021-08-28 06:25:27 +03:00
type TestSource struct {
Provider string
ClientID string
ClientSecret string
OpenIDConnectAutoDiscoveryURL string
IconURL string
}
// FromDB fills up a LDAPConfig from serialized format.
func ( source * TestSource ) FromDB ( bs [ ] byte ) error {
return json . Unmarshal ( bs , & source )
}
// ToDB exports a LDAPConfig to a serialized format.
func ( source * TestSource ) ToDB ( ) ( [ ] byte , error ) {
return json . Marshal ( source )
}
2022-01-02 16:12:35 +03:00
func TestDumpAuthSource ( t * testing . T ) {
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2021-08-28 06:25:27 +03:00
2022-01-02 16:12:35 +03:00
authSourceSchema , err := db . TableInfo ( new ( Source ) )
2021-08-28 06:25:27 +03:00
assert . NoError ( t , err )
2021-09-24 14:32:56 +03:00
RegisterTypeConfig ( OAuth2 , new ( TestSource ) )
2021-08-28 06:25:27 +03:00
2021-09-24 14:32:56 +03:00
CreateSource ( & Source {
Type : OAuth2 ,
2021-08-28 06:25:27 +03:00
Name : "TestSource" ,
IsActive : false ,
Cfg : & TestSource {
Provider : "ConvertibleSourceName" ,
ClientID : "42" ,
} ,
} )
sb := new ( strings . Builder )
2022-01-02 16:12:35 +03:00
db . DumpTables ( [ ] * schemas . Table { authSourceSchema } , sb )
2021-08-28 06:25:27 +03:00
assert . Contains ( t , sb . String ( ) , ` "Provider":"ConvertibleSourceName" ` )
}