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.
70 lines
1.4 KiB
70 lines
1.4 KiB
package ugc |
|
|
|
import ( |
|
"context" |
|
"flag" |
|
"fmt" |
|
"path/filepath" |
|
"testing" |
|
|
|
"go-common/app/job/main/tv/conf" |
|
"go-common/app/job/main/tv/model/ugc" |
|
|
|
. "github.com/smartystreets/goconvey/convey" |
|
) |
|
|
|
var ( |
|
ctx = context.TODO() |
|
d *Dao |
|
queryMid = "SELECT mid FROM ugc_archive WHERE result=1 AND valid=1 AND deleted=0 LIMIT 1" |
|
queryAid = "SELECT aid FROM ugc_archive WHERE result=1 AND valid=1 AND deleted=0 LIMIT 1" |
|
) |
|
|
|
func WithDao(f func(d *Dao)) func() { |
|
return func() { |
|
dir, _ := filepath.Abs("../../cmd/tv-job-test.toml") |
|
flag.Set("conf", dir) |
|
conf.Init() |
|
if d == nil { |
|
d = New(conf.Conf) |
|
} |
|
f(d) |
|
} |
|
} |
|
|
|
func TestDao_UpArcs(t *testing.T) { |
|
Convey("TestDao_UpArcs", t, WithDao(func(d *Dao) { |
|
var mid int64 |
|
d.DB.QueryRow(ctx, queryMid).Scan(&mid) |
|
if mid == 0 { |
|
return |
|
} |
|
res, err := d.UpArcs(ctx, mid) |
|
So(err, ShouldBeNil) |
|
So(len(res), ShouldBeGreaterThan, 0) |
|
})) |
|
} |
|
|
|
func TestDao_SetArcCMS(t *testing.T) { |
|
Convey("TestDao_SetArcCMS", t, WithDao(func(d *Dao) { |
|
var aid int64 |
|
d.DB.QueryRow(ctx, queryAid).Scan(&aid) |
|
if aid != 0 { |
|
err := d.SetArcCMS(ctx, &ugc.ArcCMS{ |
|
AID: aid, |
|
Title: "test", |
|
}) |
|
So(err, ShouldBeNil) |
|
fmt.Println(aid) |
|
} |
|
})) |
|
} |
|
|
|
func TestDao_CountArcs(t *testing.T) { |
|
Convey("TestDao_CountArcs", t, WithDao(func(d *Dao) { |
|
res, err := d.CountUpArcs(ctx, 452156) |
|
So(err, ShouldBeNil) |
|
So(res, ShouldBeGreaterThan, 0) |
|
fmt.Println(res) |
|
})) |
|
}
|
|
|