complement/tests/federation_presence_test.go
kegsay a669c750c8
Remove Deployment.Client and change Deploy (#676)
* Remove Deployment.Client and change Deploy

- `Deployment.Client` was used to get pre-registered clients. Now we want tests to register new users for each test, for dirty runs. So swap for `Deployment.Register` everywhere.
- `Deploy` was used to deploy a blueprint. We don't want this to enable dirty runs. So replace it with the number of servers you need e.g `Deploy(t, 2)`.

* Fix up more broken refactoring

* unbreak tests; make user localpart look nicer

* Alice and bob must share a room for presence

* Fix user directory test

* Fix race condition caused by making the room later than before
2023-10-17 18:07:43 +01:00

57 lines
2 KiB
Go

package tests
import (
"testing"
"github.com/tidwall/gjson"
"github.com/matrix-org/complement"
"github.com/matrix-org/complement/b"
"github.com/matrix-org/complement/client"
"github.com/matrix-org/complement/helpers"
)
func TestRemotePresence(t *testing.T) {
deployment := complement.Deploy(t, 2)
defer deployment.Destroy(t)
alice := deployment.Register(t, "hs1", helpers.RegistrationOpts{LocalpartSuffix: "alice"})
bob := deployment.Register(t, "hs2", helpers.RegistrationOpts{LocalpartSuffix: "bob"})
// for presence to be sent over federation alice and bob must share a room
roomID := alice.MustCreateRoom(t, map[string]interface{}{"preset": "public_chat"})
bob.MustJoinRoom(t, roomID, []string{"hs1"})
// sytest: Presence changes are also reported to remote room members
t.Run("Presence changes are also reported to remote room members", func(t *testing.T) {
_, bobSinceToken := bob.MustSync(t, client.SyncReq{TimeoutMillis: "0"})
statusMsg := "Update for room members"
alice.MustDo(t, "PUT", []string{"_matrix", "client", "v3", "presence", alice.UserID, "status"},
client.WithJSONBody(t, map[string]interface{}{
"status_msg": statusMsg,
"presence": "online",
}),
)
bob.MustSyncUntil(t, client.SyncReq{Since: bobSinceToken},
client.SyncPresenceHas(alice.UserID, b.Ptr("online"), func(ev gjson.Result) bool {
return ev.Get("content.status_msg").Str == statusMsg
}),
)
})
// sytest: Presence changes to UNAVAILABLE are reported to remote room members
t.Run("Presence changes to UNAVAILABLE are reported to remote room members", func(t *testing.T) {
_, bobSinceToken := bob.MustSync(t, client.SyncReq{TimeoutMillis: "0"})
alice.MustDo(t, "PUT", []string{"_matrix", "client", "v3", "presence", alice.UserID, "status"},
client.WithJSONBody(t, map[string]interface{}{
"presence": "unavailable",
}),
)
bob.MustSyncUntil(t, client.SyncReq{Since: bobSinceToken},
client.SyncPresenceHas(alice.UserID, b.Ptr("unavailable")),
)
})
}