mirror of
https://github.com/dnote/dnote
synced 2026-03-15 15:05:51 +01:00
257 lines
8.7 KiB
Go
257 lines
8.7 KiB
Go
/* Copyright (C) 2019 Monomax Software Pty Ltd
|
|
*
|
|
* 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 (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/dnote/dnote/pkg/assert"
|
|
"github.com/dnote/dnote/pkg/server/api/clock"
|
|
"github.com/dnote/dnote/pkg/server/database"
|
|
"github.com/dnote/dnote/pkg/server/testutils"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func init() {
|
|
testutils.InitTestDB()
|
|
}
|
|
|
|
func TestCreateNote(t *testing.T) {
|
|
serverTime := time.Date(2017, time.March, 14, 21, 15, 0, 0, time.UTC)
|
|
mockClock := clock.NewMock()
|
|
mockClock.SetNow(serverTime)
|
|
|
|
ts1 := time.Date(2018, time.November, 12, 10, 11, 0, 0, time.UTC).UnixNano()
|
|
ts2 := time.Date(2018, time.November, 15, 0, 1, 10, 0, time.UTC).UnixNano()
|
|
|
|
testCases := []struct {
|
|
userUSN int
|
|
addedOn *int64
|
|
editedOn *int64
|
|
expectedUSN int
|
|
expectedAddedOn int64
|
|
expectedEditedOn int64
|
|
}{
|
|
{
|
|
userUSN: 8,
|
|
addedOn: nil,
|
|
editedOn: nil,
|
|
expectedUSN: 9,
|
|
expectedAddedOn: serverTime.UnixNano(),
|
|
expectedEditedOn: 0,
|
|
},
|
|
{
|
|
userUSN: 102229,
|
|
addedOn: &ts1,
|
|
editedOn: nil,
|
|
expectedUSN: 102230,
|
|
expectedAddedOn: ts1,
|
|
expectedEditedOn: 0,
|
|
},
|
|
{
|
|
userUSN: 8099,
|
|
addedOn: &ts1,
|
|
editedOn: &ts2,
|
|
expectedUSN: 8100,
|
|
expectedAddedOn: ts1,
|
|
expectedEditedOn: ts2,
|
|
},
|
|
}
|
|
|
|
for idx, tc := range testCases {
|
|
func() {
|
|
defer testutils.ClearData()
|
|
db := database.DBConn
|
|
|
|
user := testutils.SetupUserData()
|
|
testutils.MustExec(t, db.Model(&user).Update("max_usn", tc.userUSN), fmt.Sprintf("preparing user max_usn for test case %d", idx))
|
|
|
|
anotherUser := testutils.SetupUserData()
|
|
testutils.MustExec(t, db.Model(&anotherUser).Update("max_usn", 55), fmt.Sprintf("preparing user max_usn for test case %d", idx))
|
|
|
|
b1 := database.Book{UserID: user.ID, Label: "js", Deleted: false}
|
|
testutils.MustExec(t, db.Save(&b1), fmt.Sprintf("preparing b1 for test case %d", idx))
|
|
|
|
tx := db.Begin()
|
|
if _, err := CreateNote(user, mockClock, b1.UUID, "note content", tc.addedOn, tc.editedOn, false); err != nil {
|
|
tx.Rollback()
|
|
t.Fatal(errors.Wrap(err, "deleting note"))
|
|
}
|
|
tx.Commit()
|
|
|
|
var bookCount, noteCount int
|
|
var noteRecord database.Note
|
|
var userRecord database.User
|
|
|
|
testutils.MustExec(t, db.Model(&database.Book{}).Count(&bookCount), fmt.Sprintf("counting book for test case %d", idx))
|
|
testutils.MustExec(t, db.Model(&database.Note{}).Count(¬eCount), fmt.Sprintf("counting notes for test case %d", idx))
|
|
testutils.MustExec(t, db.First(¬eRecord), fmt.Sprintf("finding note for test case %d", idx))
|
|
testutils.MustExec(t, db.Where("id = ?", user.ID).First(&userRecord), fmt.Sprintf("finding user for test case %d", idx))
|
|
|
|
assert.Equal(t, bookCount, 1, "book count mismatch")
|
|
assert.Equal(t, noteCount, 1, "note count mismatch")
|
|
assert.NotEqual(t, noteRecord.UUID, "", "note UUID should have been generated")
|
|
assert.Equal(t, noteRecord.UserID, user.ID, "note UserID mismatch")
|
|
assert.Equal(t, noteRecord.Body, "note content", "note Body mismatch")
|
|
assert.Equal(t, noteRecord.Deleted, false, "note Deleted mismatch")
|
|
assert.Equal(t, noteRecord.USN, tc.expectedUSN, "note Label mismatch")
|
|
assert.Equal(t, noteRecord.AddedOn, tc.expectedAddedOn, "note AddedOn mismatch")
|
|
assert.Equal(t, noteRecord.EditedOn, tc.expectedEditedOn, "note EditedOn mismatch")
|
|
|
|
assert.Equal(t, userRecord.MaxUSN, tc.expectedUSN, "user max_usn mismatch")
|
|
}()
|
|
}
|
|
}
|
|
|
|
func TestUpdateNote(t *testing.T) {
|
|
testCases := []struct {
|
|
userUSN int
|
|
expectedUSN int
|
|
}{
|
|
{
|
|
userUSN: 8,
|
|
expectedUSN: 9,
|
|
},
|
|
{
|
|
userUSN: 102229,
|
|
expectedUSN: 102230,
|
|
},
|
|
{
|
|
userUSN: 8099,
|
|
expectedUSN: 8100,
|
|
},
|
|
}
|
|
|
|
for idx, tc := range testCases {
|
|
func() {
|
|
defer testutils.ClearData()
|
|
db := database.DBConn
|
|
|
|
user := testutils.SetupUserData()
|
|
testutils.MustExec(t, db.Model(&user).Update("max_usn", tc.userUSN), fmt.Sprintf("preparing user max_usn for test case %d", idx))
|
|
|
|
anotherUser := testutils.SetupUserData()
|
|
testutils.MustExec(t, db.Model(&anotherUser).Update("max_usn", 55), fmt.Sprintf("preparing user max_usn for test case %d", idx))
|
|
|
|
b1 := database.Book{UserID: user.ID, Label: "js", Deleted: false}
|
|
testutils.MustExec(t, db.Save(&b1), fmt.Sprintf("preparing b1 for test case %d", idx))
|
|
|
|
note := database.Note{UserID: user.ID, Deleted: false, Body: "test content", BookUUID: b1.UUID}
|
|
testutils.MustExec(t, db.Save(¬e), fmt.Sprintf("preparing note for test case %d", idx))
|
|
|
|
c := clock.NewMock()
|
|
content := "updated test content"
|
|
|
|
tx := db.Begin()
|
|
if _, err := UpdateNote(tx, user, c, note, nil, &content); err != nil {
|
|
tx.Rollback()
|
|
t.Fatal(errors.Wrap(err, "deleting note"))
|
|
}
|
|
tx.Commit()
|
|
|
|
var bookCount, noteCount int
|
|
var noteRecord database.Note
|
|
var userRecord database.User
|
|
|
|
testutils.MustExec(t, db.Model(&database.Book{}).Count(&bookCount), fmt.Sprintf("counting book for test case %d", idx))
|
|
testutils.MustExec(t, db.Model(&database.Note{}).Count(¬eCount), fmt.Sprintf("counting notes for test case %d", idx))
|
|
testutils.MustExec(t, db.First(¬eRecord), fmt.Sprintf("finding note for test case %d", idx))
|
|
testutils.MustExec(t, db.Where("id = ?", user.ID).First(&userRecord), fmt.Sprintf("finding user for test case %d", idx))
|
|
|
|
assert.Equal(t, bookCount, 1, "book count mismatch")
|
|
assert.Equal(t, noteCount, 1, "note count mismatch")
|
|
assert.Equal(t, noteRecord.UserID, user.ID, "note UserID mismatch")
|
|
assert.Equal(t, noteRecord.Body, content, "note Body mismatch")
|
|
assert.Equal(t, noteRecord.Deleted, false, "note Deleted mismatch")
|
|
assert.Equal(t, noteRecord.USN, tc.expectedUSN, "note USN mismatch")
|
|
|
|
assert.Equal(t, userRecord.MaxUSN, tc.expectedUSN, "user MaxUSN mismatch")
|
|
}()
|
|
}
|
|
}
|
|
|
|
func TestDeleteNote(t *testing.T) {
|
|
testCases := []struct {
|
|
userUSN int
|
|
expectedUSN int
|
|
}{
|
|
{
|
|
userUSN: 3,
|
|
expectedUSN: 4,
|
|
},
|
|
{
|
|
userUSN: 9787,
|
|
expectedUSN: 9788,
|
|
},
|
|
{
|
|
userUSN: 787,
|
|
expectedUSN: 788,
|
|
},
|
|
}
|
|
|
|
for idx, tc := range testCases {
|
|
func() {
|
|
defer testutils.ClearData()
|
|
db := database.DBConn
|
|
|
|
user := testutils.SetupUserData()
|
|
testutils.MustExec(t, db.Model(&user).Update("max_usn", tc.userUSN), fmt.Sprintf("preparing user max_usn for test case %d", idx))
|
|
|
|
anotherUser := testutils.SetupUserData()
|
|
testutils.MustExec(t, db.Model(&anotherUser).Update("max_usn", 55), fmt.Sprintf("preparing user max_usn for test case %d", idx))
|
|
|
|
b1 := database.Book{UserID: user.ID, Label: "testBook"}
|
|
testutils.MustExec(t, db.Save(&b1), fmt.Sprintf("preparing b1 for test case %d", idx))
|
|
|
|
note := database.Note{UserID: user.ID, Deleted: false, Body: "test content", BookUUID: b1.UUID}
|
|
testutils.MustExec(t, db.Save(¬e), fmt.Sprintf("preparing note for test case %d", idx))
|
|
|
|
tx := db.Begin()
|
|
ret, err := DeleteNote(tx, user, note)
|
|
if err != nil {
|
|
tx.Rollback()
|
|
t.Fatal(errors.Wrap(err, "deleting note"))
|
|
}
|
|
tx.Commit()
|
|
|
|
var noteCount int
|
|
var noteRecord database.Note
|
|
var userRecord database.User
|
|
|
|
testutils.MustExec(t, db.Model(&database.Note{}).Count(¬eCount), fmt.Sprintf("counting notes for test case %d", idx))
|
|
testutils.MustExec(t, db.First(¬eRecord), fmt.Sprintf("finding note for test case %d", idx))
|
|
testutils.MustExec(t, db.Where("id = ?", user.ID).First(&userRecord), fmt.Sprintf("finding user for test case %d", idx))
|
|
|
|
assert.Equal(t, noteCount, 1, "note count mismatch")
|
|
|
|
assert.Equal(t, noteRecord.UserID, user.ID, "note user_id mismatch")
|
|
assert.Equal(t, noteRecord.Body, "", "note content mismatch")
|
|
assert.Equal(t, noteRecord.Deleted, true, "note deleted flag mismatch")
|
|
assert.Equal(t, noteRecord.USN, tc.expectedUSN, "note label mismatch")
|
|
assert.Equal(t, userRecord.MaxUSN, tc.expectedUSN, "user max_usn mismatch")
|
|
|
|
assert.Equal(t, ret.UserID, user.ID, "note user_id mismatch")
|
|
assert.Equal(t, ret.Body, "", "note content mismatch")
|
|
assert.Equal(t, ret.Deleted, true, "note deleted flag mismatch")
|
|
assert.Equal(t, ret.USN, tc.expectedUSN, "note label mismatch")
|
|
}()
|
|
}
|
|
}
|