aboutsummaryrefslogtreecommitdiff
path: root/roomserver/internal/query/query_test.go
diff options
context:
space:
mode:
authorTill <2353100+S7evinK@users.noreply.github.com>2023-03-01 17:06:47 +0100
committerGitHub <noreply@github.com>2023-03-01 17:06:47 +0100
commit6c20f8f742a7e03710fae81df6ef98bac31da2b1 (patch)
tree202e962951dc41c949a71c7f5c1deb6d1da78843 /roomserver/internal/query/query_test.go
parent1aa70b0f56825a4a5f92c38cabb1fe841cec6e18 (diff)
Refactor `StoreEvent`, add `MaybeRedactEvent`, create an `EventDatabase` (#2989)
This PR changes the following: - `StoreEvent` now only stores an event (and possibly prev event), instead of also doing redactions - Adds a `MaybeRedactEvent` (pulled out from `StoreEvent`), which should be called after storing events - a few other things
Diffstat (limited to 'roomserver/internal/query/query_test.go')
-rw-r--r--roomserver/internal/query/query_test.go6
1 files changed, 3 insertions, 3 deletions
diff --git a/roomserver/internal/query/query_test.go b/roomserver/internal/query/query_test.go
index 16761157..265f326d 100644
--- a/roomserver/internal/query/query_test.go
+++ b/roomserver/internal/query/query_test.go
@@ -80,7 +80,7 @@ func (db *getEventDB) addFakeEvents(graph map[string][]string) error {
}
// EventsFromIDs implements RoomserverInternalAPIEventDB
-func (db *getEventDB) EventsFromIDs(ctx context.Context, roomNID types.RoomNID, eventIDs []string) (res []types.Event, err error) {
+func (db *getEventDB) EventsFromIDs(ctx context.Context, roomInfo *types.RoomInfo, eventIDs []string) (res []types.Event, err error) {
for _, evID := range eventIDs {
res = append(res, types.Event{
EventNID: 0,
@@ -106,7 +106,7 @@ func TestGetAuthChainSingle(t *testing.T) {
t.Fatalf("Failed to add events to db: %v", err)
}
- result, err := GetAuthChain(context.TODO(), db.EventsFromIDs, []string{"e"})
+ result, err := GetAuthChain(context.TODO(), db.EventsFromIDs, nil, []string{"e"})
if err != nil {
t.Fatalf("getAuthChain failed: %v", err)
}
@@ -139,7 +139,7 @@ func TestGetAuthChainMultiple(t *testing.T) {
t.Fatalf("Failed to add events to db: %v", err)
}
- result, err := GetAuthChain(context.TODO(), db.EventsFromIDs, []string{"e", "f"})
+ result, err := GetAuthChain(context.TODO(), db.EventsFromIDs, nil, []string{"e", "f"})
if err != nil {
t.Fatalf("getAuthChain failed: %v", err)
}