Neaten more of the storage layer tests with assertObjectHasAttributes; more standardisation on test layout

1 files changed, 5 insertions, 6 deletions

diff --git a/tests/storage/test_room.py b/tests/storage/test_room.py
index 9979be2f65..369a73d917 100644
--- a/tests/storage/test_room.py
+++ b/tests/storage/test_room.py
@@ -51,13 +51,11 @@ class RoomStoreTestCase(unittest.TestCase):
 
     @defer.inlineCallbacks
     def test_get_room(self):
-        room = yield self.store.get_room(self.room.to_string())
-
         self.assertObjectHasAttributes(
             {"room_id": self.room.to_string(),
              "creator": self.u_creator.to_string(),
              "is_public": True},
-            room
+            (yield self.store.get_room(self.room.to_string()))
         )
 
     @defer.inlineCallbacks
@@ -66,9 +64,10 @@ class RoomStoreTestCase(unittest.TestCase):
             visibility=False
         )
 
-        room = yield self.store.get_room(self.room.to_string())
-
-        self.assertFalse(room.is_public)
+        self.assertObjectHasAttributes(
+            {"is_public": False},
+            (yield self.store.get_room(self.room.to_string()))
+        )
 
     @defer.inlineCallbacks
     def test_get_rooms(self):