Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.4 KiB
60 lines
1.4 KiB
// Copyright 2019 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. |
|
|
|
package models |
|
|
|
import ( |
|
"io/ioutil" |
|
"os" |
|
"path/filepath" |
|
"strings" |
|
"testing" |
|
|
|
"code.gitea.io/gitea/modules/auth/oauth2" |
|
"code.gitea.io/gitea/modules/setting" |
|
"xorm.io/xorm/schemas" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestDumpDatabase(t *testing.T) { |
|
assert.NoError(t, PrepareTestDatabase()) |
|
|
|
dir, err := ioutil.TempDir(os.TempDir(), "dump") |
|
assert.NoError(t, err) |
|
|
|
type Version struct { |
|
ID int64 `xorm:"pk autoincr"` |
|
Version int64 |
|
} |
|
assert.NoError(t, x.Sync2(new(Version))) |
|
|
|
for _, dbName := range setting.SupportedDatabases { |
|
dbType := setting.GetDBTypeByName(dbName) |
|
assert.NoError(t, DumpDatabase(filepath.Join(dir, dbType+".sql"), dbType)) |
|
} |
|
} |
|
|
|
func TestDumpLoginSource(t *testing.T) { |
|
assert.NoError(t, PrepareTestDatabase()) |
|
|
|
loginSourceSchema, err := x.TableInfo(new(LoginSource)) |
|
assert.NoError(t, err) |
|
|
|
CreateLoginSource(&LoginSource{ |
|
Type: LoginOAuth2, |
|
Name: "TestSource", |
|
IsActived: false, |
|
Cfg: &OAuth2Config{ |
|
Provider: "TestSourceProvider", |
|
CustomURLMapping: &oauth2.CustomURLMapping{}, |
|
}, |
|
}) |
|
|
|
sb := new(strings.Builder) |
|
|
|
x.DumpTables([]*schemas.Table{loginSourceSchema}, sb) |
|
|
|
assert.Contains(t, sb.String(), `"Provider":"TestSourceProvider"`) |
|
}
|
|
|