fix: Adjust log level for BiDi events (#2505)

appium · Dec 19, 2024 · 1a73fcd · 1a73fcd
1 parent aa4ad6d
commit 1a73fcd
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/lib/commands/bidi/models.ts b/lib/commands/bidi/models.ts
@@ -1,7 +1,8 @@
-import type { LogEntryAddedEvent, ContextUpdatedEvent } from './types';
+import type { LogEntryAddedEvent, ContextUpdatedEvent, BiDiLogLevel } from './types';
 import { NATIVE_WIN } from '../../utils';
 import { CONTEXT_UPDATED_EVENT, CONTEXT_UPDATED_EVENT_OBSOLETE, LOG_ENTRY_ADDED_EVENT } from './constants';
 import type { LogEntry } from '../types';
+import _ from 'lodash';
 
 function toContextUpdatedEvent(method: string, contextName: string): ContextUpdatedEvent {
   return {
@@ -30,7 +31,7 @@ export function makeLogEntryAddedEvent(entry: LogEntry, context: string, type: s
     method: LOG_ENTRY_ADDED_EVENT,
     params: {
       type,
-      level: entry.level,
+      level: adjustLogLevel(entry.level),
       source: {
         realm: '',
       },
@@ -39,3 +40,16 @@ export function makeLogEntryAddedEvent(entry: LogEntry, context: string, type: s
     },
   };
 }
+
+function adjustLogLevel(originalLevel: string): BiDiLogLevel {
+  const originalLevelLc = _.toLower(originalLevel);
+  switch (originalLevelLc) {
+    case 'debug':
+    case 'info':
+    case 'warn':
+    case 'error':
+      return originalLevelLc as BiDiLogLevel;
+    default:
+      return 'info';
+  }
+}
diff --git a/lib/commands/bidi/types.ts b/lib/commands/bidi/types.ts
@@ -7,9 +7,11 @@ interface LogEntrySource {
   realm: string;
 }
 
+export type BiDiLogLevel = 'debug' | 'info' | 'warn' | 'error';
+
 interface LogEntryAddedEventParams {
   type: string;
-  level: string;
+  level: BiDiLogLevel;
   source: LogEntrySource;
   text: string;
   timestamp: number;