diff options
author | 2023-10-25 16:04:53 +0200 | |
---|---|---|
committer | 2023-10-25 15:04:53 +0100 | |
commit | c7b6cd7770cad9bfdc23decffa7c4068752dbbbd (patch) | |
tree | 0f039fd34fb0287860fce06ff1c30dedd1882136 /internal/db/bundb/thread_test.go | |
parent | [bugfix] allow store smaller PNG image than 261 bytes (#2263) (#2298) (diff) | |
download | gotosocial-c7b6cd7770cad9bfdc23decffa7c4068752dbbbd.tar.xz |
[feature] Status thread mute/unmute functionality (#2278)
* add db models + functions for keeping track of threads
* give em the old linty testy
* create, remove, check mutes
* swagger
* testerino
* test mute/unmute via api
* add info log about new index creation
* thread + allow muting of any remote statuses that mention a local account
* IsStatusThreadMutedBy -> IsThreadMutedByAccount
* use common processing functions in status processor
* set = NULL
* favee!
* get rekt darlings, darlings get rekt
* testrig please, have mercy muy liege
Diffstat (limited to 'internal/db/bundb/thread_test.go')
-rw-r--r-- | internal/db/bundb/thread_test.go | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/internal/db/bundb/thread_test.go b/internal/db/bundb/thread_test.go new file mode 100644 index 000000000..4d14f73e2 --- /dev/null +++ b/internal/db/bundb/thread_test.go @@ -0,0 +1,91 @@ +// GoToSocial +// Copyright (C) GoToSocial Authors admin@gotosocial.org +// SPDX-License-Identifier: AGPL-3.0-or-later +// +// This program 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. +// +// This program 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 this program. If not, see <http://www.gnu.org/licenses/>. + +package bundb_test + +import ( + "context" + "testing" + + "github.com/stretchr/testify/suite" + "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" +) + +type ThreadTestSuite struct { + BunDBStandardTestSuite +} + +func (suite *ThreadTestSuite) TestPutThread() { + suite.NoError( + suite.db.PutThread( + context.Background(), + >smodel.Thread{ + ID: "01HCWK4HVQ4VGSS1G4VQP3AXZF", + }, + ), + ) +} + +func (suite *ThreadTestSuite) TestMuteUnmuteThread() { + var ( + threadID = suite.testThreads["local_account_1_status_1"].ID + accountID = suite.testAccounts["local_account_1"].ID + ctx = context.Background() + threadMute = >smodel.ThreadMute{ + ID: "01HD3K14B62YJHH4RR0DSZ1EQ2", + ThreadID: threadID, + AccountID: accountID, + } + ) + + // Mute the thread and ensure it's actually muted. + if err := suite.db.PutThreadMute(ctx, threadMute); err != nil { + suite.FailNow(err.Error()) + } + + muted, err := suite.db.IsThreadMutedByAccount(ctx, threadID, accountID) + if err != nil { + suite.FailNow(err.Error()) + } + + if !muted { + suite.FailNow("", "expected thread %s to be muted by account %s", threadID, accountID) + } + + _, err = suite.db.GetThreadMutedByAccount(ctx, threadID, accountID) + if err != nil { + suite.FailNow(err.Error()) + } + + // Unmute the thread and ensure it's actually unmuted. + if err := suite.db.DeleteThreadMute(ctx, threadMute.ID); err != nil { + suite.FailNow(err.Error()) + } + + muted, err = suite.db.IsThreadMutedByAccount(ctx, threadID, accountID) + if err != nil { + suite.FailNow(err.Error()) + } + + if muted { + suite.FailNow("", "expected thread %s to not be muted by account %s", threadID, accountID) + } +} + +func TestThreadTestSuite(t *testing.T) { + suite.Run(t, new(ThreadTestSuite)) +} |