""" test_wuvt.py - tests for the wuvt module author: mutantmonkey """ import re import unittest from mock import MagicMock from modules.wuvt import wuvt class TestWuvt(unittest.TestCase): def setUp(self): self.phenny = MagicMock() def test_wuvt(self): wuvt(self.phenny, None) out = self.phenny.reply.call_args[0][0] m = re.match('^DJ .* is currently playing: .* by .*$', out, flags=re.UNICODE) self.assertTrue(m)