Support for Live storage tests (Azure#6663)

Rakshith Bhyravabhotla · web-flow · commit 17e3eb33168e · 2019-08-05T15:37:38.000-07:00
* play

* playing

* some prints

* some tweaks

* try-345

* fix

* tweak for queue and file

* remove prints
diff --git a/sdk/storage/azure-storage-blob/tests/testcase.py b/sdk/storage/azure-storage-blob/tests/testcase.py
@@ -79,11 +79,11 @@ def setUp(self):
         self.fake_settings = fake_settings
 
         if settings is None:
-            self.test_mode = TestMode.playback
+            self.test_mode = os.getenv('TEST_MODE') or TestMode.playback
         else:
             self.test_mode = self.settings.TEST_MODE.lower() or TestMode.playback
 
-        if self.test_mode == TestMode.playback:
+        if self.test_mode == TestMode.playback or (self.settings is None and self.test_mode.lower() == TestMode.run_live_no_record):
             self.settings = self.fake_settings
 
         # example of qualified test name:
diff --git a/sdk/storage/azure-storage-file/tests/filetestcase.py b/sdk/storage/azure-storage-file/tests/filetestcase.py
@@ -76,11 +76,11 @@ def setUp(self):
         self.fake_settings = fake_settings
 
         if settings is None:
-            self.test_mode = TestMode.playback
+            self.test_mode = os.getenv('TEST_MODE') or TestMode.playback
         else:
             self.test_mode = self.settings.TEST_MODE.lower() or TestMode.playback
 
-        if self.test_mode == TestMode.playback:
+        if self.test_mode == TestMode.playback or (self.settings is None and self.test_mode.lower() == TestMode.run_live_no_record):
             self.settings = self.fake_settings
 
         # example of qualified test name:
diff --git a/sdk/storage/azure-storage-queue/tests/queuetestcase.py b/sdk/storage/azure-storage-queue/tests/queuetestcase.py
@@ -79,11 +79,11 @@ def setUp(self):
         self.fake_settings = fake_settings
 
         if settings is None:
-            self.test_mode = TestMode.playback
+            self.test_mode = os.getenv('TEST_MODE') or TestMode.playback
         else:
             self.test_mode = self.settings.TEST_MODE.lower() or TestMode.playback
 
-        if self.test_mode == TestMode.playback:
+        if self.test_mode == TestMode.playback or (self.settings is None and self.test_mode.lower() == TestMode.run_live_no_record):
             self.settings = self.fake_settings
 
         # example of qualified test name:
diff --git a/sdk/storage/tests.yml b/sdk/storage/tests.yml
@@ -39,4 +39,4 @@ jobs:
         CONNECTION_STRING: $(python-storage-blob-connection-string)
         BLOB_CONNECTION_STRING: $(python-storage-blob-connection-string)
         PREMIUM_CONNECTION_STRING: $(python-storage-premium-connection-string)
-        TEST_MODE: RunLiveNoRecord
+        TEST_MODE: 'RunLiveNoRecord'
