Skip to content
Snippets Groups Projects
Commit b86d05a2 authored by Erik Johnston's avatar Erik Johnston
Browse files

Clean up event accesses and tests

This is in preparation to refactor FrozenEvent to support different
event formats for different room versions
parent 1b21e771
No related branches found
No related tags found
No related merge requests found
...@@ -117,9 +117,6 @@ class Transaction(JsonEncodedObject): ...@@ -117,9 +117,6 @@ class Transaction(JsonEncodedObject):
"Require 'transaction_id' to construct a Transaction" "Require 'transaction_id' to construct a Transaction"
) )
for p in pdus:
p.transaction_id = kwargs["transaction_id"]
kwargs["pdus"] = [p.get_pdu_json() for p in pdus] kwargs["pdus"] = [p.get_pdu_json() for p in pdus]
return Transaction(**kwargs) return Transaction(**kwargs)
...@@ -311,10 +311,10 @@ class HttpPusher(object): ...@@ -311,10 +311,10 @@ class HttpPusher(object):
] ]
} }
} }
if event.type == 'm.room.member': if event.type == 'm.room.member' and event.is_state():
d['notification']['membership'] = event.content['membership'] d['notification']['membership'] = event.content['membership']
d['notification']['user_is_target'] = event.state_key == self.user_id d['notification']['user_is_target'] = event.state_key == self.user_id
if self.hs.config.push_include_content and 'content' in event: if self.hs.config.push_include_content and event.content:
d['notification']['content'] = event.content d['notification']['content'] = event.content
# We no longer send aliases separately, instead, we send the human # We no longer send aliases separately, instead, we send the human
......
...@@ -124,7 +124,7 @@ class PushRuleEvaluatorForEvent(object): ...@@ -124,7 +124,7 @@ class PushRuleEvaluatorForEvent(object):
# XXX: optimisation: cache our pattern regexps # XXX: optimisation: cache our pattern regexps
if condition['key'] == 'content.body': if condition['key'] == 'content.body':
body = self._event["content"].get("body", None) body = self._event.content.get("body", None)
if not body: if not body:
return False return False
...@@ -140,7 +140,7 @@ class PushRuleEvaluatorForEvent(object): ...@@ -140,7 +140,7 @@ class PushRuleEvaluatorForEvent(object):
if not display_name: if not display_name:
return False return False
body = self._event["content"].get("body", None) body = self._event.content.get("body", None)
if not body: if not body:
return False return False
......
...@@ -28,8 +28,8 @@ ROOM_ID = "!room:blue" ...@@ -28,8 +28,8 @@ ROOM_ID = "!room:blue"
def dict_equals(self, other): def dict_equals(self, other):
me = encode_canonical_json(self._event_dict) me = encode_canonical_json(self.get_pdu_json())
them = encode_canonical_json(other._event_dict) them = encode_canonical_json(other.get_pdu_json())
return me == them return me == them
......
...@@ -112,7 +112,7 @@ class MessageAcceptTests(unittest.TestCase): ...@@ -112,7 +112,7 @@ class MessageAcceptTests(unittest.TestCase):
"origin_server_ts": 1, "origin_server_ts": 1,
"type": "m.room.message", "type": "m.room.message",
"origin": "test.serv", "origin": "test.serv",
"content": "hewwo?", "content": {"body": "hewwo?"},
"auth_events": [], "auth_events": [],
"prev_events": [("two:test.serv", {}), (most_recent, {})], "prev_events": [("two:test.serv", {}), (most_recent, {})],
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment