From 54b3dfd290367aa082fe30b20d90bf4b43ed1838 Mon Sep 17 00:00:00 2001 From: Mark Hoekveen Date: Fri, 2 Dec 2022 21:07:55 +0100 Subject: [PATCH] tests voor aww --- General.py | 13 ++++++------- tests.py | 6 ++++++ 2 files changed, 12 insertions(+), 7 deletions(-) diff --git a/General.py b/General.py index ad4296e..c86b601 100644 --- a/General.py +++ b/General.py @@ -61,17 +61,16 @@ class General(commands.Cog): print("No permission to delete message") yield from ctx.channel.send(ctx.author.mention, file=discord.File('test.png', 'tex.png')) + def get_aww_post(self): + raww = self.r.subreddit('aww+catloaf+tuckedinkitties+babyelephantgifs+babybigcatgifs+blep+holdmynip+gingerkitty+cathighfive+catsstandingup+catsareassholes+eyebleach') + listing = list(raww.top(time_filter='month')) + return random.choice(listing) + @commands.command(pass_context=True) @asyncio.coroutine def aww(self, ctx): """ iets schattigs... """ - print(type(ctx.message.channel)) - raww = self.r.subreddit('aww+catloaf+tuckedinkitties+babyelephantgifs+babybigcatgifs+blep+holdmynip+gingerkitty+cathighfive+catsstandingup+catsareassholes+eyebleach') - post = raww.top('month', limit=250) - listing = list(post) - random.shuffle(listing) - post = listing[0] - + post = self.get_aww_post() try: yield from ctx.message.delete() except: diff --git a/tests.py b/tests.py index cf7dd14..22ebc82 100644 --- a/tests.py +++ b/tests.py @@ -29,3 +29,9 @@ class TestGeneral(unittest.TestCase): self.assertIsInstance(data, requests.Response) with self.subTest(msg="Does not convert to json"): self.assertIsInstance(json.loads(data.content), dict) + + def test_aww(self): + if not isinstance(self.cog.r, praw.Reddit): + self.skipTest("PRAW Reddit instance failed to create") + post = self.cog.get_aww_post() + self.assertIsInstance(post.url, str)