summaryrefslogtreecommitdiff
path: root/internal/db/bundb/relationship_test.go
diff options
context:
space:
mode:
authorLibravatar Vyr Cossont <VyrCossont@users.noreply.github.com>2023-07-27 01:30:39 -0700
committerLibravatar GitHub <noreply@github.com>2023-07-27 10:30:39 +0200
commit22ac4607a1c283a719eea95844e07513b8a67570 (patch)
tree61baf7949b62f84d2dff29c4d99614d946774f64 /internal/db/bundb/relationship_test.go
parent[performance] retry db queries on busy errors (#2025) (diff)
downloadgotosocial-22ac4607a1c283a719eea95844e07513b8a67570.tar.xz
[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>
Diffstat (limited to 'internal/db/bundb/relationship_test.go')
-rw-r--r--internal/db/bundb/relationship_test.go47
1 files changed, 47 insertions, 0 deletions
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))
}