class TestWaiterStateCommandBuilder(unittest.TestCase):
def setUp(self):
self.session = mock.Mock()
self.service_model = mock.Mock()
# Create some waiters.
self.model = WaiterModel(
{
"version": 2,
"waiters": {
"InstanceRunning": {
"description": "My waiter description.",
"delay": 1,
"maxAttempts": 10,
"operation": "MyOperation",
},
"BucketExists": {
"description": "My waiter description.",
"operation": "MyOperation",
"delay": 1,
"maxAttempts": 10,
},
},
}
)
self.waiter_builder = WaiterStateCommandBuilder(self.session, self.model, self.service_model)
def test_build_waiter_state_cmds(self):
subcommand_table = {}
self.waiter_builder.build_all_waiter_state_cmds(subcommand_table)
# Check the commands are in the command table
self.assertEqual(len(subcommand_table), 2)
self.assertIn("instance-running", subcommand_table)
self.assertIn("bucket-exists", subcommand_table)
# Make sure that the correct operation object was used.
self.service_model.operation_model.assert_called_with("MyOperation")
# Introspect the commands in the command table
instance_running_cmd = subcommand_table["instance-running"]
bucket_exists_cmd = subcommand_table["bucket-exists"]
# Check that the instance type is correct.
self.assertIsInstance(instance_running_cmd, WaiterStateCommand)
self.assertIsInstance(bucket_exists_cmd, WaiterStateCommand)
# Check the descriptions are set correctly.
self.assertEqual(
instance_running_cmd.DESCRIPTION,
"My waiter description. It will poll every 1 seconds until "
"a successful state has been reached. This will exit with a "
"return code of 255 after 10 failed checks.",
)
self.assertEqual(
bucket_exists_cmd.DESCRIPTION,
"My waiter description. It will poll every 1 seconds until "
"a successful state has been reached. This will exit with a "
"return code of 255 after 10 failed checks.",
)
class TestWaiterStateCommandBuilder(unittest.TestCase):
def setUp(self):
self.session = mock.Mock()
self.service_model = mock.Mock()
# Create some waiters.
self.model = WaiterModel({
'version': 2,
'waiters': {
'InstanceRunning': {
'description': 'my waiter description',
'delay': 1,
'maxAttempts': 10,
'operation': 'MyOperation',
},
'BucketExists': {
'description': 'my waiter description',
'operation': 'MyOperation',
'delay': 1,
'maxAttempts': 10,
}
}
})
self.waiter_builder = WaiterStateCommandBuilder(
self.session,
self.model,
self.service_model
)
def test_build_waiter_state_cmds(self):
subcommand_table = {}
self.waiter_builder.build_all_waiter_state_cmds(subcommand_table)
# Check the commands are in the command table
self.assertEqual(len(subcommand_table), 2)
self.assertIn('instance-running', subcommand_table)
self.assertIn('bucket-exists', subcommand_table)
# Make sure that the correct operation object was used.
self.service_model.operation_model.assert_called_with('MyOperation')
# Introspect the commands in the command table
instance_running_cmd = subcommand_table['instance-running']
bucket_exists_cmd = subcommand_table['bucket-exists']
# Check that the instance type is correct.
self.assertIsInstance(instance_running_cmd, WaiterStateCommand)
self.assertIsInstance(bucket_exists_cmd, WaiterStateCommand)
# Check the descriptions are set correctly.
self.assertEqual(
instance_running_cmd.DESCRIPTION,
'my waiter description',
)
self.assertEqual(
bucket_exists_cmd.DESCRIPTION,
'my waiter description',
)
class TestWaiterStateCommandBuilder(unittest.TestCase):
def setUp(self):
self.session = mock.Mock()
self.service_model = mock.Mock()
# Create some waiters.
self.model = WaiterModel({
'version': 2,
'waiters': {
'InstanceRunning': {
'description': 'My waiter description.',
'delay': 1,
'maxAttempts': 10,
'operation': 'MyOperation',
},
'BucketExists': {
'description': 'My waiter description.',
'operation': 'MyOperation',
'delay': 1,
'maxAttempts': 10,
}
}
})
self.waiter_builder = WaiterStateCommandBuilder(
self.session, self.model, self.service_model)
def test_build_waiter_state_cmds(self):
subcommand_table = {}
self.waiter_builder.build_all_waiter_state_cmds(subcommand_table)
# Check the commands are in the command table
self.assertEqual(len(subcommand_table), 2)
self.assertIn('instance-running', subcommand_table)
self.assertIn('bucket-exists', subcommand_table)
# Make sure that the correct operation object was used.
self.service_model.operation_model.assert_called_with('MyOperation')
# Introspect the commands in the command table
instance_running_cmd = subcommand_table['instance-running']
bucket_exists_cmd = subcommand_table['bucket-exists']
# Check that the instance type is correct.
self.assertIsInstance(instance_running_cmd, WaiterStateCommand)
self.assertIsInstance(bucket_exists_cmd, WaiterStateCommand)
# Check the descriptions are set correctly.
self.assertEqual(
instance_running_cmd.DESCRIPTION,
'My waiter description. It will poll every 1 seconds until '
'a successful state has been reached. This will exit with a '
'return code of 255 after 10 failed checks.')
self.assertEqual(
bucket_exists_cmd.DESCRIPTION,
'My waiter description. It will poll every 1 seconds until '
'a successful state has been reached. This will exit with a '
'return code of 255 after 10 failed checks.')
class TestWaiterStateCommandBuilder(unittest.TestCase):
def setUp(self):
self.service_object = mock.Mock()
# Create some waiters.
self.model = WaiterModel({
'version': 2,
'waiters': {
'InstanceRunning': {
'description': 'my waiter description',
'delay': 1,
'maxAttempts': 10,
'operation': 'MyOperation',
},
'BucketExists': {
'description': 'my waiter description',
'operation': 'MyOperation',
'delay': 1,
'maxAttempts': 10,
}
}
})
self.waiter_builder = WaiterStateCommandBuilder(
self.model,
self.service_object
)
def test_build_waiter_state_cmds(self):
subcommand_table = {}
self.waiter_builder.build_all_waiter_state_cmds(subcommand_table)
# Check the commands are in the command table
self.assertEqual(len(subcommand_table), 2)
self.assertIn('instance-running', subcommand_table)
self.assertIn('bucket-exists', subcommand_table)
# Make sure that the correct operation object was used.
self.service_object.get_operation.assert_called_with('MyOperation')
# Introspect the commands in the command table
instance_running_cmd = subcommand_table['instance-running']
bucket_exists_cmd = subcommand_table['bucket-exists']
# Check that the instance type is correct.
self.assertIsInstance(instance_running_cmd, WaiterStateCommand)
self.assertIsInstance(bucket_exists_cmd, WaiterStateCommand)
# Check the descriptions are set correctly.
self.assertEqual(
instance_running_cmd.DESCRIPTION,
'my waiter description',
)
self.assertEqual(
bucket_exists_cmd.DESCRIPTION,
'my waiter description',
)