Fix senderID/key conversion unit tests

This commit is contained in:
Devon Hudson 2023-06-14 17:11:27 +01:00
parent 3f4df25b31
commit 8cf6c381e2
No known key found for this signature in database
GPG Key ID: CD06B18E77F6A628
2 changed files with 4 additions and 3 deletions

View File

@ -183,7 +183,7 @@ func TestUserRoomKeys(t *testing.T) {
assert.NoError(t, err)
wantKeys := map[spec.RoomID]map[string]string{
*roomID: {
string(key.Public().(ed25519.PublicKey)): userID.String(),
spec.Base64Bytes(key.Public().(ed25519.PublicKey)).Encode(): userID.String(),
},
}
assert.Equal(t, wantKeys, userIDs)

View File

@ -13,6 +13,7 @@ import (
"github.com/matrix-org/dendrite/roomserver/types"
"github.com/matrix-org/dendrite/setup/config"
"github.com/matrix-org/dendrite/test"
"github.com/matrix-org/gomatrixserverlib/spec"
"github.com/stretchr/testify/assert"
ed255192 "golang.org/x/crypto/ed25519"
)
@ -101,8 +102,8 @@ func TestUserRoomKeysTable(t *testing.T) {
assert.NotNil(t, gotKeys)
wantKeys := map[string]types.UserRoomKeyPair{
string(key2.Public().(ed25519.PublicKey)): {RoomNID: roomNID, EventStateKeyNID: userNID},
string(key3.Public().(ed25519.PublicKey)): {RoomNID: roomNID, EventStateKeyNID: userNID2},
string(spec.Base64Bytes(key2.Public().(ed25519.PublicKey)).Encode()): {RoomNID: roomNID, EventStateKeyNID: userNID},
string(spec.Base64Bytes(key3.Public().(ed25519.PublicKey)).Encode()): {RoomNID: roomNID, EventStateKeyNID: userNID2},
}
assert.Equal(t, wantKeys, gotKeys)