def test_open(self): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() # test consumer._open() # validation consumer._reader.open.assert_called_once_with()
def test_open(self): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() # test consumer._open() # validation consumer._reader.open.assert_called_once_with()
def test_open_exception(self, sleep): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.open.side_effect = [ValueError, None] # test consumer._open() # validation sleep.assert_called_once_with(60) self.assertEqual(consumer._reader.open.call_count, 2)
def test_open_exception(self, sleep): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.open.side_effect = [ValueError, None] # test consumer._open() # validation sleep.assert_called_once_with(60) self.assertEqual(consumer._reader.open.call_count, 2)
def test_read_exception(self, sleep): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.side_effect = IndexError consumer._open = Mock() consumer._close = Mock() # test consumer._read() # validation consumer._close.assert_called_once_with() consumer._open.assert_called_once_with() sleep.assert_called_once_with(60)
def test_read_exception(self, sleep): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.side_effect = IndexError consumer._open = Mock() consumer._close = Mock() # test consumer._read() # validation consumer._close.assert_called_once_with() consumer._open.assert_called_once_with() sleep.assert_called_once_with(60)
def test_run(self, reader): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._open = Mock() consumer._close = Mock() consumer._read = Mock(side_effect=StopIteration) # test try: consumer.run() except StopIteration: pass # validation reader.assert_called_once_with(node, url) consumer._open.assert_called_once_with() consumer._read.assert_called_once_with() consumer._close.assert_called_once_with()
def test_run(self, reader): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._open = Mock() consumer._close = Mock() consumer._read = Mock(side_effect=StopIteration) # test try: consumer.run() except StopIteration: pass # validation reader.assert_called_once_with(node, url) consumer._open.assert_called_once_with() consumer._read.assert_called_once_with() consumer._close.assert_called_once_with()