mirror of
https://github.com/dnote/dnote
synced 2026-03-15 15:05:51 +01:00
340 lines
15 KiB
Go
340 lines
15 KiB
Go
/* Copyright (C) 2019 Monomax Software Pty Ltd
|
|
*
|
|
* This file is part of Dnote CLI.
|
|
*
|
|
* Dnote CLI is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Dnote CLI is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Dnote CLI. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"testing"
|
|
|
|
"github.com/dnote/dnote/pkg/assert"
|
|
"github.com/dnote/dnote/pkg/cli/consts"
|
|
"github.com/dnote/dnote/pkg/cli/database"
|
|
"github.com/dnote/dnote/pkg/cli/testutils"
|
|
"github.com/dnote/dnote/pkg/cli/utils"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
var binaryName = "test-dnote"
|
|
|
|
var opts = testutils.RunDnoteCmdOptions{
|
|
HomeDir: "./tmp",
|
|
DnoteDir: "./tmp/.dnote",
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
if err := exec.Command("go", "build", "--tags", "fts5", "-o", binaryName).Run(); err != nil {
|
|
log.Print(errors.Wrap(err, "building a binary").Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
func TestInit(t *testing.T) {
|
|
// Execute
|
|
testutils.RunDnoteCmd(t, opts, binaryName)
|
|
defer testutils.RemoveDir(t, opts.HomeDir)
|
|
|
|
db := database.OpenTestDB(t, opts.DnoteDir)
|
|
|
|
// Test
|
|
ok, err := utils.FileExists(opts.DnoteDir)
|
|
if err != nil {
|
|
t.Fatal(errors.Wrap(err, "checking if dnote dir exists"))
|
|
}
|
|
if !ok {
|
|
t.Errorf("dnote directory was not initialized")
|
|
}
|
|
|
|
ok, err = utils.FileExists(fmt.Sprintf("%s/%s", opts.DnoteDir, consts.ConfigFilename))
|
|
if err != nil {
|
|
t.Fatal(errors.Wrap(err, "checking if dnote config exists"))
|
|
}
|
|
if !ok {
|
|
t.Errorf("config file was not initialized")
|
|
}
|
|
|
|
var notesTableCount, booksTableCount, systemTableCount int
|
|
database.MustScan(t, "counting notes",
|
|
db.QueryRow("SELECT count(*) FROM sqlite_master WHERE type = ? AND name = ?", "table", "notes"), ¬esTableCount)
|
|
database.MustScan(t, "counting books",
|
|
db.QueryRow("SELECT count(*) FROM sqlite_master WHERE type = ? AND name = ?", "table", "books"), &booksTableCount)
|
|
database.MustScan(t, "counting system",
|
|
db.QueryRow("SELECT count(*) FROM sqlite_master WHERE type = ? AND name = ?", "table", "system"), &systemTableCount)
|
|
|
|
assert.Equal(t, notesTableCount, 1, "notes table count mismatch")
|
|
assert.Equal(t, booksTableCount, 1, "books table count mismatch")
|
|
assert.Equal(t, systemTableCount, 1, "system table count mismatch")
|
|
|
|
// test that all default system configurations are generated
|
|
var lastUpgrade, lastMaxUSN, lastSyncAt string
|
|
database.MustScan(t, "scanning last upgrade",
|
|
db.QueryRow("SELECT value FROM system WHERE key = ?", consts.SystemLastUpgrade), &lastUpgrade)
|
|
database.MustScan(t, "scanning last max usn",
|
|
db.QueryRow("SELECT value FROM system WHERE key = ?", consts.SystemLastMaxUSN), &lastMaxUSN)
|
|
database.MustScan(t, "scanning last sync at",
|
|
db.QueryRow("SELECT value FROM system WHERE key = ?", consts.SystemLastSyncAt), &lastSyncAt)
|
|
|
|
assert.NotEqual(t, lastUpgrade, "", "last upgrade should not be empty")
|
|
assert.NotEqual(t, lastMaxUSN, "", "last max usn should not be empty")
|
|
assert.NotEqual(t, lastSyncAt, "", "last sync at should not be empty")
|
|
}
|
|
|
|
func TestAddNote_NewBook_BodyFlag(t *testing.T) {
|
|
// Set up
|
|
testutils.RunDnoteCmd(t, opts, binaryName, "add", "js", "-c", "foo")
|
|
defer testutils.RemoveDir(t, opts.HomeDir)
|
|
|
|
db := database.OpenTestDB(t, opts.DnoteDir)
|
|
|
|
// Execute
|
|
|
|
// Test
|
|
var noteCount, bookCount int
|
|
database.MustScan(t, "counting books", db.QueryRow("SELECT count(*) FROM books"), &bookCount)
|
|
database.MustScan(t, "counting notes", db.QueryRow("SELECT count(*) FROM notes"), ¬eCount)
|
|
|
|
assert.Equalf(t, bookCount, 1, "book count mismatch")
|
|
assert.Equalf(t, noteCount, 1, "note count mismatch")
|
|
|
|
var book database.Book
|
|
database.MustScan(t, "getting book", db.QueryRow("SELECT uuid, dirty FROM books where label = ?", "js"), &book.UUID, &book.Dirty)
|
|
var note database.Note
|
|
database.MustScan(t, "getting note",
|
|
db.QueryRow("SELECT uuid, body, added_on, dirty FROM notes where book_uuid = ?", book.UUID), ¬e.UUID, ¬e.Body, ¬e.AddedOn, ¬e.Dirty)
|
|
|
|
assert.Equal(t, book.Dirty, true, "Book dirty mismatch")
|
|
|
|
assert.NotEqual(t, note.UUID, "", "Note should have UUID")
|
|
assert.Equal(t, note.Body, "foo", "Note body mismatch")
|
|
assert.Equal(t, note.Dirty, true, "Note dirty mismatch")
|
|
assert.NotEqual(t, note.AddedOn, int64(0), "Note added_on mismatch")
|
|
}
|
|
|
|
func TestAddNote_ExistingBook_BodyFlag(t *testing.T) {
|
|
// Setup
|
|
db := database.InitTestDB(t, fmt.Sprintf("%s/%s", opts.DnoteDir, consts.DnoteDBFileName), nil)
|
|
testutils.Setup3(t, db)
|
|
|
|
// Execute
|
|
testutils.RunDnoteCmd(t, opts, binaryName, "add", "js", "-c", "foo")
|
|
defer testutils.RemoveDir(t, opts.HomeDir)
|
|
|
|
// Test
|
|
|
|
var noteCount, bookCount int
|
|
database.MustScan(t, "counting books", db.QueryRow("SELECT count(*) FROM books"), &bookCount)
|
|
database.MustScan(t, "counting notes", db.QueryRow("SELECT count(*) FROM notes"), ¬eCount)
|
|
|
|
assert.Equalf(t, bookCount, 1, "book count mismatch")
|
|
assert.Equalf(t, noteCount, 2, "note count mismatch")
|
|
|
|
var n1, n2 database.Note
|
|
database.MustScan(t, "getting n1",
|
|
db.QueryRow("SELECT uuid, body, added_on, dirty FROM notes WHERE book_uuid = ? AND uuid = ?", "js-book-uuid", "43827b9a-c2b0-4c06-a290-97991c896653"), &n1.UUID, &n1.Body, &n1.AddedOn, &n1.Dirty)
|
|
database.MustScan(t, "getting n2",
|
|
db.QueryRow("SELECT uuid, body, added_on, dirty FROM notes WHERE book_uuid = ? AND body = ?", "js-book-uuid", "foo"), &n2.UUID, &n2.Body, &n2.AddedOn, &n2.Dirty)
|
|
|
|
var book database.Book
|
|
database.MustScan(t, "getting book", db.QueryRow("SELECT dirty FROM books where label = ?", "js"), &book.Dirty)
|
|
|
|
assert.Equal(t, book.Dirty, false, "Book dirty mismatch")
|
|
|
|
assert.NotEqual(t, n1.UUID, "", "n1 should have UUID")
|
|
assert.Equal(t, n1.Body, "Booleans have toString()", "n1 body mismatch")
|
|
assert.Equal(t, n1.AddedOn, int64(1515199943), "n1 added_on mismatch")
|
|
assert.Equal(t, n1.Dirty, false, "n1 dirty mismatch")
|
|
|
|
assert.NotEqual(t, n2.UUID, "", "n2 should have UUID")
|
|
assert.Equal(t, n2.Body, "foo", "n2 body mismatch")
|
|
assert.Equal(t, n2.Dirty, true, "n2 dirty mismatch")
|
|
}
|
|
|
|
func TestEditNote_BodyFlag(t *testing.T) {
|
|
// Setup
|
|
db := database.InitTestDB(t, fmt.Sprintf("%s/%s", opts.DnoteDir, consts.DnoteDBFileName), nil)
|
|
testutils.Setup4(t, db)
|
|
|
|
// Execute
|
|
testutils.RunDnoteCmd(t, opts, binaryName, "edit", "2", "-c", "foo bar")
|
|
defer testutils.RemoveDir(t, opts.HomeDir)
|
|
|
|
// Test
|
|
|
|
var noteCount, bookCount int
|
|
database.MustScan(t, "counting books", db.QueryRow("SELECT count(*) FROM books"), &bookCount)
|
|
database.MustScan(t, "counting notes", db.QueryRow("SELECT count(*) FROM notes"), ¬eCount)
|
|
|
|
assert.Equalf(t, bookCount, 1, "book count mismatch")
|
|
assert.Equalf(t, noteCount, 2, "note count mismatch")
|
|
|
|
var n1, n2 database.Note
|
|
database.MustScan(t, "getting n1",
|
|
db.QueryRow("SELECT uuid, body, added_on, dirty FROM notes where book_uuid = ? AND uuid = ?", "js-book-uuid", "43827b9a-c2b0-4c06-a290-97991c896653"), &n1.UUID, &n1.Body, &n1.AddedOn, &n1.Dirty)
|
|
database.MustScan(t, "getting n2",
|
|
db.QueryRow("SELECT uuid, body, added_on, dirty FROM notes where book_uuid = ? AND uuid = ?", "js-book-uuid", "f0d0fbb7-31ff-45ae-9f0f-4e429c0c797f"), &n2.UUID, &n2.Body, &n2.AddedOn, &n2.Dirty)
|
|
|
|
assert.Equal(t, n1.UUID, "43827b9a-c2b0-4c06-a290-97991c896653", "n1 should have UUID")
|
|
assert.Equal(t, n1.Body, "Booleans have toString()", "n1 body mismatch")
|
|
assert.Equal(t, n1.Dirty, false, "n1 dirty mismatch")
|
|
|
|
assert.Equal(t, n2.UUID, "f0d0fbb7-31ff-45ae-9f0f-4e429c0c797f", "Note should have UUID")
|
|
assert.Equal(t, n2.Body, "foo bar", "Note body mismatch")
|
|
assert.Equal(t, n2.Dirty, true, "n2 dirty mismatch")
|
|
assert.NotEqual(t, n2.EditedOn, 0, "Note edited_on mismatch")
|
|
}
|
|
|
|
func TestRemoveNote(t *testing.T) {
|
|
// Setup
|
|
db := database.InitTestDB(t, fmt.Sprintf("%s/%s", opts.DnoteDir, consts.DnoteDBFileName), nil)
|
|
testutils.Setup2(t, db)
|
|
|
|
// Execute
|
|
testutils.WaitDnoteCmd(t, opts, testutils.UserConfirm, binaryName, "remove", "js", "1")
|
|
defer testutils.RemoveDir(t, opts.HomeDir)
|
|
|
|
// Test
|
|
var noteCount, bookCount, jsNoteCount, linuxNoteCount int
|
|
database.MustScan(t, "counting books", db.QueryRow("SELECT count(*) FROM books"), &bookCount)
|
|
database.MustScan(t, "counting notes", db.QueryRow("SELECT count(*) FROM notes"), ¬eCount)
|
|
database.MustScan(t, "counting js notes", db.QueryRow("SELECT count(*) FROM notes WHERE book_uuid = ?", "js-book-uuid"), &jsNoteCount)
|
|
database.MustScan(t, "counting linux notes", db.QueryRow("SELECT count(*) FROM notes WHERE book_uuid = ?", "linux-book-uuid"), &linuxNoteCount)
|
|
|
|
assert.Equalf(t, bookCount, 2, "book count mismatch")
|
|
assert.Equalf(t, noteCount, 3, "note count mismatch")
|
|
assert.Equal(t, jsNoteCount, 2, "js book should have 2 notes")
|
|
assert.Equal(t, linuxNoteCount, 1, "linux book book should have 1 note")
|
|
|
|
var b1, b2 database.Book
|
|
var n1, n2, n3 database.Note
|
|
database.MustScan(t, "getting b1",
|
|
db.QueryRow("SELECT label, deleted, usn FROM books WHERE uuid = ?", "js-book-uuid"),
|
|
&b1.Label, &b1.Deleted, &b1.USN)
|
|
database.MustScan(t, "getting b2",
|
|
db.QueryRow("SELECT label, deleted, usn FROM books WHERE uuid = ?", "linux-book-uuid"),
|
|
&b2.Label, &b2.Deleted, &b2.USN)
|
|
database.MustScan(t, "getting n1",
|
|
db.QueryRow("SELECT uuid, body, added_on, deleted, dirty, usn FROM notes WHERE book_uuid = ? AND uuid = ?", "js-book-uuid", "f0d0fbb7-31ff-45ae-9f0f-4e429c0c797f"),
|
|
&n1.UUID, &n1.Body, &n1.AddedOn, &n1.Deleted, &n1.Dirty, &n1.USN)
|
|
database.MustScan(t, "getting n2",
|
|
db.QueryRow("SELECT uuid, body, added_on, deleted, dirty, usn FROM notes WHERE book_uuid = ? AND uuid = ?", "js-book-uuid", "43827b9a-c2b0-4c06-a290-97991c896653"),
|
|
&n2.UUID, &n2.Body, &n2.AddedOn, &n2.Deleted, &n2.Dirty, &n2.USN)
|
|
database.MustScan(t, "getting n3",
|
|
db.QueryRow("SELECT uuid, body, added_on, deleted, dirty, usn FROM notes WHERE book_uuid = ? AND uuid = ?", "linux-book-uuid", "3e065d55-6d47-42f2-a6bf-f5844130b2d2"),
|
|
&n3.UUID, &n3.Body, &n3.AddedOn, &n3.Deleted, &n3.Dirty, &n3.USN)
|
|
|
|
assert.Equal(t, b1.Label, "js", "b1 label mismatch")
|
|
assert.Equal(t, b1.Deleted, false, "b1 deleted mismatch")
|
|
assert.Equal(t, b1.Dirty, false, "b1 Dirty mismatch")
|
|
assert.Equal(t, b1.USN, 111, "b1 usn mismatch")
|
|
|
|
assert.Equal(t, b2.Label, "linux", "b2 label mismatch")
|
|
assert.Equal(t, b2.Deleted, false, "b2 deleted mismatch")
|
|
assert.Equal(t, b2.Dirty, false, "b2 Dirty mismatch")
|
|
assert.Equal(t, b2.USN, 122, "b2 usn mismatch")
|
|
|
|
assert.Equal(t, n1.UUID, "f0d0fbb7-31ff-45ae-9f0f-4e429c0c797f", "n1 should have UUID")
|
|
assert.Equal(t, n1.Body, "", "n1 body mismatch")
|
|
assert.Equal(t, n1.Deleted, true, "n1 deleted mismatch")
|
|
assert.Equal(t, n1.Dirty, true, "n1 Dirty mismatch")
|
|
assert.Equal(t, n1.USN, 11, "n1 usn mismatch")
|
|
|
|
assert.Equal(t, n2.UUID, "43827b9a-c2b0-4c06-a290-97991c896653", "n2 should have UUID")
|
|
assert.Equal(t, n2.Body, "n2 body", "n2 body mismatch")
|
|
assert.Equal(t, n2.Deleted, false, "n2 deleted mismatch")
|
|
assert.Equal(t, n2.Dirty, false, "n2 Dirty mismatch")
|
|
assert.Equal(t, n2.USN, 12, "n2 usn mismatch")
|
|
|
|
assert.Equal(t, n3.UUID, "3e065d55-6d47-42f2-a6bf-f5844130b2d2", "n3 should have UUID")
|
|
assert.Equal(t, n3.Body, "n3 body", "n3 body mismatch")
|
|
assert.Equal(t, n3.Deleted, false, "n3 deleted mismatch")
|
|
assert.Equal(t, n3.Dirty, false, "n3 Dirty mismatch")
|
|
assert.Equal(t, n3.USN, 13, "n3 usn mismatch")
|
|
}
|
|
|
|
func TestRemoveBook(t *testing.T) {
|
|
// Setup
|
|
db := database.InitTestDB(t, fmt.Sprintf("%s/%s", opts.DnoteDir, consts.DnoteDBFileName), nil)
|
|
testutils.Setup2(t, db)
|
|
|
|
// Execute
|
|
testutils.WaitDnoteCmd(t, opts, testutils.UserConfirm, binaryName, "remove", "-b", "js")
|
|
defer testutils.RemoveDir(t, opts.HomeDir)
|
|
|
|
// Test
|
|
var noteCount, bookCount, jsNoteCount, linuxNoteCount int
|
|
database.MustScan(t, "counting books", db.QueryRow("SELECT count(*) FROM books"), &bookCount)
|
|
database.MustScan(t, "counting notes", db.QueryRow("SELECT count(*) FROM notes"), ¬eCount)
|
|
database.MustScan(t, "counting js notes", db.QueryRow("SELECT count(*) FROM notes WHERE book_uuid = ?", "js-book-uuid"), &jsNoteCount)
|
|
database.MustScan(t, "counting linux notes", db.QueryRow("SELECT count(*) FROM notes WHERE book_uuid = ?", "linux-book-uuid"), &linuxNoteCount)
|
|
|
|
assert.Equalf(t, bookCount, 2, "book count mismatch")
|
|
assert.Equalf(t, noteCount, 3, "note count mismatch")
|
|
assert.Equal(t, jsNoteCount, 2, "js book should have 2 notes")
|
|
assert.Equal(t, linuxNoteCount, 1, "linux book book should have 1 note")
|
|
|
|
var b1, b2 database.Book
|
|
var n1, n2, n3 database.Note
|
|
database.MustScan(t, "getting b1",
|
|
db.QueryRow("SELECT label, dirty, deleted, usn FROM books WHERE uuid = ?", "js-book-uuid"),
|
|
&b1.Label, &b1.Dirty, &b1.Deleted, &b1.USN)
|
|
database.MustScan(t, "getting b2",
|
|
db.QueryRow("SELECT label, dirty, deleted, usn FROM books WHERE uuid = ?", "linux-book-uuid"),
|
|
&b2.Label, &b2.Dirty, &b2.Deleted, &b2.USN)
|
|
database.MustScan(t, "getting n1",
|
|
db.QueryRow("SELECT uuid, body, added_on, dirty, deleted, usn FROM notes WHERE book_uuid = ? AND uuid = ?", "js-book-uuid", "f0d0fbb7-31ff-45ae-9f0f-4e429c0c797f"),
|
|
&n1.UUID, &n1.Body, &n1.AddedOn, &n1.Deleted, &n1.Dirty, &n1.USN)
|
|
database.MustScan(t, "getting n2",
|
|
db.QueryRow("SELECT uuid, body, added_on, dirty, deleted, usn FROM notes WHERE book_uuid = ? AND uuid = ?", "js-book-uuid", "43827b9a-c2b0-4c06-a290-97991c896653"),
|
|
&n2.UUID, &n2.Body, &n2.AddedOn, &n2.Deleted, &n2.Dirty, &n2.USN)
|
|
database.MustScan(t, "getting n3",
|
|
db.QueryRow("SELECT uuid, body, added_on, dirty, deleted, usn FROM notes WHERE book_uuid = ? AND uuid = ?", "linux-book-uuid", "3e065d55-6d47-42f2-a6bf-f5844130b2d2"),
|
|
&n3.UUID, &n3.Body, &n3.AddedOn, &n3.Deleted, &n3.Dirty, &n3.USN)
|
|
|
|
assert.NotEqual(t, b1.Label, "js", "b1 label mismatch")
|
|
assert.Equal(t, b1.Dirty, true, "b1 Dirty mismatch")
|
|
assert.Equal(t, b1.Deleted, true, "b1 deleted mismatch")
|
|
assert.Equal(t, b1.USN, 111, "b1 usn mismatch")
|
|
|
|
assert.Equal(t, b2.Label, "linux", "b2 label mismatch")
|
|
assert.Equal(t, b2.Dirty, false, "b2 Dirty mismatch")
|
|
assert.Equal(t, b2.Deleted, false, "b2 deleted mismatch")
|
|
assert.Equal(t, b2.USN, 122, "b2 usn mismatch")
|
|
|
|
assert.Equal(t, n1.UUID, "f0d0fbb7-31ff-45ae-9f0f-4e429c0c797f", "n1 should have UUID")
|
|
assert.Equal(t, n1.Body, "", "n1 body mismatch")
|
|
assert.Equal(t, n1.Dirty, true, "n1 Dirty mismatch")
|
|
assert.Equal(t, n1.Deleted, true, "n1 deleted mismatch")
|
|
assert.Equal(t, n1.USN, 11, "n1 usn mismatch")
|
|
|
|
assert.Equal(t, n2.UUID, "43827b9a-c2b0-4c06-a290-97991c896653", "n2 should have UUID")
|
|
assert.Equal(t, n2.Body, "", "n2 body mismatch")
|
|
assert.Equal(t, n2.Dirty, true, "n2 Dirty mismatch")
|
|
assert.Equal(t, n2.Deleted, true, "n2 deleted mismatch")
|
|
assert.Equal(t, n2.USN, 12, "n2 usn mismatch")
|
|
|
|
assert.Equal(t, n3.UUID, "3e065d55-6d47-42f2-a6bf-f5844130b2d2", "n3 should have UUID")
|
|
assert.Equal(t, n3.Body, "n3 body", "n3 body mismatch")
|
|
assert.Equal(t, n3.Dirty, false, "n3 Dirty mismatch")
|
|
assert.Equal(t, n3.Deleted, false, "n3 deleted mismatch")
|
|
assert.Equal(t, n3.USN, 13, "n3 usn mismatch")
|
|
}
|