tests/integration: activitypub: test maximum response size limit

This commit is contained in:
famfo 2025-02-26 01:28:17 +01:00
parent c8968e54fb
commit a018da0378
No known key found for this signature in database

View file

@ -0,0 +1,50 @@
// Copyright 2025 The Forgejo Authors. All rights reserved.
// SPDX-License-Identifier: MIT
package integration
import (
"net/url"
"testing"
"forgejo.org/models/db"
"forgejo.org/models/unittest"
user_model "forgejo.org/models/user"
"forgejo.org/modules/activitypub"
"forgejo.org/modules/setting"
"forgejo.org/modules/test"
"forgejo.org/routers"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestActivityPubClientBodySize(t *testing.T) {
defer test.MockVariableValue(&setting.Federation.Enabled, true)()
defer test.MockVariableValue(&testWebRoutes, routers.NormalRoutes())()
onGiteaRun(t, func(t *testing.T, u *url.URL) {
user1 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
clientFactory, err := activitypub.GetClientFactory(db.DefaultContext)
require.NoError(t, err)
apClient, err := clientFactory.WithKeys(db.DefaultContext, user1, user1.APActorKeyID())
require.NoError(t, err)
url := u.JoinPath("/api/v1/nodeinfo").String()
// Request with normal MaxSize
resp, err := apClient.GetBody(url)
require.NoError(t, err)
assert.Contains(t, string(resp), "forgejo")
// Set MaxSize to something very low to always fail
defer test.MockVariableValue(&setting.Federation.MaxSize, 100)()
// Request with low MaxSize
_, err = apClient.GetBody(url)
require.Error(t, err)
assert.ErrorContains(t, err, "Request returned")
})
}