def test_add_raises_ValueError(self): hub = Hub() hub._add = Mock() hub._add.side_effect = ValueError() hub._discard = Mock() hub.add([2], Mock(), READ) hub._discard.assert_called_with(2)
def test_add_raises_ValueError(self): hub = Hub() hub.poller = Mock(name='hub.poller') hub.poller.register.side_effect = ValueError() hub._discard = Mock(name='hub.discard') hub.add(2, Mock(), READ) hub._discard.assert_called_with(2)