1 files changed, 7 insertions, 0 deletions
diff --git a/tests/events/__init__.py b/synapse/events/snapshot.py
index 9bff9ec169..b9fb29be01 100644
--- a/tests/events/__init__.py
+++ b/synapse/events/snapshot.py
@@ -13,3 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+
+class EventContext(object):
+
+    def __init__(self, current_state=None, auth_events=None):
+        self.current_state = current_state
+        self.auth_events = auth_events
+        self.state_group = None