unittest-queue.py

import unittest
from unittest.mock import patch, call


class MyQueue(object):

    def send(self, message):
        pass


class MessageSender(object):

    def __init__(self, queue):
        self.queue = queue

    def send_all(self, messages):
        for message in messages:
            self.queue.send(message)


class MessageSenderTest(unittest.TestCase):

    @patch('MyQueue')
    def test_can_send_all_messages(self, queue):
        messages = ['foo', 'bar', 'baz']
        sender = MessageSender(queue)
        sender.send_all(messages)
        queue.send.assert_has_calls([call.send(message) for message in messages])


if __name__ == '__main__':
    unittest.main()