Adding test for managed disk config

qubole · Oct 1, 2020 · d758a49 · d758a49
1 parent b05d9be
commit d758a49
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/test_clusterv2.py b/tests/test_clusterv2.py
@@ -219,6 +219,26 @@ def test_azure_storage_config(self):
                                                       }
                                                  })
 
+    def test_azure_managed_disk_config(self):
+        sys.argv = ['qds.py', '--version', 'v2', '--cloud', 'AZURE', 'cluster', 'create', '--label', 'test_label',
+                    '--storage-access-key', 'testkey', '--storage-account-name', 'test_account_name',
+                    '--managed-disk-account-type' 'test_managed_disk']
+        Qubole.cloud = None
+        print_command()
+        Connection._api_call = Mock(return_value={})
+        qds.main()
+        Connection._api_call.assert_called_with('POST', 'clusters',
+                                                {'cluster_info':
+                                                     {'label': ['test_label']},
+                                                 'cloud_config':
+                                                     {'storage_config':
+                                                          {'storage_access_key': 'testkey',
+                                                           'storage_account_name': 'test_account_name',
+                                                           'managed_disk_account_type': 'test_managed_disk'
+                                                           }
+                                                      }
+                                                 })
+
     def test_azure_network_config(self):
         sys.argv = ['qds.py', '--version', 'v2', '--cloud', 'AZURE', 'cluster', 'create', '--label', 'test_label',
                     '--vnet-name', 'testvnet', '--subnet-name', 'testsubnet',