gtsocial-umbx

Unnamed repository; edit this file 'description' to name the repository.
Log | Files | Refs | README | LICENSE

mention_test.go (2113B)

      1 // GoToSocial
      2 // Copyright (C) GoToSocial Authors admin@gotosocial.org
      3 // SPDX-License-Identifier: AGPL-3.0-or-later
      4 //
      5 // This program is free software: you can redistribute it and/or modify
      6 // it under the terms of the GNU Affero General Public License as published by
      7 // the Free Software Foundation, either version 3 of the License, or
      8 // (at your option) any later version.
      9 //
     10 // This program is distributed in the hope that it will be useful,
     11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
     12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     13 // GNU Affero General Public License for more details.
     14 //
     15 // You should have received a copy of the GNU Affero General Public License
     16 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
     17 
     18 package bundb_test
     19 
     20 import (
     21 	"context"
     22 	"testing"
     23 
     24 	"github.com/stretchr/testify/suite"
     25 )
     26 
     27 type MentionTestSuite struct {
     28 	BunDBStandardTestSuite
     29 }
     30 
     31 func (suite *MentionTestSuite) TestGetMention() {
     32 	m := suite.testMentions["local_user_2_mention_zork"]
     33 
     34 	dbMention, err := suite.db.GetMention(context.Background(), m.ID)
     35 	suite.NoError(err)
     36 	suite.NotNil(dbMention)
     37 	suite.Equal(m.ID, dbMention.ID)
     38 	suite.Equal(m.OriginAccountID, dbMention.OriginAccountID)
     39 	suite.NotNil(dbMention.OriginAccount)
     40 	suite.Equal(m.TargetAccountID, dbMention.TargetAccountID)
     41 	suite.NotNil(dbMention.TargetAccount)
     42 	suite.Equal(m.StatusID, dbMention.StatusID)
     43 	suite.NotNil(dbMention.Status)
     44 }
     45 
     46 func (suite *MentionTestSuite) TestGetMentions() {
     47 	m := suite.testMentions["local_user_2_mention_zork"]
     48 
     49 	dbMentions, err := suite.db.GetMentions(context.Background(), []string{m.ID})
     50 	suite.NoError(err)
     51 	suite.Len(dbMentions, 1)
     52 	dbMention := dbMentions[0]
     53 	suite.Equal(m.ID, dbMention.ID)
     54 	suite.Equal(m.OriginAccountID, dbMention.OriginAccountID)
     55 	suite.NotNil(dbMention.OriginAccount)
     56 	suite.Equal(m.TargetAccountID, dbMention.TargetAccountID)
     57 	suite.NotNil(dbMention.TargetAccount)
     58 	suite.Equal(m.StatusID, dbMention.StatusID)
     59 	suite.NotNil(dbMention.Status)
     60 }
     61 
     62 func TestMentionTestSuite(t *testing.T) {
     63 	suite.Run(t, new(MentionTestSuite))
     64 }