diff options
author | zeripath <art27@cantab.net> | 2021-08-28 04:25:27 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-08-28 04:25:27 +0100 |
commit | 19049413827b130809ffe249900a5c89ef8a90df (patch) | |
tree | 4821622f10fbb25f82910278d8c8d3b486ab8500 | |
parent | 88abb0dc8a388795d5a1fa8bd1e4f0b7d207a0f7 (diff) | |
download | gitea-19049413827b130809ffe249900a5c89ef8a90df.tar.gz gitea-19049413827b130809ffe249900a5c89ef8a90df.zip |
Add test to ensure that dumping of login sources remains correct (#16847)
#16831 has occurred because of a missed regression. This PR adds a simple test to
try to prevent this occuring again.
Signed-off-by: Andrew Thornton <art27@cantab.net>
-rw-r--r-- | models/models_test.go | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/models/models_test.go b/models/models_test.go index 9793394e0b..c75c299e4f 100644 --- a/models/models_test.go +++ b/models/models_test.go @@ -5,12 +5,15 @@ package models import ( + "encoding/json" "io/ioutil" "os" "path/filepath" + "strings" "testing" "code.gitea.io/gitea/modules/setting" + "xorm.io/xorm/schemas" "github.com/stretchr/testify/assert" ) @@ -32,3 +35,46 @@ func TestDumpDatabase(t *testing.T) { assert.NoError(t, DumpDatabase(filepath.Join(dir, dbType+".sql"), dbType)) } } + +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) +} + +func TestDumpLoginSource(t *testing.T) { + assert.NoError(t, PrepareTestDatabase()) + + loginSourceSchema, err := x.TableInfo(new(LoginSource)) + assert.NoError(t, err) + + RegisterLoginTypeConfig(LoginOAuth2, new(TestSource)) + + CreateLoginSource(&LoginSource{ + Type: LoginOAuth2, + Name: "TestSource", + IsActive: false, + Cfg: &TestSource{ + Provider: "ConvertibleSourceName", + ClientID: "42", + }, + }) + + sb := new(strings.Builder) + + x.DumpTables([]*schemas.Table{loginSourceSchema}, sb) + + assert.Contains(t, sb.String(), `"Provider":"ConvertibleSourceName"`) +} |