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.
52 lines
1.1 KiB
52 lines
1.1 KiB
5 years ago
|
package task
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"github.com/DATA-DOG/go-sqlmock"
|
||
|
|
model "github.com/HFO4/cloudreve/models"
|
||
|
|
"github.com/HFO4/cloudreve/pkg/cache"
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
var mock sqlmock.Sqlmock
|
||
|
|
|
||
|
|
// 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)
|
||
|
|
defer db.Close()
|
||
|
|
m.Run()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInit(t *testing.T) {
|
||
|
|
asserts := assert.New(t)
|
||
|
|
cache.Set("setting_max_worker_num", "10", 0)
|
||
|
|
mock.ExpectQuery("SELECT(.+)").WithArgs(Queued).WillReturnRows(sqlmock.NewRows([]string{"type"}).AddRow(-1))
|
||
|
|
Init()
|
||
|
|
asserts.NoError(mock.ExpectationsWereMet())
|
||
|
|
asserts.Len(TaskPoll.idleWorker, 10)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPool_Submit(t *testing.T) {
|
||
|
|
asserts := assert.New(t)
|
||
|
|
pool := &Pool{
|
||
|
|
idleWorker: make(chan int, 1),
|
||
|
|
}
|
||
|
|
pool.Add(1)
|
||
|
|
job := &MockJob{
|
||
|
|
DoFunc: func() {
|
||
|
|
|
||
|
|
},
|
||
|
|
}
|
||
|
|
asserts.NotPanics(func() {
|
||
|
|
pool.Submit(job)
|
||
|
|
})
|
||
|
|
}
|