1 files changed, 7 insertions, 1 deletions
diff --git a/tests/upload_status_blob.py b/tests/upload_status_blob.py
index af8c580..59cb732 100644
--- a/tests/upload_status_blob.py
+++ b/tests/upload_status_blob.py
@@ -30,8 +30,14 @@ and defined the following 2 variables like:
 """
 from status_blob_url import blockBlobUrl, pageBlobUrl
 
+class MockConfig(object):
+    def get(self, keyName):
+        return None
+
+waagent.Config = MockConfig()
+
 if __name__ == '__main__':
     waagent.LoggerInit('/dev/stdout', '/dev/null', verbose=True)
     status = "a" * 512
     waagent.UploadStatusBlob(blockBlobUrl, status.encode("utf-8"))
-    waagent.UploadStatusBlob(pageBlobUrl, status.encode("utf-8"))
+    #waagent.UploadStatusBlob(pageBlobUrl, status.encode("utf-8"))