def test_service_object_to_waiter_model(self): service_object = mock.Mock() session = mock.Mock() service_object.session = session service_object.service_name = 'service' service_object.api_version = '2014-01-01' get_waiter_model_from_service_object(service_object) session.get_waiter_model.assert_called_with('service', '2014-01-01')
def test_can_handle_data_errors(self): service_object = mock.Mock() session = mock.Mock() service_object.session = session service_object.service_name = 'service' service_object.api_version = '2014-01-01' session.get_waiter_model.side_effect = DataNotFoundError( data_path='foo') self.assertIsNone(get_waiter_model_from_service_object(service_object))
def test_can_handle_data_errors(self): service_object = mock.Mock() session = mock.Mock() service_object.session = session service_object.service_name = 'service' service_object.api_version = '2014-01-01' session.get_waiter_model.side_effect = DataNotFoundError( data_path='foo') self.assertIsNone( get_waiter_model_from_service_object(service_object))