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(''), ]