Fix messageKeys import

cryostatio · Apr 21, 2022 · 3299420 · 3299420
1 parent f54d0b7
commit 3299420
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/app/SecurityPanel/StoreJmxCredentials.tsx b/src/app/SecurityPanel/StoreJmxCredentials.tsx
@@ -83,7 +83,7 @@ export const StoreJmxCredentials = () => {
   }, [context, context.targets, setTargets, refreshStoredTargetsList]);
 
   React.useEffect(() => {
-    const sub = context.notificationChannel.messages('TargetCredentialsStored' as NotificationCategory.TargetCredentialsStored).subscribe((v) => {
+    const sub = context.notificationChannel.messages(NotificationCategory.TargetCredentialsStored).subscribe((v) => {
       const updatedTarget = targets.filter((t) => t.connectUrl === v.message.target).pop();
       if(!updatedTarget) {
         return;
@@ -94,7 +94,7 @@ export const StoreJmxCredentials = () => {
   }, [context, context.notificationChannel, targets, setStoredTargets]);
 
   React.useEffect(() => {
-    const sub = context.notificationChannel.messages('TargetCredentialsDeleted' as NotificationCategory.TargetCredentialsDeleted).subscribe((v) => {
+    const sub = context.notificationChannel.messages(NotificationCategory.TargetCredentialsDeleted).subscribe((v) => {
       setStoredTargets(old => old.filter(t => t.connectUrl !== v.message.target));
     });
     return () => sub.unsubscribe();

diff --git a/src/test/SecurityPanel/StoreJmxCredentials.test.tsx b/src/test/SecurityPanel/StoreJmxCredentials.test.tsx
@@ -95,6 +95,7 @@ jest.mock('@app/Shared/Services/NotificationChannel.service', () => {
   const mockNotification = { message: { target: 'service:jmx:rmi://someUrl' } } as NotificationMessage;
 
   return {
+    ...jest.requireActual('@app/Shared/Services/NotificationChannel.service'),
     NotificationChannel: jest.fn(() => {
       return {
         messages: jest