mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-05 01:52:46 +00:00
aeb65bceae
* [feature/frontend] Better web threading model * fix test * bwap * tweaks * more tweaks to wording * typo * indenting * adjust wording * aaa
207 lines
5.8 KiB
Go
207 lines
5.8 KiB
Go
// 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 status_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
|
"github.com/superseriousbusiness/gotosocial/internal/processing/status"
|
|
)
|
|
|
|
type topoSortTestSuite struct {
|
|
suite.Suite
|
|
}
|
|
|
|
func statusIDs(apiStatuses []*gtsmodel.Status) []string {
|
|
ids := make([]string, 0, len(apiStatuses))
|
|
for _, apiStatus := range apiStatuses {
|
|
ids = append(ids, apiStatus.ID)
|
|
}
|
|
return ids
|
|
}
|
|
|
|
func (suite *topoSortTestSuite) TestBranched() {
|
|
// https://commons.wikimedia.org/wiki/File:Sorted_binary_tree_ALL_RGB.svg
|
|
f := >smodel.Status{ID: "F"}
|
|
b := >smodel.Status{ID: "B", InReplyToID: f.ID}
|
|
a := >smodel.Status{ID: "A", InReplyToID: b.ID}
|
|
d := >smodel.Status{ID: "D", InReplyToID: b.ID}
|
|
c := >smodel.Status{ID: "C", InReplyToID: d.ID}
|
|
e := >smodel.Status{ID: "E", InReplyToID: d.ID}
|
|
g := >smodel.Status{ID: "G", InReplyToID: f.ID}
|
|
i := >smodel.Status{ID: "I", InReplyToID: g.ID}
|
|
h := >smodel.Status{ID: "H", InReplyToID: i.ID}
|
|
|
|
expected := statusIDs([]*gtsmodel.Status{f, b, a, d, c, e, g, i, h})
|
|
list := []*gtsmodel.Status{a, b, c, d, e, f, g, h, i}
|
|
status.TopoSort(list, "")
|
|
actual := statusIDs(list)
|
|
|
|
suite.Equal(expected, actual)
|
|
}
|
|
|
|
func (suite *topoSortTestSuite) TestBranchedWithSelfReplyChain() {
|
|
targetAccount := >smodel.Account{ID: "1"}
|
|
otherAccount := >smodel.Account{ID: "2"}
|
|
|
|
f := >smodel.Status{
|
|
ID: "F",
|
|
Account: targetAccount,
|
|
}
|
|
b := >smodel.Status{
|
|
ID: "B",
|
|
Account: targetAccount,
|
|
AccountID: targetAccount.ID,
|
|
InReplyToID: f.ID,
|
|
InReplyToAccountID: f.Account.ID,
|
|
}
|
|
d := >smodel.Status{
|
|
ID: "D",
|
|
Account: targetAccount,
|
|
AccountID: targetAccount.ID,
|
|
InReplyToID: b.ID,
|
|
InReplyToAccountID: b.Account.ID,
|
|
}
|
|
e := >smodel.Status{
|
|
ID: "E",
|
|
Account: targetAccount,
|
|
AccountID: targetAccount.ID,
|
|
InReplyToID: d.ID,
|
|
InReplyToAccountID: d.Account.ID,
|
|
}
|
|
c := >smodel.Status{
|
|
ID: "C",
|
|
Account: otherAccount,
|
|
AccountID: otherAccount.ID,
|
|
InReplyToID: d.ID,
|
|
InReplyToAccountID: d.Account.ID,
|
|
}
|
|
a := >smodel.Status{
|
|
ID: "A",
|
|
Account: otherAccount,
|
|
AccountID: otherAccount.ID,
|
|
InReplyToID: b.ID,
|
|
InReplyToAccountID: b.Account.ID,
|
|
}
|
|
g := >smodel.Status{
|
|
ID: "G",
|
|
Account: otherAccount,
|
|
AccountID: otherAccount.ID,
|
|
InReplyToID: f.ID,
|
|
InReplyToAccountID: f.Account.ID,
|
|
}
|
|
i := >smodel.Status{
|
|
ID: "I",
|
|
Account: targetAccount,
|
|
AccountID: targetAccount.ID,
|
|
InReplyToID: g.ID,
|
|
InReplyToAccountID: g.Account.ID,
|
|
}
|
|
h := >smodel.Status{
|
|
ID: "H",
|
|
Account: otherAccount,
|
|
AccountID: otherAccount.ID,
|
|
InReplyToID: i.ID,
|
|
InReplyToAccountID: i.Account.ID,
|
|
}
|
|
|
|
expected := statusIDs([]*gtsmodel.Status{f, b, d, e, c, a, g, i, h})
|
|
list := []*gtsmodel.Status{a, b, c, d, e, f, g, h, i}
|
|
status.TopoSort(list, targetAccount.ID)
|
|
actual := statusIDs(list)
|
|
|
|
suite.Equal(expected, actual)
|
|
}
|
|
|
|
func (suite *topoSortTestSuite) TestDisconnected() {
|
|
f := >smodel.Status{ID: "F"}
|
|
b := >smodel.Status{ID: "B", InReplyToID: f.ID}
|
|
dID := "D"
|
|
e := >smodel.Status{ID: "E", InReplyToID: dID}
|
|
|
|
expected := statusIDs([]*gtsmodel.Status{e, f, b})
|
|
list := []*gtsmodel.Status{b, e, f}
|
|
status.TopoSort(list, "")
|
|
actual := statusIDs(list)
|
|
|
|
suite.Equal(expected, actual)
|
|
}
|
|
|
|
func (suite *topoSortTestSuite) TestTrivialCycle() {
|
|
xID := "X"
|
|
x := >smodel.Status{ID: xID, InReplyToID: xID}
|
|
|
|
expected := statusIDs([]*gtsmodel.Status{x})
|
|
list := []*gtsmodel.Status{x}
|
|
status.TopoSort(list, "")
|
|
actual := statusIDs(list)
|
|
|
|
suite.ElementsMatch(expected, actual)
|
|
}
|
|
|
|
func (suite *topoSortTestSuite) TestCycle() {
|
|
yID := "Y"
|
|
x := >smodel.Status{ID: "X", InReplyToID: yID}
|
|
y := >smodel.Status{ID: yID, InReplyToID: x.ID}
|
|
|
|
expected := statusIDs([]*gtsmodel.Status{x, y})
|
|
list := []*gtsmodel.Status{x, y}
|
|
status.TopoSort(list, "")
|
|
actual := statusIDs(list)
|
|
|
|
suite.ElementsMatch(expected, actual)
|
|
}
|
|
|
|
func (suite *topoSortTestSuite) TestMixedCycle() {
|
|
yID := "Y"
|
|
x := >smodel.Status{ID: "X", InReplyToID: yID}
|
|
y := >smodel.Status{ID: yID, InReplyToID: x.ID}
|
|
z := >smodel.Status{ID: "Z"}
|
|
|
|
expected := statusIDs([]*gtsmodel.Status{x, y, z})
|
|
list := []*gtsmodel.Status{x, y, z}
|
|
status.TopoSort(list, "")
|
|
actual := statusIDs(list)
|
|
|
|
suite.ElementsMatch(expected, actual)
|
|
}
|
|
|
|
func (suite *topoSortTestSuite) TestEmpty() {
|
|
expected := statusIDs([]*gtsmodel.Status{})
|
|
list := []*gtsmodel.Status{}
|
|
status.TopoSort(list, "")
|
|
actual := statusIDs(list)
|
|
|
|
suite.Equal(expected, actual)
|
|
}
|
|
|
|
func (suite *topoSortTestSuite) TestNil() {
|
|
expected := statusIDs(nil)
|
|
var list []*gtsmodel.Status
|
|
status.TopoSort(list, "")
|
|
actual := statusIDs(list)
|
|
|
|
suite.Equal(expected, actual)
|
|
}
|
|
|
|
func TestTopoSortTestSuite(t *testing.T) {
|
|
suite.Run(t, &topoSortTestSuite{})
|
|
}
|