def test_add_waiters_no_service_object(self):
command_table = {}
self.command_object.service_object = None
add_waiters(command_table, self.session, self.command_object)
# Make sure that no wait command was added since no service object
# was passed in.
self.assertEqual(command_table, {})
def test_add_waiters_no_waiter_names(self):
self.session.get_waiter_model.return_value = WaiterModel({
'version': 2,
# No waiters are specified.
'waiters': {}
})
command_table = {}
add_waiters(command_table, self.session, self.command_object)
# Make sure that no wait command was added since the service object
# has no waiters.
self.assertEqual(command_table, {})
def test_add_waiters_no_waiter_names(self):
self.session.get_waiter_model.return_value = WaiterModel(
{
'version': 2,
# No waiters are specified.
'waiters': {}
}
)
command_table = {}
add_waiters(command_table, self.session, self.command_object)
# Make sure that no wait command was added since the service object
# has no waiters.
self.assertEqual(command_table, {})
def test_add_waiter_no_waiter_config(self):
self.session.get_waiter_model.side_effect = DataNotFoundError(
data_path='foo')
command_table = {}
add_waiters(command_table, self.session, self.command_object)
self.assertEqual(command_table, {})
def test_add_waiters(self):
command_table = {}
add_waiters(command_table, self.session, self.command_object)
# Make sure a wait command was added.
self.assertIn('wait', command_table)
self.assertIsInstance(command_table['wait'], WaitCommand)
