def test_key_and_connection(self):
from azure.storage.file import FileService
hook = AzureFileShareHook(wasb_conn_id='wasb_test_key')
self.assertEqual(hook.conn_id, 'wasb_test_key')
self.assertIsNone(hook._conn)
self.assertIsInstance(hook.get_conn(), FileService)
def test_key_and_connection(self):
from azure.storage.file import FileService
hook = AzureFileShareHook(wasb_conn_id='wasb_test_key')
assert hook.conn_id == 'wasb_test_key'
assert hook._conn is None
assert isinstance(hook.get_conn(), FileService)
def test_sas_token(self):
from azure.storage.file import FileService
hook = AzureFileShareHook(wasb_conn_id='wasb_test_sas_token')
self.assertEqual(hook.conn_id, 'wasb_test_sas_token')
self.assertIsInstance(hook.get_conn(), FileService)
def test_sas_token(self):
from azure.storage.file import FileService
hook = AzureFileShareHook(wasb_conn_id='wasb_test_sas_token')
assert hook.conn_id == 'wasb_test_sas_token'
assert isinstance(hook.get_conn(), FileService)