From 22ac4607a1c283a719eea95844e07513b8a67570 Mon Sep 17 00:00:00 2001
From: Vyr Cossont <VyrCossont@users.noreply.github.com>
Date: Thu, 27 Jul 2023 01:30:39 -0700
Subject: [feature] Support setting private notes on accounts (#1982)

* Support setting private notes on accounts

* Reformat comment whitespace

* Add missing license headers

* Use apiutil.ParseID

* Rename Note model and cache to AccountNote

* Update golden cache config in test/envparsing.sh

* Rename gtsmodel/note.go to gtsmodel/accountnote.go

* Update AccountNote uniqueness constraint name

Now has same prefix as other indexes on this table.

---------

Co-authored-by: tobi <31960611+tsmethurst@users.noreply.github.com>
---
 internal/db/bundb/relationship_test.go | 47 ++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

(limited to 'internal/db/bundb/relationship_test.go')

diff --git a/internal/db/bundb/relationship_test.go b/internal/db/bundb/relationship_test.go
index d3f4a31d1..cf2df5144 100644
--- a/internal/db/bundb/relationship_test.go
+++ b/internal/db/bundb/relationship_test.go
@@ -912,6 +912,53 @@ func (suite *RelationshipTestSuite) TestUpdateFollow() {
 	suite.True(relationship.Notifying)
 }
 
+func (suite *RelationshipTestSuite) TestGetNote() {
+	ctx := context.Background()
+
+	// Retrieve a fixture note
+	account1 := suite.testAccounts["local_account_1"].ID
+	account2 := suite.testAccounts["local_account_2"].ID
+	expectedNote := suite.testAccountNotes["local_account_2_note_on_1"]
+	note, err := suite.db.GetNote(ctx, account2, account1)
+	suite.NoError(err)
+	suite.NotNil(note)
+	suite.Equal(expectedNote.ID, note.ID)
+	suite.Equal(expectedNote.Comment, note.Comment)
+}
+
+func (suite *RelationshipTestSuite) TestPutNote() {
+	ctx := context.Background()
+
+	// put a note in
+	account1 := suite.testAccounts["local_account_1"].ID
+	account2 := suite.testAccounts["local_account_2"].ID
+	err := suite.db.PutNote(ctx, &gtsmodel.AccountNote{
+		ID:              "01H539R2NA0M83JX15Y5RWKE97",
+		AccountID:       account1,
+		TargetAccountID: account2,
+		Comment:         "foo",
+	})
+	suite.NoError(err)
+
+	// make sure the note is in the db
+	note, err := suite.db.GetNote(ctx, account1, account2)
+	suite.NoError(err)
+	suite.NotNil(note)
+	suite.Equal("01H539R2NA0M83JX15Y5RWKE97", note.ID)
+	suite.Equal("foo", note.Comment)
+
+	// update the note
+	note.Comment = "bar"
+	err = suite.db.PutNote(ctx, note)
+	suite.NoError(err)
+
+	// make sure the comment changes
+	note, err = suite.db.GetNote(ctx, account1, account2)
+	suite.NoError(err)
+	suite.NotNil(note)
+	suite.Equal("bar", note.Comment)
+}
+
 func TestRelationshipTestSuite(t *testing.T) {
 	suite.Run(t, new(RelationshipTestSuite))
 }
-- 
cgit v1.2.3