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) assert.NoError(t, err)
wantKeys := map[spec.RoomID]map[string]string{ wantKeys := map[spec.RoomID]map[string]string{
*roomID: { *roomID: {
string(key.Public().(ed25519.PublicKey)): userID.String(), spec.Base64Bytes(key.Public().(ed25519.PublicKey)).Encode(): userID.String(),
}, },
} }
assert.Equal(t, wantKeys, userIDs) assert.Equal(t, wantKeys, userIDs)

View File

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