def test_wait_for_action_complete_error(self): fake_client = FakeActionClient() fake_client.action['results'][0]['error'] = "id not found" pending_action = fake_client.enqueue_units('git/0', 'list-users', None) with self.assertRaisesRegexp(ValueError, 'id not found'): wait_for_action_complete( fake_client, pending_action['results'][0]['action']['tag'])
def test_wait_for_action_complete_timeout(self): fake_client = FakeActionClient() pending_action = fake_client.enqueue_units('git/0', 'list-users', None) with self.assertRaisesRegexp( Exception, 'Timed out waiting for action to complete.'): wait_for_action_complete( fake_client, pending_action['results'][0]['action']['tag'], 0)
def test_wait_for_action_complete(self): fake_client = FakeActionClient() pending_action = fake_client.enqueue_units('git/0', 'list-users', None) result = wait_for_action_complete( fake_client, pending_action['results'][0]['action']['tag']) expected = {'users': 'user, someuser'} self.assertEqual(result['results'][0]['output'], expected)