models/user: add function to get FederatedUser by local userID

This commit is contained in:
famfo 2025-02-28 17:02:50 +01:00
parent 1af31f86da
commit b29449c9c6
No known key found for this signature in database
4 changed files with 181 additions and 17 deletions

View file

@ -10,8 +10,9 @@ import (
"testing"
"forgejo.org/models/db"
"forgejo.org/models/forgefed"
"forgejo.org/models/unittest"
user_model "forgejo.org/models/user"
"forgejo.org/models/user"
"forgejo.org/modules/activitypub"
"forgejo.org/modules/setting"
"forgejo.org/modules/test"
@ -29,7 +30,7 @@ func TestFederationHttpSigValidation(t *testing.T) {
userID := 2
userURL := fmt.Sprintf("%vapi/v1/activitypub/user-id/%v", u, userID)
user1 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
user1 := unittest.AssertExistsAndLoadBean(t, &user.User{ID: 1})
clientFactory, err := activitypub.GetClientFactory(db.DefaultContext)
require.NoError(t, err)
@ -46,6 +47,23 @@ func TestFederationHttpSigValidation(t *testing.T) {
require.NoError(t, err)
assert.Equal(t, http.StatusOK, resp.StatusCode)
// HACK HACK HACK: the host part of the URL gets set to which IP forgejo is
// listening on, NOT localhost, which is the Domain given to forgejo which
// is then used for eg. the keyID all requests
applicationKeyID := fmt.Sprintf("http://localhost:%v/api/v1/activitypub/actor#main-key", u.Port())
actorKeyID := fmt.Sprintf("http://localhost:%v/api/v1/activitypub/user-id/1#main-key", u.Port())
// Check for cached public keys
host, err := forgefed.FindFederationHostByKeyID(db.DefaultContext, applicationKeyID)
require.NoError(t, err)
assert.NotNil(t, host)
assert.True(t, host.PublicKey.Valid)
user, err := user.GetFederatedUserByKeyID(db.DefaultContext, actorKeyID)
require.NoError(t, err)
assert.NotNil(t, user)
assert.True(t, user.PublicKey.Valid)
// Disable signature validation
defer test.MockVariableValue(&setting.Federation.SignatureEnforced, false)()