forgejo/models/models_test.go

// 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"`)
}