Add v2 room version

This commit is contained in:
Erik Johnston 2018-12-18 18:10:37 +00:00
parent 678ad155a2
commit df89f8afb8
2 changed files with 5 additions and 1 deletions

View File

@ -102,6 +102,7 @@ class ThirdPartyEntityKind(object):
class RoomVersions(object): class RoomVersions(object):
V1 = "1" V1 = "1"
V2 = "2"
VDH_TEST = "vdh-test-version" VDH_TEST = "vdh-test-version"
STATE_V2_TEST = "state-v2-test" STATE_V2_TEST = "state-v2-test"
@ -113,6 +114,7 @@ DEFAULT_ROOM_VERSION = RoomVersions.V1
# until we have a working v2. # until we have a working v2.
KNOWN_ROOM_VERSIONS = { KNOWN_ROOM_VERSIONS = {
RoomVersions.V1, RoomVersions.V1,
RoomVersions.V2,
RoomVersions.VDH_TEST, RoomVersions.VDH_TEST,
RoomVersions.STATE_V2_TEST, RoomVersions.STATE_V2_TEST,
} }

View File

@ -607,7 +607,9 @@ def resolve_events_with_store(room_version, state_sets, event_map, state_res_sto
return v1.resolve_events_with_store( return v1.resolve_events_with_store(
state_sets, event_map, state_res_store.get_events, state_sets, event_map, state_res_store.get_events,
) )
elif room_version in (RoomVersions.VDH_TEST, RoomVersions.STATE_V2_TEST): elif room_version in (
RoomVersions.VDH_TEST, RoomVersions.STATE_V2_TEST, RoomVersions.V2,
):
return v2.resolve_events_with_store( return v2.resolve_events_with_store(
state_sets, event_map, state_res_store, state_sets, event_map, state_res_store,
) )