mirror of https://github.com/mautrix/go.git
114 lines
3.1 KiB
Go
114 lines
3.1 KiB
Go
// Copyright (c) 2022 Tulir Asokan
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package mautrix_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"maunium.net/go/mautrix"
|
|
"maunium.net/go/mautrix/crypto/canonicaljson"
|
|
)
|
|
|
|
const sampleData = `{
|
|
"capabilities": {
|
|
"m.room_versions": {
|
|
"default": "9",
|
|
"available": {
|
|
"1": "stable",
|
|
"2": "stable",
|
|
"3": "stable",
|
|
"4": "stable",
|
|
"5": "stable",
|
|
"6": "stable",
|
|
"org.matrix.msc2176": "unstable",
|
|
"7": "stable",
|
|
"8": "stable",
|
|
"9": "stable",
|
|
"org.matrix.msc2716v3": "unstable",
|
|
"org.matrix.msc3787": "unstable",
|
|
"10": "stable"
|
|
}
|
|
},
|
|
"m.change_password": {
|
|
"enabled": true
|
|
},
|
|
"m.set_displayname": {
|
|
"enabled": true
|
|
},
|
|
"m.3pid_changes": {
|
|
"enabled": false
|
|
},
|
|
"fi.mau.custom_field": {
|
|
"🐈️": true
|
|
}
|
|
}
|
|
}`
|
|
|
|
var sampleObject = mautrix.RespCapabilities{
|
|
RoomVersions: &mautrix.CapRoomVersions{
|
|
Default: "9",
|
|
Available: map[string]mautrix.CapRoomVersionStability{
|
|
"1": "stable",
|
|
"2": "stable",
|
|
"3": "stable",
|
|
"4": "stable",
|
|
"5": "stable",
|
|
"6": "stable",
|
|
"org.matrix.msc2176": "unstable",
|
|
"7": "stable",
|
|
"8": "stable",
|
|
"9": "stable",
|
|
"org.matrix.msc2716v3": "unstable",
|
|
"org.matrix.msc3787": "unstable",
|
|
"10": "stable",
|
|
},
|
|
},
|
|
ChangePassword: &mautrix.CapBooleanTrue{Enabled: true},
|
|
SetDisplayname: &mautrix.CapBooleanTrue{Enabled: true},
|
|
ThreePIDChanges: &mautrix.CapBooleanTrue{Enabled: false},
|
|
Custom: map[string]interface{}{
|
|
"fi.mau.custom_field": map[string]interface{}{
|
|
"🐈️": true,
|
|
},
|
|
},
|
|
}
|
|
|
|
func TestRespCapabilities_UnmarshalJSON(t *testing.T) {
|
|
var caps mautrix.RespCapabilities
|
|
err := json.Unmarshal([]byte(sampleData), &caps)
|
|
require.NoError(t, err)
|
|
fmt.Println(caps)
|
|
|
|
require.NotNil(t, caps.RoomVersions)
|
|
assert.Equal(t, "9", caps.RoomVersions.Default)
|
|
assert.True(t, caps.RoomVersions.IsStable("10"))
|
|
|
|
// Omitted capabilities still support IsEnabled(), and this one defaults to true
|
|
assert.Nil(t, caps.SetAvatarURL)
|
|
assert.True(t, caps.SetAvatarURL.IsEnabled())
|
|
|
|
assert.True(t, caps.SetDisplayname.IsEnabled())
|
|
assert.False(t, caps.ThreePIDChanges.IsEnabled())
|
|
|
|
assert.Contains(t, caps.Custom, "fi.mau.custom_field")
|
|
assert.NotContains(t, caps.Custom, "m.room_versions")
|
|
}
|
|
|
|
func TestRespCapabilities_MarshalJSON(t *testing.T) {
|
|
data, err := json.Marshal(&sampleObject)
|
|
require.NoError(t, err)
|
|
marshaledString := string(canonicaljson.CanonicalJSONAssumeValid(data))
|
|
origString := string(canonicaljson.CanonicalJSONAssumeValid([]byte(sampleData)))
|
|
assert.Equal(t, marshaledString, origString)
|
|
assert.Len(t, sampleObject.Custom, 1)
|
|
}
|