test import updates

Azure · Jul 31, 2018 · 39a9695 · 39a9695
1 parent bca73e4
commit 39a9695
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/azure-keyvault/tests/keyvault_testcase.py b/azure-keyvault/tests/keyvault_testcase.py
@@ -19,3 +19,15 @@ def tearDown(self):
         super(KeyvaultTestCase, self).tearDown()
 
 
+    def _get_keyvault_client(self, api_version):
+        super(KeyvaultTestCase, self).setUp()
+        self.list_test_size = 2
+        if self.is_live:
+            self.client = self.create_basic_client(KeyVaultClient, api_version=api_version)
+        else:
+
+            def _auth_callback(server, resource, scope):
+                return AccessToken('Bearer', 'fake-token')
+            self.client = KeyVaultClient(KeyVaultAuthentication(authorization_callback=_auth_callback), api_version=api_version)
+
+
diff --git a/azure-keyvault/tests/test_internal.py b/azure-keyvault/tests/test_internal.py
@@ -1,13 +1,14 @@
-import unittest
+import json
 import os
 import random
 import string
-import json
-import uuid
 import time
-from azure.keyvault.custom.internal import _bytes_to_int, _int_to_bytes, _int_to_bigendian_8_bytes, \
+import unittest
+import uuid
+
+from azure.keyvault._internal import _bytes_to_int, _int_to_bytes, _int_to_bigendian_8_bytes, \
     _bstr_to_b64url, _b64_to_bstr, _b64_to_str, _str_to_b64url, _a128cbc_hs256_decrypt, _a128cbc_hs256_encrypt, \
-    _RsaKey, _JwsHeader, _JweHeader, _JwsObject, _JweObject
+    _RsaKey, _JwsHeader, _JweHeader
 
 
 class EncodingTests(unittest.TestCase):

diff --git a/azure-keyvault/tests/test_secrets.py b/azure-keyvault/tests/test_secrets.py
@@ -29,6 +29,7 @@ def _validate_secret_list(self, secrets, expected):
                 self.assertEqual(attributes, secret.attributes)
                 del expected[secret.id]
 
+
     @ResourceGroupPreparer()
     @KeyVaultPreparer()
     def test_secret_crud_operations(self, vault, **kwargs):