Fix: failed unit test due to import cycle
This commit is contained in:
parent
fce38209bc
commit
4dbe867020
1 changed files with 0 additions and 21 deletions
|
@ -2,37 +2,16 @@ package invoker
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"database/sql"
|
|
||||||
"github.com/DATA-DOG/go-sqlmock"
|
|
||||||
model "github.com/cloudreve/Cloudreve/v3/models"
|
|
||||||
"github.com/jinzhu/gorm"
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
var mock sqlmock.Sqlmock
|
|
||||||
var mockDB *gorm.DB
|
|
||||||
|
|
||||||
type TestScript int
|
type TestScript int
|
||||||
|
|
||||||
func (script TestScript) Run(ctx context.Context) {
|
func (script TestScript) Run(ctx context.Context) {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// TestMain 初始化数据库Mock
|
|
||||||
func TestMain(m *testing.M) {
|
|
||||||
var db *sql.DB
|
|
||||||
var err error
|
|
||||||
db, mock, err = sqlmock.New()
|
|
||||||
if err != nil {
|
|
||||||
panic("An error was not expected when opening a stub database connection")
|
|
||||||
}
|
|
||||||
model.DB, _ = gorm.Open("mysql", db)
|
|
||||||
mockDB = model.DB
|
|
||||||
defer db.Close()
|
|
||||||
m.Run()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRunDBScript(t *testing.T) {
|
func TestRunDBScript(t *testing.T) {
|
||||||
asserts := assert.New(t)
|
asserts := assert.New(t)
|
||||||
Register("test", TestScript(0))
|
Register("test", TestScript(0))
|
||||||
|
|
Loading…
Add table
Reference in a new issue