(pickle_with_and_without_subtree_filesystem)
| 1507 | |
| 1508 | @pytest.mark.azure |
| 1509 | def test_azurefs_options(pickle_with_and_without_subtree_filesystem): |
| 1510 | pickle_module = pickle_with_and_without_subtree_filesystem |
| 1511 | from pyarrow.fs import AzureFileSystem |
| 1512 | |
| 1513 | fs1 = AzureFileSystem(account_name='fake-account-name') |
| 1514 | assert isinstance(fs1, AzureFileSystem) |
| 1515 | assert pickle_module.loads(pickle_module.dumps(fs1)) == fs1 |
| 1516 | |
| 1517 | fs2 = AzureFileSystem(account_name='fake-account-name', |
| 1518 | account_key='fakeaccountkey') |
| 1519 | assert isinstance(fs2, AzureFileSystem) |
| 1520 | assert pickle_module.loads(pickle_module.dumps(fs2)) == fs2 |
| 1521 | assert fs2 != fs1 |
| 1522 | |
| 1523 | fs3 = AzureFileSystem(account_name='fake-account', account_key='fakeaccount', |
| 1524 | blob_storage_authority='fake-blob-authority', |
| 1525 | dfs_storage_authority='fake-dfs-authority', |
| 1526 | blob_storage_scheme='https', |
| 1527 | dfs_storage_scheme='https') |
| 1528 | assert isinstance(fs3, AzureFileSystem) |
| 1529 | assert pickle_module.loads(pickle_module.dumps(fs3)) == fs3 |
| 1530 | assert fs3 != fs2 |
| 1531 | |
| 1532 | fs4 = AzureFileSystem(account_name='fake-account-name', |
| 1533 | sas_token='fakesastoken') |
| 1534 | assert isinstance(fs4, AzureFileSystem) |
| 1535 | assert pickle_module.loads(pickle_module.dumps(fs4)) == fs4 |
| 1536 | assert fs4 != fs3 |
| 1537 | |
| 1538 | fs5 = AzureFileSystem( |
| 1539 | account_name='fake-account-name', |
| 1540 | tenant_id='fake-tenant-id', |
| 1541 | client_id='fake-client-id', |
| 1542 | client_secret='fake-client-secret' |
| 1543 | ) |
| 1544 | assert isinstance(fs5, AzureFileSystem) |
| 1545 | assert pickle_module.loads(pickle_module.dumps(fs5)) == fs5 |
| 1546 | assert fs5 != fs4 |
| 1547 | |
| 1548 | fs6 = AzureFileSystem( |
| 1549 | account_name='fake-account-name', |
| 1550 | client_id='fake-client-id' |
| 1551 | ) |
| 1552 | assert isinstance(fs6, AzureFileSystem) |
| 1553 | assert pickle_module.loads(pickle_module.dumps(fs6)) == fs6 |
| 1554 | assert fs6 != fs5 |
| 1555 | |
| 1556 | with pytest.raises(ValueError, match="client_id must be specified"): |
| 1557 | AzureFileSystem( |
| 1558 | account_name='fake-account-name', |
| 1559 | tenant_id='fake-tenant-id' |
| 1560 | ) |
| 1561 | |
| 1562 | with pytest.raises(ValueError, match="client_id must be specified"): |
| 1563 | AzureFileSystem( |
| 1564 | account_name='fake-account-name', |
| 1565 | client_secret='fake-client-secret' |
| 1566 | ) |
nothing calls this directly
no test coverage detected