From a202d88fe527c83179d2ebf57ac0d146c3173f9d Mon Sep 17 00:00:00 2001
From: Kegsay <kegan@matrix.org>
Date: Fri, 24 Apr 2020 10:38:58 +0100
Subject: Use a single storage.Database interface (#978)

---
 roomserver/query/query_test.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'roomserver/query/query_test.go')

diff --git a/roomserver/query/query_test.go b/roomserver/query/query_test.go
index 7e040c6f..8fb6a082 100644
--- a/roomserver/query/query_test.go
+++ b/roomserver/query/query_test.go
@@ -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, []string{"e"})
+	result, err := getAuthChain(context.TODO(), db.EventsFromIDs, []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, []string{"e", "f"})
+	result, err := getAuthChain(context.TODO(), db.EventsFromIDs, []string{"e", "f"})
 	if err != nil {
 		t.Fatalf("getAuthChain failed: %v", err)
 	}
-- 
cgit v1.2.3