def test_verify_no_traffic(self, mocks, mocker):
subject = TrafficHandler('snow', self.mock_traffic_commands)
subject.verify()
assert self.mock_print.mock_calls == [
mocker.call("\tyou didn't request traffic on snow")
]
def test_verify(self, mocks, mocker):
subject = TrafficHandler('snow', self.mock_traffic_commands)
mock_first_traffic = mocker.Mock()
mock_first_traffic.status.return_value = 'first: SUCCESS'
mock_second_traffic = mocker.Mock()
mock_second_traffic.status.return_value = 'second: FAILURE'
subject.traffic_handles = [mock_first_traffic, mock_second_traffic]
subject.verify()
assert self.mock_print.mock_calls == [
mocker.call('snow - first: SUCCESS'),
mocker.call('snow - second: FAILURE'),
mocker.call(''),
]