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.
cloudreve/models/download_test.go

172 lines
4.0 KiB

package model
import (
"errors"
"github.com/DATA-DOG/go-sqlmock"
"github.com/jinzhu/gorm"
"github.com/stretchr/testify/assert"
"testing"
)
func TestDownload_Create(t *testing.T) {
asserts := assert.New(t)
// 成功
{
mock.ExpectBegin()
mock.ExpectExec("INSERT(.+)").WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectCommit()
download := Download{GID: "1"}
id, err := download.Create()
asserts.NoError(mock.ExpectationsWereMet())
asserts.NoError(err)
asserts.EqualValues(1, id)
}
// 失败
{
mock.ExpectBegin()
mock.ExpectExec("INSERT(.+)").WillReturnError(errors.New("error"))
mock.ExpectRollback()
download := Download{GID: "1"}
id, err := download.Create()
asserts.NoError(mock.ExpectationsWereMet())
asserts.Error(err)
asserts.EqualValues(0, id)
}
}
func TestDownload_AfterFind(t *testing.T) {
asserts := assert.New(t)
// 成功
{
download := Download{Attrs: `{"gid":"123"}`}
err := download.AfterFind()
asserts.NoError(err)
asserts.Equal("123", download.StatusInfo.Gid)
}
// 忽略空值
{
download := Download{Attrs: ``}
err := download.AfterFind()
asserts.NoError(err)
asserts.Equal("", download.StatusInfo.Gid)
}
// 解析失败
{
download := Download{Attrs: `?`}
err := download.BeforeSave()
asserts.Error(err)
asserts.Equal("", download.StatusInfo.Gid)
}
// 关联任务
{
mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "error"}).AddRow(1, "error"))
download := Download{TaskID: 1}
download.BeforeSave()
asserts.NoError(mock.ExpectationsWereMet())
asserts.Equal("error", download.Task.Error)
}
}
func TestDownload_Save(t *testing.T) {
asserts := assert.New(t)
// 成功
{
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectCommit()
download := Download{
Model: gorm.Model{
ID: 1,
},
Attrs: `{"gid":"123"}`,
}
err := download.Save()
asserts.NoError(mock.ExpectationsWereMet())
asserts.NoError(err)
asserts.Equal("123", download.StatusInfo.Gid)
}
// 失败
{
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)").WillReturnError(errors.New("error"))
mock.ExpectRollback()
download := Download{
Model: gorm.Model{
ID: 1,
},
}
err := download.Save()
asserts.NoError(mock.ExpectationsWereMet())
asserts.Error(err)
}
}
func TestGetDownloadsByStatus(t *testing.T) {
asserts := assert.New(t)
mock.ExpectQuery("SELECT(.+)").WithArgs(0, 1).WillReturnRows(sqlmock.NewRows([]string{"gid"}).AddRow("0").AddRow("1"))
res := GetDownloadsByStatus(0, 1)
asserts.NoError(mock.ExpectationsWereMet())
asserts.Len(res, 2)
}
func TestGetDownloadByGid(t *testing.T) {
asserts := assert.New(t)
mock.ExpectQuery("SELECT(.+)").WithArgs(2, "gid").WillReturnRows(sqlmock.NewRows([]string{"g_id"}).AddRow("1"))
res, err := GetDownloadByGid("gid", 2)
asserts.NoError(mock.ExpectationsWereMet())
asserts.NoError(err)
asserts.Equal(res.GID, "1")
}
func TestDownload_GetOwner(t *testing.T) {
asserts := assert.New(t)
// 已经有User对象
{
download := &Download{User: &User{Nick: "nick"}}
user := download.GetOwner()
asserts.NotNil(user)
asserts.Equal("nick", user.Nick)
}
// 无User对象
{
download := &Download{UserID: 3}
mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"nick"}).AddRow("nick"))
user := download.GetOwner()
asserts.NoError(mock.ExpectationsWereMet())
asserts.NotNil(user)
asserts.Equal("nick", user.Nick)
}
}
func TestGetDownloadsByStatusAndUser(t *testing.T) {
asserts := assert.New(t)
// 列出全部
{
mock.ExpectQuery("SELECT(.+)").WithArgs(1, 1, 2).WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(2).AddRow(3))
res := GetDownloadsByStatusAndUser(0, 1, 1, 2)
asserts.NoError(mock.ExpectationsWereMet())
asserts.Len(res, 2)
}
// 列出全部,分页
{
mock.ExpectQuery("SELECT(.+)DESC(.+)").WithArgs(1, 1, 2).WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(2).AddRow(3))
res := GetDownloadsByStatusAndUser(2, 1, 1, 2)
asserts.NoError(mock.ExpectationsWereMet())
asserts.Len(res, 2)
}
}