test: Improve integration test commons (box/box-codegen#686) (#541)

Author: box-sdk-build

.codegen.json

@@ -1 +1 @@
-{ "engineHash": "3955ab2", "specHash": "c303afc", "version": "1.13.0" }
+{ "engineHash": "4148197", "specHash": "c303afc", "version": "1.13.0" }

test/commons.py

@@ -60,6 +60,25 @@
 
 from box_sdk_gen.schemas.terms_of_service import TermsOfService
 
+from box_sdk_gen.networking.auth import Authentication
+
+from box_sdk_gen.box.ccg_auth import BoxCCGAuth
+
+from box_sdk_gen.box.ccg_auth import CCGConfig
+
+from box_sdk_gen.internal.utils import is_browser
+
+
+def get_ccg_auth() -> BoxCCGAuth:
+    ccg_config: CCGConfig = CCGConfig(
+        client_id=get_env_var('CLIENT_ID'),
+        client_secret=get_env_var('CLIENT_SECRET'),
+        enterprise_id=get_env_var('ENTERPRISE_ID'),
+    )
+    auth: BoxCCGAuth = BoxCCGAuth(config=ccg_config)
+    return auth
+
+
 from box_sdk_gen.box.jwt_auth import BoxJWTAuth
 
 from box_sdk_gen.box.jwt_auth import JWTConfig
@@ -74,13 +93,19 @@ def get_jwt_auth() -> BoxJWTAuth:
 
 
 def get_default_client_with_user_subject(user_id: str) -> BoxClient:
+    if is_browser():
+        ccg_auth: BoxCCGAuth = get_ccg_auth()
+        ccg_auth_user: BoxCCGAuth = ccg_auth.with_user_subject(user_id)
+        return BoxClient(auth=ccg_auth_user)
     auth: BoxJWTAuth = get_jwt_auth()
     auth_user: BoxJWTAuth = auth.with_user_subject(user_id)
     return BoxClient(auth=auth_user)
 
 
 def get_default_client() -> BoxClient:
-    client: BoxClient = BoxClient(auth=get_jwt_auth())
+    client: BoxClient = BoxClient(
+        auth=get_ccg_auth() if is_browser() else get_jwt_auth()
+    )
     return client
 
 

test/files.py

@@ -14,6 +14,8 @@
 
 from box_sdk_gen.managers.files import GetFileThumbnailUrlExtension
 
+from box_sdk_gen.internal.utils import Buffer
+
 from box_sdk_gen.managers.files import GetFileThumbnailByIdExtension
 
 from box_sdk_gen.schemas.trash_file import TrashFile
@@ -30,6 +32,10 @@
 
 from box_sdk_gen.internal.utils import read_byte_stream
 
+from box_sdk_gen.internal.utils import generate_byte_stream_from_buffer
+
+from box_sdk_gen.internal.utils import generate_byte_buffer
+
 from box_sdk_gen.internal.utils import buffer_equals
 
 from box_sdk_gen.internal.utils import ByteStream
@@ -69,19 +75,17 @@ def testGetFileThumbnailUrl():
 
 def testGetFileThumbnail():
     thumbnail_file_name: str = get_uuid()
-    thumbnail_content_stream: ByteStream = generate_byte_stream(1024 * 1024)
+    thumbnail_buffer: Buffer = generate_byte_buffer(1024 * 1024)
+    thumbnail_content_stream: ByteStream = generate_byte_stream_from_buffer(
+        thumbnail_buffer
+    )
     thumbnail_file: FileFull = upload_file(
         thumbnail_file_name, thumbnail_content_stream
     )
     thumbnail: Optional[ByteStream] = client.files.get_file_thumbnail_by_id(
         thumbnail_file.id, GetFileThumbnailByIdExtension.PNG
     )
-    assert (
-        not buffer_equals(
-            read_byte_stream(thumbnail), read_byte_stream(thumbnail_content_stream)
-        )
-        == True
-    )
+    assert not buffer_equals(read_byte_stream(thumbnail), thumbnail_buffer) == True
     client.files.delete_file_by_id(thumbnail_file.id)