def test_load_file(self, mock_service): mock_instance = mock_service.return_value hook = AzureFileShareHook(azure_fileshare_conn_id='azure_fileshare_extras') hook.load_file('path', 'share', 'directory', 'file', max_connections=1) mock_instance.create_file_from_path.assert_called_once_with( 'share', 'directory', 'file', 'path', max_connections=1 )