mirror of
https://github.com/dnote/dnote
synced 2026-03-14 22:45:50 +01:00
b03ca999a5* Remove encrypted fields from notes and books * Remove public from notes * Use consistent flags
127 lines
3.3 KiB
Go
127 lines
3.3 KiB
Go
/* Copyright (C) 2019, 2020, 2021, 2022, 2023, 2024, 2025 Dnote contributors
|
|
*
|
|
* This file is part of Dnote.
|
|
*
|
|
* Dnote is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Dnote 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 Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with Dnote. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package operations
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/dnote/dnote/pkg/assert"
|
|
"github.com/dnote/dnote/pkg/server/database"
|
|
"github.com/dnote/dnote/pkg/server/testutils"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func TestGetNote(t *testing.T) {
|
|
db := testutils.InitMemoryDB(t)
|
|
|
|
user := testutils.SetupUserData(db)
|
|
anotherUser := testutils.SetupUserData(db)
|
|
|
|
b1 := database.Book{
|
|
UUID: testutils.MustUUID(t),
|
|
UserID: user.ID,
|
|
Label: "js",
|
|
}
|
|
testutils.MustExec(t, db.Save(&b1), "preparing b1")
|
|
|
|
note := database.Note{
|
|
UUID: testutils.MustUUID(t),
|
|
UserID: user.ID,
|
|
BookUUID: b1.UUID,
|
|
Body: "note content",
|
|
Deleted: false,
|
|
}
|
|
testutils.MustExec(t, db.Save(¬e), "preparing note")
|
|
|
|
var noteRecord database.Note
|
|
testutils.MustExec(t, db.Where("uuid = ?", note.UUID).Preload("Book").Preload("User").First(¬eRecord), "finding note")
|
|
|
|
testCases := []struct {
|
|
name string
|
|
user database.User
|
|
note database.Note
|
|
expectedOK bool
|
|
expectedNote database.Note
|
|
}{
|
|
{
|
|
name: "owner accessing note",
|
|
user: user,
|
|
note: note,
|
|
expectedOK: true,
|
|
expectedNote: noteRecord,
|
|
},
|
|
{
|
|
name: "non-owner accessing note",
|
|
user: anotherUser,
|
|
note: note,
|
|
expectedOK: false,
|
|
expectedNote: database.Note{},
|
|
},
|
|
{
|
|
name: "guest accessing note",
|
|
user: database.User{},
|
|
note: note,
|
|
expectedOK: false,
|
|
expectedNote: database.Note{},
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
note, ok, err := GetNote(db, tc.note.UUID, &tc.user)
|
|
if err != nil {
|
|
t.Fatal(errors.Wrap(err, "executing"))
|
|
}
|
|
|
|
assert.Equal(t, ok, tc.expectedOK, "ok mismatch")
|
|
assert.DeepEqual(t, note, tc.expectedNote, "note mismatch")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGetNote_nonexistent(t *testing.T) {
|
|
db := testutils.InitMemoryDB(t)
|
|
|
|
user := testutils.SetupUserData(db)
|
|
|
|
b1 := database.Book{
|
|
UUID: testutils.MustUUID(t),
|
|
UserID: user.ID,
|
|
Label: "js",
|
|
}
|
|
testutils.MustExec(t, db.Save(&b1), "preparing b1")
|
|
|
|
n1 := database.Note{
|
|
UUID: "4fd19336-671e-4ff3-8f22-662b80e22edc",
|
|
UserID: user.ID,
|
|
BookUUID: b1.UUID,
|
|
Body: "n1 content",
|
|
Deleted: false,
|
|
}
|
|
testutils.MustExec(t, db.Save(&n1), "preparing n1")
|
|
|
|
nonexistentUUID := "4fd19336-671e-4ff3-8f22-662b80e22edd"
|
|
note, ok, err := GetNote(db, nonexistentUUID, &user)
|
|
if err != nil {
|
|
t.Fatal(errors.Wrap(err, "executing"))
|
|
}
|
|
|
|
assert.Equal(t, ok, false, "ok mismatch")
|
|
assert.DeepEqual(t, note, database.Note{}, "note mismatch")
|
|
}
|