2021-04-19 20:42:19 +03:00
/ *
GoToSocial
Copyright ( C ) 2021 GoToSocial Authors admin @ gotosocial . org
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
import (
"encoding/json"
"fmt"
"io/ioutil"
"net/http"
"net/http/httptest"
"strings"
"testing"
"github.com/gin-gonic/gin"
"github.com/sirupsen/logrus"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/suite"
"github.com/superseriousbusiness/gotosocial/internal/apimodule/status"
"github.com/superseriousbusiness/gotosocial/internal/config"
"github.com/superseriousbusiness/gotosocial/internal/db"
"github.com/superseriousbusiness/gotosocial/internal/db/gtsmodel"
"github.com/superseriousbusiness/gotosocial/internal/distributor"
"github.com/superseriousbusiness/gotosocial/internal/mastotypes"
mastomodel "github.com/superseriousbusiness/gotosocial/internal/mastotypes/mastomodel"
"github.com/superseriousbusiness/gotosocial/internal/media"
"github.com/superseriousbusiness/gotosocial/internal/oauth"
"github.com/superseriousbusiness/gotosocial/internal/storage"
"github.com/superseriousbusiness/gotosocial/testrig"
)
type StatusFavedByTestSuite struct {
// standard suite interfaces
suite . Suite
config * config . Config
db db . DB
log * logrus . Logger
storage storage . Storage
mastoConverter mastotypes . Converter
2021-04-20 19:14:23 +03:00
mediaHandler media . Handler
2021-04-19 20:42:19 +03:00
oauthServer oauth . Server
distributor distributor . Distributor
// standard suite models
testTokens map [ string ] * oauth . Token
testClients map [ string ] * oauth . Client
testApplications map [ string ] * gtsmodel . Application
testUsers map [ string ] * gtsmodel . User
testAccounts map [ string ] * gtsmodel . Account
testAttachments map [ string ] * gtsmodel . MediaAttachment
testStatuses map [ string ] * gtsmodel . Status
// module being tested
2021-04-20 19:14:23 +03:00
statusModule * status . Module
2021-04-19 20:42:19 +03:00
}
// SetupSuite sets some variables on the suite that we can use as consts (more or less) throughout
func ( suite * StatusFavedByTestSuite ) SetupSuite ( ) {
// setup standard items
suite . config = testrig . NewTestConfig ( )
suite . db = testrig . NewTestDB ( )
suite . log = testrig . NewTestLog ( )
suite . storage = testrig . NewTestStorage ( )
suite . mastoConverter = testrig . NewTestMastoConverter ( suite . db )
suite . mediaHandler = testrig . NewTestMediaHandler ( suite . db , suite . storage )
suite . oauthServer = testrig . NewTestOauthServer ( suite . db )
suite . distributor = testrig . NewTestDistributor ( )
// setup module being tested
2021-04-20 19:14:23 +03:00
suite . statusModule = status . New ( suite . config , suite . db , suite . mediaHandler , suite . mastoConverter , suite . distributor , suite . log ) . ( * status . Module )
2021-04-19 20:42:19 +03:00
}
func ( suite * StatusFavedByTestSuite ) TearDownSuite ( ) {
testrig . StandardDBTeardown ( suite . db )
testrig . StandardStorageTeardown ( suite . storage )
}
func ( suite * StatusFavedByTestSuite ) SetupTest ( ) {
testrig . StandardDBSetup ( suite . db )
testrig . StandardStorageSetup ( suite . storage , "../../../../testrig/media" )
suite . testTokens = testrig . NewTestTokens ( )
suite . testClients = testrig . NewTestClients ( )
suite . testApplications = testrig . NewTestApplications ( )
suite . testUsers = testrig . NewTestUsers ( )
suite . testAccounts = testrig . NewTestAccounts ( )
suite . testAttachments = testrig . NewTestAttachments ( )
suite . testStatuses = testrig . NewTestStatuses ( )
}
// TearDownTest drops tables to make sure there's no data in the db
func ( suite * StatusFavedByTestSuite ) TearDownTest ( ) {
testrig . StandardDBTeardown ( suite . db )
testrig . StandardStorageTeardown ( suite . storage )
}
/ *
ACTUAL TESTS
* /
func ( suite * StatusFavedByTestSuite ) TestGetFavedBy ( ) {
t := suite . testTokens [ "local_account_2" ]
2021-04-20 19:14:23 +03:00
oauthToken := oauth . TokenToOauthToken ( t )
2021-04-19 20:42:19 +03:00
targetStatus := suite . testStatuses [ "admin_account_status_1" ] // this status is faved by local_account_1
// setup
recorder := httptest . NewRecorder ( )
ctx , _ := gin . CreateTestContext ( recorder )
ctx . Set ( oauth . SessionAuthorizedApplication , suite . testApplications [ "application_2" ] )
ctx . Set ( oauth . SessionAuthorizedToken , oauthToken )
ctx . Set ( oauth . SessionAuthorizedUser , suite . testUsers [ "local_account_2" ] )
ctx . Set ( oauth . SessionAuthorizedAccount , suite . testAccounts [ "local_account_2" ] )
ctx . Request = httptest . NewRequest ( http . MethodPost , fmt . Sprintf ( "http://localhost:8080%s" , strings . Replace ( status . FavouritedPath , ":id" , targetStatus . ID , 1 ) ) , nil ) // the endpoint we're hitting
// normally the router would populate these params from the path values,
// but because we're calling the function directly, we need to set them manually.
ctx . Params = gin . Params {
gin . Param {
Key : status . IDKey ,
Value : targetStatus . ID ,
} ,
}
suite . statusModule . StatusFavedByGETHandler ( ctx )
// check response
suite . EqualValues ( http . StatusOK , recorder . Code )
result := recorder . Result ( )
defer result . Body . Close ( )
b , err := ioutil . ReadAll ( result . Body )
assert . NoError ( suite . T ( ) , err )
accts := [ ] mastomodel . Account { }
err = json . Unmarshal ( b , & accts )
assert . NoError ( suite . T ( ) , err )
assert . Len ( suite . T ( ) , accts , 1 )
assert . Equal ( suite . T ( ) , "the_mighty_zork" , accts [ 0 ] . Username )
}
func TestStatusFavedByTestSuite ( t * testing . T ) {
suite . Run ( t , new ( StatusFavedByTestSuite ) )
}