Cleanup some tests

praw-dev · Jun 15, 2021 · f4cec73 · f4cec73
1 parent 9aa6651
commit f4cec73
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/tests/unit/models/reddit/test_collections.py b/tests/unit/models/reddit/test_collections.py
@@ -63,7 +63,7 @@ def test_neq(self):
 class TestSubredditCollections(UnitTest):
     async def test_call(self):
         collections = SubredditCollections(
-            self.reddit, self.reddit.subreddit("placeholder")
+            self.reddit, await self.reddit.subreddit("placeholder")
         )
         with pytest.raises(TypeError):
             await collections()

diff --git a/tests/unit/models/reddit/test_removal_reasons.py b/tests/unit/models/reddit/test_removal_reasons.py
@@ -47,10 +47,14 @@ def test_exception(self):
                 _data={},
             )
         with pytest.raises(ValueError):
-            RemovalReason(self.reddit, subreddit=self.reddit.subreddit("a"), id="")
+            RemovalReason(
+                self.reddit, subreddit=Subreddit(self.reddit, display_name="a"), id=""
+            )
         with pytest.raises(ValueError):
             RemovalReason(
-                self.reddit, subreddit=self.reddit.subreddit("a"), reason_id=""
+                self.reddit,
+                subreddit=Subreddit(self.reddit, display_name="a"),
+                reason_id="",
             )
 
     async def test__get(self):

diff --git a/tests/unit/util/test_token_manager.py b/tests/unit/util/test_token_manager.py
@@ -36,13 +36,13 @@ def test_reddit__must_only_be_set_once(self):
 
 
 class TestFileTokenManager(UnitTest):
-    def setUp(self):
+    async def setUp(self):
         aiofiles.threadpool.wrap.register(mock.MagicMock)(
             lambda *args, **kwargs: aiofiles.threadpool.AsyncBufferedIOBase(
                 *args, **kwargs
             )
         )
-        super(TestFileTokenManager, self).setUp()
+        await super(TestFileTokenManager, self).setUp()
 
     async def test_post_refresh_token_callback__writes_to_file(self):
         authorizer = DummyAuthorizer("token_value")