packs/GoogleDrive: format file and fix test

Signed-off-by: Gal Nakash <[email protected]>
demisto · Aug 20, 2023 · 3775cfc · 3775cfc
1 parent a0cbb8f
commit 3775cfc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Packs/GoogleDrive/Integrations/GoogleDrive/GoogleDrive.py b/Packs/GoogleDrive/Integrations/GoogleDrive/GoogleDrive.py
@@ -1247,6 +1247,7 @@ def modify_label_command(client: 'GSuiteClient', args: dict[str, str]) -> Comman
 
     return CommandResults(
         outputs=outputs_context,
+        raw_response=response,
         readable_output=table_hr_md,
     )
 

diff --git a/Packs/GoogleDrive/Integrations/GoogleDrive/GoogleDrive_test.py b/Packs/GoogleDrive/Integrations/GoogleDrive/GoogleDrive_test.py
@@ -794,7 +794,6 @@ def test_list_labels(self, mocker_http_request, gsuite_client):
         assert 'GoogleDrive.Drive' in result.outputs
         assert len(result.outputs['GoogleDrive.Drive']['DriveLabels']) == 1
 
-
     @patch(MOCKER_HTTP_METHOD)
     def test_modify_labels_command(self, mocker_http_request, gsuite_client):
         """
@@ -823,8 +822,9 @@ def test_modify_labels_command(self, mocker_http_request, gsuite_client):
         }
         result = modify_label_command(gsuite_client, args)
 
-        assert 'modifiedLabels' in result.outputs
-        assert result.outputs.get('modifiedLabels')[0].get('id') == 'vFmXsMA1fQMz1BdE59YSkisZV4DiKdpxxLQRNNEbbFcb'
+        assert 'modifiedLabels' in result.outputs.get('GoogleDrive.Labels')
+        assert result.outputs.get('GoogleDrive.Labels').get('modifiedLabels')[0].get('id') \
+               == 'vFmXsMA1fQMz1BdE59YSkisZV4DiKdpxxLQRNNEbbFcb'
 
         assert result.raw_response == mock_response