Function TestListBooks

pkg/cli/cmd/view/book_test.go:154–184

(t *testing.T)

}

func TestListBooks(t *testing.T) {
	// Setup
	db := database.InitTestMemoryDB(t)
	defer db.Close()

	b1UUID := "js-book-uuid"
	b2UUID := "linux-book-uuid"

	database.MustExec(t, "inserting book 1", db, "INSERT INTO books (uuid, label) VALUES (?, ?)", b1UUID, "javascript")
	database.MustExec(t, "inserting book 2", db, "INSERT INTO books (uuid, label) VALUES (?, ?)", b2UUID, "linux")

	// Add notes to test count
	database.MustExec(t, "inserting note 1", db, "INSERT INTO notes (uuid, book_uuid, body, added_on) VALUES (?, ?, ?, ?)", "note-1", b1UUID, "note body 1", 1515199943)
	database.MustExec(t, "inserting note 2", db, "INSERT INTO notes (uuid, book_uuid, body, added_on) VALUES (?, ?, ?, ?)", "note-2", b1UUID, "note body 2", 1515199944)

	ctx := context.DnoteCtx{DB: db}
	var buf bytes.Buffer

	// Execute
	err := listBooks(ctx, &buf, false)
	if err != nil {
		t.Fatal(err)
	}

	got := buf.String()

	// Verify output
	assert.Equal(t, strings.Contains(got, "javascript"), true, "should contain javascript book")
	assert.Equal(t, strings.Contains(got, "linux"), true, "should contain linux book")
	assert.Equal(t, strings.Contains(got, "(2)"), true, "should show 2 notes for javascript")
}