fix(classification): Code review changes

box · Oct 30, 2024 · 0ea6b82 · 0ea6b82
1 parent fdce2c6
commit 0ea6b82
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/features/classification/security-controls/__tests__/SecurityControls.test.js b/src/features/classification/security-controls/__tests__/SecurityControls.test.js
@@ -47,8 +47,8 @@ describe('features/classification/security-controls/SecurityControls', () => {
 
     test('should render SecurityControls with Integration label when using SHORT controlsFormat and shouldDisplayAppsAsIntegrations is true', () => {
         wrapper.setProps({ controlsFormat: SHORT, shouldDisplayAppsAsIntegrations: true });
-        expect(wrapper.find('SecurityControlsItem').prop('message').id).toBe(
-            messages.shortSharingDownloadIntegration.id,
+        expect(wrapper.find('SecurityControlsItem').prop('message').defaultMessage).toBe(
+            'Sharing, download and integration restrictions apply',
         );
     });
 
@@ -74,7 +74,7 @@ describe('features/classification/security-controls/SecurityControls', () => {
             wrapper
                 .find('SecurityControlsModal')
                 .prop('modalItems')
-                .find(item => item.message.id === messages.integrationDownloadAllowlist.id),
+                .find(item => item.message.defaultMessage === 'Only select integrations are allowed: {appNames}'),
         ).toBeDefined();
     });
 
@@ -89,7 +89,7 @@ describe('features/classification/security-controls/SecurityControls', () => {
             wrapper.findWhere(
                 node =>
                     node.type() === 'SecurityControlsItem' &&
-                    node.prop('message').id === messages.integrationDownloadAllowlist.id,
+                    node.prop('message').defaultMessage === 'Only select integrations are allowed: {appNames}',
             ),
         ).toBeDefined();
     });