Improve api

Author: cnasikas

x-pack/plugins/security_solution/public/cases/components/use_push_to_service/index.test.tsx

@@ -10,9 +10,7 @@ import { usePushToService, ReturnUsePushToService, UsePushToService } from '.';
 import { TestProviders } from '../../../common/mock';
 import { usePostPushToService } from '../../containers/use_post_push_to_service';
 import { basicPush, actionLicenses } from '../../containers/mock';
-import * as i18n from './translations';
 import { useGetActionLicense } from '../../containers/use_get_action_license';
-import { getKibanaConfigError, getLicenseError } from './helpers';
 import { connectorsMock } from '../../containers/configure/mock';
 
 jest.mock('react-router-dom', () => {

x-pack/plugins/security_solution/public/common/containers/local_storage/use_messages_storage.test.tsx

@@ -54,4 +54,32 @@ describe('useLocalStorage', () => {
       expect(getMessages('case')).toEqual(['id-1', 'id-2']);
     });
   });
+
+  it('should remove a message', async () => {
+    await act(async () => {
+      const { result, waitForNextUpdate } = renderHook<string, UseMessagesStorage>(() =>
+        useMessagesStorage()
+      );
+      await waitForNextUpdate();
+      const { getMessages, addMessage, removeMessage } = result.current;
+      addMessage('case', 'id-1');
+      addMessage('case', 'id-2');
+      removeMessage('case', 'id-2');
+      expect(getMessages('case')).toEqual(['id-1']);
+    });
+  });
+
+  it('should clear all messages', async () => {
+    await act(async () => {
+      const { result, waitForNextUpdate } = renderHook<string, UseMessagesStorage>(() =>
+        useMessagesStorage()
+      );
+      await waitForNextUpdate();
+      const { getMessages, addMessage, clearAllMessages } = result.current;
+      addMessage('case', 'id-1');
+      addMessage('case', 'id-2');
+      clearAllMessages('case');
+      expect(getMessages('case')).toEqual([]);
+    });
+  });
 });

x-pack/plugins/security_solution/public/common/containers/local_storage/use_messages_storage.tsx

@@ -10,15 +10,15 @@ import { useKibana } from '../../lib/kibana';
 export interface UseMessagesStorage {
   getMessages: (plugin: string) => string[];
   addMessage: (plugin: string, id: string) => void;
+  removeMessage: (plugin: string, id: string) => void;
+  clearAllMessages: (plugin: string) => void;
 }
 
 export const useMessagesStorage = (): UseMessagesStorage => {
   const { storage } = useKibana().services;
 
   const getMessages = useCallback(
-    (plugin: string): string[] => {
-      return storage.get(`${plugin}-messages`) ?? [];
-    },
+    (plugin: string): string[] => storage.get(`${plugin}-messages`) ?? [],
     [storage]
   );
 
@@ -30,8 +30,23 @@ export const useMessagesStorage = (): UseMessagesStorage => {
     [storage]
   );
 
+  const removeMessage = useCallback(
+    (plugin: string, id: string) => {
+      const pluginStorage = storage.get(`${plugin}-messages`) ?? [];
+      storage.set(`${plugin}-messages`, [...pluginStorage.filter((val: string) => val !== id)]);
+    },
+    [storage]
+  );
+
+  const clearAllMessages = useCallback(
+    (plugin: string): string[] => storage.remove(`${plugin}-messages`),
+    [storage]
+  );
+
   return {
     getMessages,
     addMessage,
+    clearAllMessages,
+    removeMessage,
   };
 };