@@ -43,7 +43,7 @@ def test_add_filter(self) -> None:
4343 self .EXAMPLE_FILTER_JSON ,
4444 )
4545
46- self .assertEqual (channel .result [ " code" ], b" 200" )
46+ self .assertEqual (channel .code , 200 )
4747 self .assertEqual (channel .json_body , {"filter_id" : "0" })
4848 filter = self .get_success (
4949 self .store .get_user_filter (user_localpart = "apple" , filter_id = 0 )
@@ -58,7 +58,7 @@ def test_add_filter_for_other_user(self) -> None:
5858 self .EXAMPLE_FILTER_JSON ,
5959 )
6060
61- self .assertEqual (channel .result [ " code" ], b" 403" )
61+ self .assertEqual (channel .code , 403 )
6262 self .assertEqual (channel .json_body ["errcode" ], Codes .FORBIDDEN )
6363
6464 def test_add_filter_non_local_user (self ) -> None :
@@ -71,7 +71,7 @@ def test_add_filter_non_local_user(self) -> None:
7171 )
7272
7373 self .hs .is_mine = _is_mine
74- self .assertEqual (channel .result [ " code" ], b" 403" )
74+ self .assertEqual (channel .code , 403 )
7575 self .assertEqual (channel .json_body ["errcode" ], Codes .FORBIDDEN )
7676
7777 def test_get_filter (self ) -> None :
@@ -85,15 +85,15 @@ def test_get_filter(self) -> None:
8585 "GET" , "/_matrix/client/r0/user/%s/filter/%s" % (self .user_id , filter_id )
8686 )
8787
88- self .assertEqual (channel .result [ " code" ], b" 200" )
88+ self .assertEqual (channel .code , 200 )
8989 self .assertEqual (channel .json_body , self .EXAMPLE_FILTER )
9090
9191 def test_get_filter_non_existant (self ) -> None :
9292 channel = self .make_request (
9393 "GET" , "/_matrix/client/r0/user/%s/filter/12382148321" % (self .user_id )
9494 )
9595
96- self .assertEqual (channel .result [ " code" ], b" 404" )
96+ self .assertEqual (channel .code , 404 )
9797 self .assertEqual (channel .json_body ["errcode" ], Codes .NOT_FOUND )
9898
9999 # Currently invalid params do not have an appropriate errcode
@@ -103,12 +103,12 @@ def test_get_filter_invalid_id(self) -> None:
103103 "GET" , "/_matrix/client/r0/user/%s/filter/foobar" % (self .user_id )
104104 )
105105
106- self .assertEqual (channel .result [ " code" ], b" 400" )
106+ self .assertEqual (channel .code , 400 )
107107
108108 # No ID also returns an invalid_id error
109109 def test_get_filter_no_id (self ) -> None :
110110 channel = self .make_request (
111111 "GET" , "/_matrix/client/r0/user/%s/filter/" % (self .user_id )
112112 )
113113
114- self .assertEqual (channel .result [ " code" ], b" 400" )
114+ self .assertEqual (channel .code , 400 )
0 commit comments