/* Copyright (C) 2019, 2020, 2021 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 . */ package api import ( "fmt" "net/http" "testing" "github.com/dnote/dnote/pkg/assert" "github.com/dnote/dnote/pkg/clock" "github.com/dnote/dnote/pkg/server/app" "github.com/dnote/dnote/pkg/server/database" "github.com/dnote/dnote/pkg/server/testutils" ) func TestCreateNote(t *testing.T) { defer testutils.ClearData(testutils.DB) // Setup server := MustNewServer(t, &app.App{ Clock: clock.NewMock(), }) defer server.Close() user := testutils.SetupUserData() testutils.MustExec(t, testutils.DB.Model(&user).Update("max_usn", 101), "preparing user max_usn") b1 := database.Book{ UserID: user.ID, Label: "js", USN: 58, } testutils.MustExec(t, testutils.DB.Save(&b1), "preparing b1") // Execute dat := fmt.Sprintf(`{"book_uuid": "%s", "content": "note content"}`, b1.UUID) req := testutils.MakeReq(server.URL, "POST", "/v3/notes", dat) res := testutils.HTTPAuthDo(t, req, user) // Test assert.StatusCodeEquals(t, res, http.StatusCreated, "") var noteRecord database.Note var bookRecord database.Book var userRecord database.User var bookCount, noteCount int testutils.MustExec(t, testutils.DB.Model(&database.Book{}).Count(&bookCount), "counting books") testutils.MustExec(t, testutils.DB.Model(&database.Note{}).Count(¬eCount), "counting notes") testutils.MustExec(t, testutils.DB.First(¬eRecord), "finding note") testutils.MustExec(t, testutils.DB.Where("id = ?", b1.ID).First(&bookRecord), "finding book") testutils.MustExec(t, testutils.DB.Where("id = ?", user.ID).First(&userRecord), "finding user record") assert.Equalf(t, bookCount, 1, "book count mismatch") assert.Equalf(t, noteCount, 1, "note count mismatch") assert.Equal(t, bookRecord.Label, b1.Label, "book name mismatch") assert.Equal(t, bookRecord.UUID, b1.UUID, "book uuid mismatch") assert.Equal(t, bookRecord.UserID, b1.UserID, "book user_id mismatch") assert.Equal(t, bookRecord.USN, 58, "book usn mismatch") assert.NotEqual(t, noteRecord.UUID, "", "note uuid should have been generated") assert.Equal(t, noteRecord.BookUUID, b1.UUID, "note book_uuid mismatch") assert.Equal(t, noteRecord.Body, "note content", "note content mismatch") assert.Equal(t, noteRecord.USN, 102, "note usn mismatch") } func TestUpdateNote(t *testing.T) { updatedBody := "some updated content" b1UUID := "37868a8e-a844-4265-9a4f-0be598084733" b2UUID := "8f3bd424-6aa5-4ed5-910d-e5b38ab09f8c" testCases := []struct { payload string noteUUID string noteBookUUID string noteBody string notePublic bool noteDeleted bool expectedNoteBody string expectedNoteBookName string expectedNoteBookUUID string expectedNotePublic bool }{ { payload: fmt.Sprintf(`{ "content": "%s" }`, updatedBody), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: false, noteBody: "original content", noteDeleted: false, expectedNoteBookUUID: b1UUID, expectedNoteBody: "some updated content", expectedNoteBookName: "css", expectedNotePublic: false, }, { payload: fmt.Sprintf(`{ "book_uuid": "%s" }`, b1UUID), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: false, noteBody: "original content", noteDeleted: false, expectedNoteBookUUID: b1UUID, expectedNoteBody: "original content", expectedNoteBookName: "css", expectedNotePublic: false, }, { payload: fmt.Sprintf(`{ "book_uuid": "%s" }`, b2UUID), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: false, noteBody: "original content", noteDeleted: false, expectedNoteBookUUID: b2UUID, expectedNoteBody: "original content", expectedNoteBookName: "js", expectedNotePublic: false, }, { payload: fmt.Sprintf(`{ "book_uuid": "%s", "content": "%s" }`, b2UUID, updatedBody), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: false, noteBody: "original content", noteDeleted: false, expectedNoteBookUUID: b2UUID, expectedNoteBody: "some updated content", expectedNoteBookName: "js", expectedNotePublic: false, }, { payload: fmt.Sprintf(`{ "book_uuid": "%s", "content": "%s" }`, b1UUID, updatedBody), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: false, noteBody: "", noteDeleted: true, expectedNoteBookUUID: b1UUID, expectedNoteBody: updatedBody, expectedNoteBookName: "js", expectedNotePublic: false, }, { payload: fmt.Sprintf(`{ "public": %t }`, true), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: false, noteBody: "original content", noteDeleted: false, expectedNoteBookUUID: b1UUID, expectedNoteBody: "original content", expectedNoteBookName: "css", expectedNotePublic: true, }, { payload: fmt.Sprintf(`{ "public": %t }`, false), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: true, noteBody: "original content", noteDeleted: false, expectedNoteBookUUID: b1UUID, expectedNoteBody: "original content", expectedNoteBookName: "css", expectedNotePublic: false, }, { payload: fmt.Sprintf(`{ "content": "%s", "public": %t }`, updatedBody, false), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: true, noteBody: "original content", noteDeleted: false, expectedNoteBookUUID: b1UUID, expectedNoteBody: updatedBody, expectedNoteBookName: "css", expectedNotePublic: false, }, { payload: fmt.Sprintf(`{ "book_uuid": "%s", "content": "%s", "public": %t }`, b2UUID, updatedBody, true), noteUUID: "ab50aa32-b232-40d8-b10f-10a7f9134053", noteBookUUID: b1UUID, notePublic: false, noteBody: "original content", noteDeleted: false, expectedNoteBookUUID: b2UUID, expectedNoteBody: updatedBody, expectedNoteBookName: "js", expectedNotePublic: true, }, } for idx, tc := range testCases { t.Run(fmt.Sprintf("test case %d", idx), func(t *testing.T) { defer testutils.ClearData(testutils.DB) // Setup server := MustNewServer(t, &app.App{ Clock: clock.NewMock(), }) defer server.Close() user := testutils.SetupUserData() testutils.MustExec(t, testutils.DB.Model(&user).Update("max_usn", 101), "preparing user max_usn") b1 := database.Book{ UUID: b1UUID, UserID: user.ID, Label: "css", } testutils.MustExec(t, testutils.DB.Save(&b1), "preparing b1") b2 := database.Book{ UUID: b2UUID, UserID: user.ID, Label: "js", } testutils.MustExec(t, testutils.DB.Save(&b2), "preparing b2") note := database.Note{ UserID: user.ID, UUID: tc.noteUUID, BookUUID: tc.noteBookUUID, Body: tc.noteBody, Deleted: tc.noteDeleted, Public: tc.notePublic, } testutils.MustExec(t, testutils.DB.Save(¬e), "preparing note") // Execute endpoint := fmt.Sprintf("/v3/notes/%s", note.UUID) req := testutils.MakeReq(server.URL, "PATCH", endpoint, tc.payload) res := testutils.HTTPAuthDo(t, req, user) // Test assert.StatusCodeEquals(t, res, http.StatusOK, "status code mismatch for test case") var bookRecord database.Book var noteRecord database.Note var userRecord database.User var noteCount, bookCount int testutils.MustExec(t, testutils.DB.Model(&database.Book{}).Count(&bookCount), "counting books") testutils.MustExec(t, testutils.DB.Model(&database.Note{}).Count(¬eCount), "counting notes") testutils.MustExec(t, testutils.DB.Where("uuid = ?", note.UUID).First(¬eRecord), "finding note") testutils.MustExec(t, testutils.DB.Where("id = ?", b1.ID).First(&bookRecord), "finding book") testutils.MustExec(t, testutils.DB.Where("id = ?", user.ID).First(&userRecord), "finding user record") assert.Equalf(t, bookCount, 2, "book count mismatch") assert.Equalf(t, noteCount, 1, "note count mismatch") assert.Equal(t, noteRecord.UUID, tc.noteUUID, "note uuid mismatch for test case") assert.Equal(t, noteRecord.Body, tc.expectedNoteBody, "note content mismatch for test case") assert.Equal(t, noteRecord.BookUUID, tc.expectedNoteBookUUID, "note book_uuid mismatch for test case") assert.Equal(t, noteRecord.Public, tc.expectedNotePublic, "note public mismatch for test case") assert.Equal(t, noteRecord.USN, 102, "note usn mismatch for test case") assert.Equal(t, userRecord.MaxUSN, 102, "user max_usn mismatch for test case") }) } } func TestDeleteNote(t *testing.T) { b1UUID := "37868a8e-a844-4265-9a4f-0be598084733" testCases := []struct { content string deleted bool originalUSN int expectedUSN int expectedMaxUSN int }{ { content: "n1 content", deleted: false, originalUSN: 12, expectedUSN: 982, expectedMaxUSN: 982, }, { content: "", deleted: true, originalUSN: 12, expectedUSN: 982, expectedMaxUSN: 982, }, } for _, tc := range testCases { t.Run(fmt.Sprintf("originally deleted %t", tc.deleted), func(t *testing.T) { defer testutils.ClearData(testutils.DB) // Setup server := MustNewServer(t, &app.App{ Clock: clock.NewMock(), }) defer server.Close() user := testutils.SetupUserData() testutils.MustExec(t, testutils.DB.Model(&user).Update("max_usn", 981), "preparing user max_usn") b1 := database.Book{ UUID: b1UUID, UserID: user.ID, Label: "js", } testutils.MustExec(t, testutils.DB.Save(&b1), "preparing b1") note := database.Note{ UserID: user.ID, BookUUID: b1.UUID, Body: tc.content, Deleted: tc.deleted, USN: tc.originalUSN, } testutils.MustExec(t, testutils.DB.Save(¬e), "preparing note") // Execute endpoint := fmt.Sprintf("/v3/notes/%s", note.UUID) req := testutils.MakeReq(server.URL, "DELETE", endpoint, "") res := testutils.HTTPAuthDo(t, req, user) // Test assert.StatusCodeEquals(t, res, http.StatusOK, "") var bookRecord database.Book var noteRecord database.Note var userRecord database.User var bookCount, noteCount int testutils.MustExec(t, testutils.DB.Model(&database.Book{}).Count(&bookCount), "counting books") testutils.MustExec(t, testutils.DB.Model(&database.Note{}).Count(¬eCount), "counting notes") testutils.MustExec(t, testutils.DB.Where("uuid = ?", note.UUID).First(¬eRecord), "finding note") testutils.MustExec(t, testutils.DB.Where("id = ?", b1.ID).First(&bookRecord), "finding book") testutils.MustExec(t, testutils.DB.Where("id = ?", user.ID).First(&userRecord), "finding user record") assert.Equalf(t, bookCount, 1, "book count mismatch") assert.Equalf(t, noteCount, 1, "note count mismatch") assert.Equal(t, noteRecord.UUID, note.UUID, "note uuid mismatch for test case") assert.Equal(t, noteRecord.Body, "", "note content mismatch for test case") assert.Equal(t, noteRecord.Deleted, true, "note deleted mismatch for test case") assert.Equal(t, noteRecord.BookUUID, note.BookUUID, "note book_uuid mismatch for test case") assert.Equal(t, noteRecord.UserID, note.UserID, "note user_id mismatch for test case") assert.Equal(t, noteRecord.USN, tc.expectedUSN, "note usn mismatch for test case") assert.Equal(t, userRecord.MaxUSN, tc.expectedMaxUSN, "user max_usn mismatch for test case") }) } }