aboutsummaryrefslogtreecommitdiff
path: root/roomserver
diff options
context:
space:
mode:
authorDevon Hudson <devonhudson@librem.one>2023-06-14 17:11:27 +0100
committerDevon Hudson <devonhudson@librem.one>2023-06-14 17:11:27 +0100
commit8cf6c381e21d0710f0290c97dfa5616036749a81 (patch)
tree822aaaf093b10c9b0c30b8dea4d2f324ea63d857 /roomserver
parent3f4df25b31a403e936488a1920d1aed3de471a71 (diff)
Fix senderID/key conversion unit tests
Diffstat (limited to 'roomserver')
-rw-r--r--roomserver/storage/shared/storage_test.go2
-rw-r--r--roomserver/storage/tables/user_room_keys_table_test.go5
2 files changed, 4 insertions, 3 deletions
diff --git a/roomserver/storage/shared/storage_test.go b/roomserver/storage/shared/storage_test.go
index c7b915c7..612e4ef0 100644
--- a/roomserver/storage/shared/storage_test.go
+++ b/roomserver/storage/shared/storage_test.go
@@ -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)
diff --git a/roomserver/storage/tables/user_room_keys_table_test.go b/roomserver/storage/tables/user_room_keys_table_test.go
index 8802a3c6..2809771b 100644
--- a/roomserver/storage/tables/user_room_keys_table_test.go
+++ b/roomserver/storage/tables/user_room_keys_table_test.go
@@ -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)