From 1ccbf3328f905764fcf9dadc2f3734ddb91ab478 Mon Sep 17 00:00:00 2001 From: Umputun Date: Mon, 17 Dec 2018 03:02:15 -0600 Subject: [PATCH] test for sequential dbl import --- backend/app/migrator/remark_test.go | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/backend/app/migrator/remark_test.go b/backend/app/migrator/remark_test.go index 78a50766..7819466a 100644 --- a/backend/app/migrator/remark_test.go +++ b/backend/app/migrator/remark_test.go @@ -48,9 +48,7 @@ func TestRemark_Import(t *testing.T) { 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) + b := prep(t) // write some recs r := Remark{DataStore: &service.DataStore{Interface: b, AdminStore: admin.NewStaticStore("12345", []string{}, "")}} size, err := r.Import(buf, "radio-t") assert.Nil(t, err) @@ -62,6 +60,14 @@ func TestRemark_Import(t *testing.T) { assert.Equal(t, "efbc17f177ee1a1c0ee6e1e025749966ec071adc", comments[0].ID) assert.Equal(t, "afbc17f177ee1a1c0ee6e1e025749966ec071adc", comments[1].ID) assert.Equal(t, "efbc17f177ee1a1c0ee6e1e025749966ec071adc", comments[1].ParentID) + + // try import again + buf.WriteString(r1) + buf.WriteString(r2) + buf.WriteString("{}") + size, err = r.Import(buf, "radio-t") + assert.Nil(t, err) + assert.Equal(t, 2, size) } func TestRemark_ImportManyWithError(t *testing.T) { @@ -76,9 +82,7 @@ func TestRemark_ImportManyWithError(t *testing.T) { 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) + b := prep(t) // write some recs r := Remark{DataStore: &service.DataStore{Interface: b, AdminStore: admin.NewStaticStore("12345", []string{}, "")}} n, err := r.Import(buf, "radio-t") assert.EqualError(t, err, "failed to save 2 comments")