mirror of
https://github.com/eiblog/eiblog.git
synced 2026-02-04 13:52:26 +08:00
184 lines
3.5 KiB
Go
184 lines
3.5 KiB
Go
// Package store provides ...
|
|
package store
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/eiblog/eiblog/pkg/model"
|
|
)
|
|
|
|
var (
|
|
store Store
|
|
acct *model.Account
|
|
blogger *model.Blogger
|
|
series *model.Serie
|
|
article *model.Article
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
store, err = NewStore("mongodb", "mongodb://127.0.0.1:27017")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// account
|
|
acct = &model.Account{
|
|
Username: "deepzz",
|
|
Password: "deepzz",
|
|
Email: "deepzz@example.com",
|
|
PhoneN: "12345678900",
|
|
Address: "address",
|
|
CreatedAt: time.Now(),
|
|
}
|
|
// blogger
|
|
blogger = &model.Blogger{
|
|
BlogName: "Deepzz",
|
|
SubTitle: "不抛弃,不放弃",
|
|
BeiAn: "beian",
|
|
BTitle: "Deepzz's Blog",
|
|
Copyright: "Copyright",
|
|
}
|
|
// series
|
|
series = &model.Serie{
|
|
Slug: "slug",
|
|
Name: "series name",
|
|
Desc: "series desc",
|
|
CreatedAt: time.Now(),
|
|
}
|
|
// article
|
|
article = &model.Article{
|
|
Author: "deepzz",
|
|
Slug: "slug",
|
|
Title: "title",
|
|
Count: 0,
|
|
Content: "### count",
|
|
SerieID: 0,
|
|
Tags: nil,
|
|
IsDraft: false,
|
|
|
|
UpdatedAt: time.Now(),
|
|
CreatedAt: time.Now(),
|
|
}
|
|
}
|
|
|
|
func TestLoadInsertAccount(t *testing.T) {
|
|
ok, err := store.LoadInsertAccount(context.Background(), acct)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log(ok)
|
|
}
|
|
|
|
func TestUpdateAccount(t *testing.T) {
|
|
err := store.UpdateAccount(context.Background(), "deepzz", map[string]interface{}{
|
|
"phonn": "09876543211",
|
|
"loginua": "chrome",
|
|
"password": "123456",
|
|
"logintime": time.Now(),
|
|
"logouttime": time.Now(),
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestLoadInsertBlogger(t *testing.T) {
|
|
ok, err := store.LoadInsertBlogger(context.Background(), blogger)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log(ok)
|
|
}
|
|
|
|
func TestUpdateBlogger(t *testing.T) {
|
|
err := store.UpdateBlogger(context.Background(), map[string]interface{}{
|
|
"blogname": "blogname",
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestInsertSeries(t *testing.T) {
|
|
err := store.InsertSerie(context.Background(), series)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestRemoveSeries(t *testing.T) {
|
|
err := store.RemoveSerie(context.Background(), 1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestUpdateSeries(t *testing.T) {
|
|
err := store.UpdateSerie(context.Background(), 2, map[string]interface{}{
|
|
"desc": "update desc",
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestLoadAllSeries(t *testing.T) {
|
|
series, err := store.LoadAllSerie(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("load all series: %d", len(series))
|
|
}
|
|
|
|
func TestInsertArticle(t *testing.T) {
|
|
article.ID = 12
|
|
err := store.InsertArticle(context.Background(), article, 10)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestRemoveArticle(t *testing.T) {
|
|
err := store.RemoveArticle(context.Background(), 11)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestDeleteArticle(t *testing.T) {
|
|
err := store.RemoveArticle(context.Background(), 12)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestCleanArticles(t *testing.T) {
|
|
err := store.CleanArticles(context.Background(), time.Now())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestUpdateArticle(t *testing.T) {
|
|
err := store.UpdateArticle(context.Background(), 13, map[string]interface{}{
|
|
"title": "new title",
|
|
"updatetime": time.Now(),
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestLoadAllArticle(t *testing.T) {
|
|
_, total, err := store.LoadArticleList(context.Background(), SearchArticles{
|
|
Page: 1,
|
|
Limit: 1000,
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("load all articles: %d", total)
|
|
}
|