113 lines
4.3 KiB
Go
113 lines
4.3 KiB
Go
package migrator
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/coreos/bbolt"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/umputun/remark/app/store"
|
|
"github.com/umputun/remark/app/store/engine"
|
|
"github.com/umputun/remark/app/store/service"
|
|
)
|
|
|
|
var testDb = "/tmp/test-remark.db"
|
|
|
|
func TestRemark_Export(t *testing.T) {
|
|
b := prep(t)
|
|
r := Remark{DataStore: b}
|
|
|
|
buf := &bytes.Buffer{}
|
|
size, err := r.Export(buf, "radio-t")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 2, size)
|
|
|
|
c1, err := buf.ReadString('\n')
|
|
assert.Nil(t, err)
|
|
log.Print(c1)
|
|
exp := `{"id":"efbc17f177ee1a1c0ee6e1e025749966ec071adc","pid":"","text":"some text, <a href=\"http://radio-t.com\" rel=\"nofollow\">link</a>","user":{"name":"user name","id":"user1","picture":"","admin":false},"locator":{"site":"radio-t","url":"https://radio-t.com"},"score":0,"votes":{},"time":"2017-12-20T15:18:22-06:00"}` + "\n"
|
|
assert.Equal(t, exp, c1)
|
|
}
|
|
|
|
func TestRemark_Import(t *testing.T) {
|
|
r1 := `{"id":"efbc17f177ee1a1c0ee6e1e025749966ec071adc","pid":"","text":"some text, <a href=\"http://radio-t.com\" rel=\"nofollow\">link</a>","user":{"name":"user name","id":"user1","picture":"","profile":"","admin":false},"locator":{"site":"radio-t","url":"https://radio-t.com"},"score":0,"votes":{},"time":"2017-12-20T15:18:22-06:00"}` + "\n"
|
|
|
|
r2 := `{"id":"afbc17f177ee1a1c0ee6e1e025749966ec071adc","pid":"efbc17f177ee1a1c0ee6e1e025749966ec071adc","text":"some text2, <a href=\"http://radio-t.com\" rel=\"nofollow\">link</a>","user":{"name":"user name","id":"user1","picture":"","profile":"","admin":false},"locator":{"site":"radio-t","url":"https://radio-t.com"},"score":0,"votes":{},"time":"2017-12-20T15:18:23-06:00"}` + "\n"
|
|
|
|
buf := &bytes.Buffer{}
|
|
buf.WriteString(r1)
|
|
buf.WriteString(r2)
|
|
buf.WriteString("{}")
|
|
|
|
os.Remove(testDb)
|
|
b, err := engine.NewBoltDB(bolt.Options{}, engine.BoltSite{SiteID: "radio-t", FileName: testDb})
|
|
assert.Nil(t, err)
|
|
r := Remark{DataStore: &service.DataStore{Interface: b}}
|
|
size, err := r.Import(buf, "radio-t")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 2, size)
|
|
|
|
comments, err := b.Find(store.Locator{SiteID: "radio-t", URL: "https://radio-t.com"}, "time")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 2, len(comments))
|
|
assert.Equal(t, "efbc17f177ee1a1c0ee6e1e025749966ec071adc", comments[0].ID)
|
|
assert.Equal(t, "afbc17f177ee1a1c0ee6e1e025749966ec071adc", comments[1].ID)
|
|
assert.Equal(t, "efbc17f177ee1a1c0ee6e1e025749966ec071adc", comments[1].ParentID)
|
|
}
|
|
|
|
func TestRemark_ImportManyWithError(t *testing.T) {
|
|
goodRec := `{"id":"%d","pid":"","text":"some text, <a href=\"http://radio-t.com\" rel=\"nofollow\">link</a>","user":{"name":"user name","id":"user1","picture":"","profile":"","admin":false},"locator":{"site":"radio-t","url":"https://radio-t.com"},"score":0,"votes":{},"time":"2017-12-20T15:18:22-06:00"}` + "\n"
|
|
|
|
buf := &bytes.Buffer{}
|
|
for i := 0; i < 1200; i++ {
|
|
buf.WriteString(fmt.Sprintf(goodRec, i))
|
|
}
|
|
buf.WriteString("bad1\n")
|
|
buf.WriteString("bad2\n")
|
|
|
|
os.Remove(testDb)
|
|
b, err := engine.NewBoltDB(bolt.Options{}, engine.BoltSite{SiteID: "radio-t", FileName: testDb})
|
|
assert.Nil(t, err)
|
|
r := Remark{DataStore: &service.DataStore{Interface: b}}
|
|
n, err := r.Import(buf, "radio-t")
|
|
assert.EqualError(t, err, "failed to save 2 comments")
|
|
assert.Equal(t, 1200, n)
|
|
comments, err := b.Find(store.Locator{SiteID: "radio-t", URL: "https://radio-t.com"}, "time")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1200, len(comments))
|
|
}
|
|
|
|
// makes new boltdb, put two records
|
|
func prep(t *testing.T) *service.DataStore {
|
|
os.Remove(testDb)
|
|
|
|
boltStore, err := engine.NewBoltDB(bolt.Options{}, engine.BoltSite{SiteID: "radio-t", FileName: testDb})
|
|
assert.Nil(t, err)
|
|
|
|
b := &service.DataStore{Interface: boltStore}
|
|
|
|
comment := store.Comment{
|
|
ID: "efbc17f177ee1a1c0ee6e1e025749966ec071adc",
|
|
Text: `some text, <a href="http://radio-t.com">link</a>`,
|
|
Timestamp: time.Date(2017, 12, 20, 15, 18, 22, 0, time.Local),
|
|
Locator: store.Locator{URL: "https://radio-t.com", SiteID: "radio-t"},
|
|
User: store.User{ID: "user1", Name: "user name"},
|
|
}
|
|
_, err = b.Create(comment)
|
|
assert.Nil(t, err)
|
|
|
|
comment = store.Comment{
|
|
Text: "some text2", Timestamp: time.Date(2017, 12, 20, 15, 18, 23, 0, time.Local),
|
|
Locator: store.Locator{URL: "https://radio-t.com", SiteID: "radio-t"},
|
|
User: store.User{ID: "user1", Name: "user name"},
|
|
}
|
|
_, err = b.Create(comment)
|
|
assert.Nil(t, err)
|
|
|
|
return b
|
|
}
|