Python AzureCosmosDBHook.delete_database Examples

Example #1

File: test_azure_cosmos.py Project: yqian1991/airflow

 def test_delete_database(self, mock_cosmos):
     hook = AzureCosmosDBHook(
         azure_cosmos_conn_id='azure_cosmos_test_key_id')
     hook.delete_database(self.test_database_name)
     expected_calls = [mock.call().DeleteDatabase('dbs/test_database_name')]
     mock_cosmos.assert_any_call(self.test_end_point,
                                 {'masterKey': self.test_master_key})
     mock_cosmos.assert_has_calls(expected_calls)

Example #2

File: test_azure_cosmos.py Project: ysktir/airflow-1

 def test_delete_database_exception(self, mock_cosmos):
     hook = AzureCosmosDBHook(
         azure_cosmos_conn_id='azure_cosmos_test_key_id')
     with pytest.raises(AirflowException):
         hook.delete_database(None)