modules/fm-orchestrator
1
0
Fork 0
mirror of https://pagure.io/fm-orchestrator.git synced 2026-05-16 13:56:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Change the format of the unit tests to pytest

Browse Source
This commit is contained in:
mprahl
2018-01-30 19:03:28 -05:00
parent fb8e9cf02c
commit e91d09f7ca
23 changed files with 719 additions and 796 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/test_scheduler/test_consumer.py
View File

@@ -18,13 +18,12 @@
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
import unittest
from mock import patch, MagicMock
from module_build_service.scheduler.consumer import MBSConsumer
from module_build_service.messaging import KojiTagChange, KojiRepoChange
class TestConsumer(unittest.TestCase):
class TestConsumer:
@patch('module_build_service.messaging.conf.messaging', new='fedmsg')
def test_get_abstracted_msg_fedmsg(self):
@@ -57,10 +56,10 @@ class TestConsumer(unittest.TestCase):
}
}
msg_obj = consumer.get_abstracted_msg(msg)
self.assertIsInstance(msg_obj, KojiTagChange)
self.assertEqual(msg_obj.msg_id, msg['msg_id'])
self.assertEqual(msg_obj.tag, msg['msg']['tag'])
self.assertEqual(msg_obj.artifact, msg['msg']['name'])
assert isinstance(msg_obj, KojiTagChange)
assert msg_obj.msg_id == msg['msg_id']
assert msg_obj.tag == msg['msg']['tag']
assert msg_obj.artifact == msg['msg']['name']
@patch('module_build_service.scheduler.consumer.models')
@patch.object(MBSConsumer, 'process_message')
@@ -94,8 +93,8 @@ class TestConsumer(unittest.TestCase):
}
}
consumer.consume(msg)
self.assertEqual(process_message.call_count, 1)
assert process_message.call_count == 1
msg_obj = process_message.call_args[0][1]
self.assertIsInstance(msg_obj, KojiRepoChange)
self.assertEqual(msg_obj.msg_id, msg['body']['msg_id'])
self.assertEqual(msg_obj.repo_tag, msg['body']['msg']['tag'])
assert isinstance(msg_obj, KojiRepoChange)
assert msg_obj.msg_id == msg['body']['msg_id']
assert msg_obj.repo_tag == msg['body']['msg']['tag']