141 lines
4.2 KiB
Go
141 lines
4.2 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 bundb_test
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"crypto/rsa"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/superseriousbusiness/gotosocial/internal/db"
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
|
)
|
|
|
|
type BasicTestSuite struct {
|
|
BunDBStandardTestSuite
|
|
}
|
|
|
|
func (suite *BasicTestSuite) TestGetAccountByID() {
|
|
testAccount := suite.testAccounts["local_account_1"]
|
|
|
|
a := >smodel.Account{}
|
|
err := suite.db.GetByID(context.Background(), testAccount.ID, a)
|
|
suite.NoError(err)
|
|
}
|
|
|
|
func (suite *BasicTestSuite) TestPutAccountWithBunDefaultFields() {
|
|
key, err := rsa.GenerateKey(rand.Reader, 2048)
|
|
if err != nil {
|
|
suite.FailNow(err.Error())
|
|
}
|
|
|
|
// Create an account that only just matches constraints.
|
|
testAccount := >smodel.Account{
|
|
ID: "01GADR1AH9VCKH8YYCM86XSZ00",
|
|
Username: "test",
|
|
URI: "https://example.org/users/test",
|
|
URL: "https://example.org/@test",
|
|
InboxURI: "https://example.org/users/test/inbox",
|
|
OutboxURI: "https://example.org/users/test/outbox",
|
|
ActorType: "Person",
|
|
PublicKeyURI: "https://example.org/test#main-key",
|
|
PublicKey: &key.PublicKey,
|
|
}
|
|
|
|
if err := suite.db.Put(context.Background(), testAccount); err != nil {
|
|
suite.FailNow(err.Error())
|
|
}
|
|
|
|
a := >smodel.Account{}
|
|
if err := suite.db.GetByID(context.Background(), testAccount.ID, a); err != nil {
|
|
suite.FailNow(err.Error())
|
|
}
|
|
|
|
// check all fields are set as expected, including database defaults
|
|
suite.Equal(testAccount.ID, a.ID)
|
|
suite.WithinDuration(time.Now(), a.CreatedAt, 5*time.Second)
|
|
suite.WithinDuration(time.Now(), a.UpdatedAt, 5*time.Second)
|
|
suite.Equal(testAccount.Username, a.Username)
|
|
suite.Empty(a.Domain)
|
|
suite.Empty(a.AvatarMediaAttachmentID)
|
|
suite.Nil(a.AvatarMediaAttachment)
|
|
suite.Empty(a.AvatarRemoteURL)
|
|
suite.Empty(a.HeaderMediaAttachmentID)
|
|
suite.Nil(a.HeaderMediaAttachment)
|
|
suite.Empty(a.HeaderRemoteURL)
|
|
suite.Empty(a.DisplayName)
|
|
suite.Nil(a.Fields)
|
|
suite.Empty(a.Note)
|
|
suite.Empty(a.NoteRaw)
|
|
suite.Empty(a.AlsoKnownAsURIs)
|
|
suite.Empty(a.MovedToURI)
|
|
suite.False(*a.Bot)
|
|
// Locked is especially important, since it's a bool that defaults
|
|
// to true, which is why we use pointers for bools in the first place
|
|
suite.True(*a.Locked)
|
|
suite.False(*a.Discoverable)
|
|
suite.Equal(testAccount.URI, a.URI)
|
|
suite.Equal(testAccount.URL, a.URL)
|
|
suite.Zero(testAccount.FetchedAt)
|
|
suite.Equal(testAccount.InboxURI, a.InboxURI)
|
|
suite.Equal(testAccount.OutboxURI, a.OutboxURI)
|
|
suite.Empty(a.FollowingURI)
|
|
suite.Empty(a.FollowersURI)
|
|
suite.Empty(a.FeaturedCollectionURI)
|
|
suite.Equal(testAccount.ActorType, a.ActorType)
|
|
suite.Nil(a.PrivateKey)
|
|
suite.EqualValues(key.PublicKey, *a.PublicKey)
|
|
suite.Equal(testAccount.PublicKeyURI, a.PublicKeyURI)
|
|
suite.Zero(a.SensitizedAt)
|
|
suite.Zero(a.SilencedAt)
|
|
suite.Zero(a.SuspendedAt)
|
|
suite.Empty(a.SuspensionOrigin)
|
|
}
|
|
|
|
func (suite *BasicTestSuite) TestGetAllStatuses() {
|
|
s := []*gtsmodel.Status{}
|
|
err := suite.db.GetAll(context.Background(), &s)
|
|
suite.NoError(err)
|
|
suite.Len(s, 23)
|
|
}
|
|
|
|
func (suite *BasicTestSuite) TestGetAllNotNull() {
|
|
where := []db.Where{{
|
|
Key: "domain",
|
|
Value: nil,
|
|
Not: true,
|
|
}}
|
|
|
|
a := []*gtsmodel.Account{}
|
|
|
|
err := suite.db.GetWhere(context.Background(), where, &a)
|
|
suite.NoError(err)
|
|
suite.NotEmpty(a)
|
|
|
|
for _, acct := range a {
|
|
suite.NotEmpty(acct.Domain)
|
|
}
|
|
}
|
|
|
|
func TestBasicTestSuite(t *testing.T) {
|
|
suite.Run(t, new(BasicTestSuite))
|
|
}
|