package util_test import ( "context" "encoding/json" "net/http" "net/http/httptest" "testing" "time" "github.com/matrix-org/gomatrixserverlib" "github.com/matrix-org/util" "golang.org/x/crypto/bcrypt" "github.com/matrix-org/dendrite/internal/pushgateway" "github.com/matrix-org/dendrite/setup/config" "github.com/matrix-org/dendrite/test" "github.com/matrix-org/dendrite/test/testrig" "github.com/matrix-org/dendrite/userapi/api" "github.com/matrix-org/dendrite/userapi/storage" userUtil "github.com/matrix-org/dendrite/userapi/util" ) func TestNotifyUserCountsAsync(t *testing.T) { alice := test.NewUser(t) aliceLocalpart, serverName, err := gomatrixserverlib.SplitID('@', alice.ID) if err != nil { t.Error(err) } ctx := context.Background() // Create a test room, just used to provide events room := test.NewRoom(t, alice) dummyEvent := room.Events()[len(room.Events())-1] appID := util.RandomString(8) pushKey := util.RandomString(8) test.WithAllDatabases(t, func(t *testing.T, dbType test.DBType) { receivedRequest := make(chan bool, 1) // create a test server which responds to our /notify call srv := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { var data pushgateway.NotifyRequest if err := json.NewDecoder(r.Body).Decode(&data); err != nil { t.Error(err) } notification := data.Notification // Validate the request if notification.Counts == nil { t.Fatal("no unread notification counts in request") } if unread := notification.Counts.Unread; unread != 1 { t.Errorf("expected one unread notification, got %d", unread) } if len(notification.Devices) == 0 { t.Fatal("expected devices in request") } // We only created one push device, so access it directly device := notification.Devices[0] if device.AppID != appID { t.Errorf("unexpected app_id: %s, want %s", device.AppID, appID) } if device.PushKey != pushKey { t.Errorf("unexpected push_key: %s, want %s", device.PushKey, pushKey) } // Return empty result, otherwise the call is handled as failed if _, err := w.Write([]byte("{}")); err != nil { t.Error(err) } close(receivedRequest) })) defer srv.Close() // Create DB and Dendrite base connStr, close := test.PrepareDBConnectionString(t, dbType) defer close() base, _, _ := testrig.Base(nil) defer base.Close() db, err := storage.NewUserAPIDatabase(base, &config.DatabaseOptions{ ConnectionString: config.DataSource(connStr), }, "test", bcrypt.MinCost, 0, 0, "") if err != nil { t.Error(err) } // Prepare pusher with our test server URL if err := db.UpsertPusher(ctx, api.Pusher{ Kind: api.HTTPKind, AppID: appID, PushKey: pushKey, Data: map[string]interface{}{ "url": srv.URL, }, }, aliceLocalpart, serverName); err != nil { t.Error(err) } // Insert a dummy event if err := db.InsertNotification(ctx, aliceLocalpart, serverName, dummyEvent.EventID(), 0, nil, &api.Notification{ Event: gomatrixserverlib.HeaderedToClientEvent(dummyEvent, gomatrixserverlib.FormatAll), }); err != nil { t.Error(err) } // Notify the user about a new notification if err := userUtil.NotifyUserCountsAsync(ctx, pushgateway.NewHTTPClient(true), aliceLocalpart, serverName, db); err != nil { t.Error(err) } select { case <-time.After(time.Second * 5): t.Error("timed out waiting for response") case <-receivedRequest: } }) }