def test_check_volume_state_creating(self):
"""Test the appropriate error for still creating volumes."""
mock_volume = helper.generate_mock_volume(state='creating')
with self.assertRaises(AwsVolumeNotReadyError):
ec2.check_volume_state(mock_volume)
def test_check_volume_state_error(self):
"""Test the appropriate error for other volume states."""
mock_volume = helper.generate_mock_volume(state='error')
with self.assertRaises(AwsVolumeError):
ec2.check_volume_state(mock_volume)
def test_check_volume_state_available(self):
"""Test that a volue is available."""
mock_volume = helper.generate_mock_volume(state='available')
self.assertIsNone(ec2.check_volume_state(mock_volume))