def _get_table_client_from_storage_account(storage_account, session):
primary_key = StorageUtilities.get_storage_primary_key(storage_account['resourceGroup'],
storage_account['name'],
session)
return TableService(
account_name=storage_account['name'],
account_key=primary_key
)
def test_get_storage_primary_key(self):
key1 = StorageAccountKey()
key1.key_name = "key1"
key1.value = "mock_storage_key"
data = StorageAccountListKeysResult()
data.keys = [key1]
with patch(self._get_storage_client_string() + '.list_keys', return_value=data) \
as list_keys_mock:
primary_key = StorageUtilities.get_storage_primary_key(
'mock_rg_group', 'mock_account', self.session)
list_keys_mock.assert_called_with('mock_rg_group', 'mock_account')
self.assertEqual(primary_key, data.keys[0].value)
