From 512eb552e0ca740e1d285dc1b66a56579bcf68ec Mon Sep 17 00:00:00 2001
From: Daniel Watkins <daniel.watkins@canonical.com>
Date: Fri, 8 May 2015 16:52:49 +0100
Subject: Fix retrying.

---
 cloudinit/sources/helpers/azure.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'cloudinit')

diff --git a/cloudinit/sources/helpers/azure.py b/cloudinit/sources/helpers/azure.py
index dfdfa7c2..2ce728f5 100644
--- a/cloudinit/sources/helpers/azure.py
+++ b/cloudinit/sources/helpers/azure.py
@@ -242,14 +242,19 @@ class WALinuxAgentShim(object):
         self.openssl_manager = OpenSSLManager()
         http_client = AzureEndpointHttpClient(self.openssl_manager.certificate)
         LOG.info('Registering with Azure...')
-        for i in range(10):
+        attempts = 0
+        while True:
             try:
                 response = http_client.get(
                     'http://{}/machine/?comp=goalstate'.format(self.endpoint))
             except Exception:
-                time.sleep(i + 1)
+                if attempts < 10:
+                    time.sleep(attempts + 1)
+                else:
+                    raise
             else:
                 break
+            attempts += 1
         LOG.debug('Successfully fetched GoalState XML.')
         goal_state = GoalState(response.contents, http_client)
         public_keys = []
-- 
cgit v1.2.3