expose mfa for inline tests

elastic · Sep 17, 2024 · 6ec81a0 · 6ec81a0
1 parent ca24ebd
commit 6ec81a0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/__tests__/cli.test.ts b/__tests__/cli.test.ts
@@ -92,6 +92,19 @@ describe('CLI', () => {
       expect(await cli.exitCode).toBe(0);
     });
 
+    it('generate mfa totp token', async () => {
+      const cli = new CLIMock(true)
+        .stdin(
+          `step('gen mfa totp', async () => {
+          const token = mfa.totp('JBSWY3DPEHPK3PXP');
+          expect(token.length).toBe(6);
+        })`
+        )
+        .args(['--inline', '--params', JSON.stringify(serverParams)])
+        .run();
+      expect(await cli.exitCode).toBe(0);
+    });
+
     it('exit with 1 on syntax errors', async () => {
       const cli = new CLIMock()
         .stdin(`step('syntax error', async () => {}})`)

diff --git a/src/loader.ts b/src/loader.ts
@@ -29,6 +29,7 @@ import { CliArgs } from './common_types';
 import { step, journey } from './core';
 import { log } from './core/logger';
 import { expect } from './core/expect';
+import * as mfa from "./core/mfa";
 import {
   isDepInstalled,
   isDirectory,
@@ -91,6 +92,7 @@ const loadInlineScript = source => {
     'params',
     'expect',
     'request',
+    'mfa',
     source
   );
   journey('inline', ({ page, context, browser, params, request }) => {
@@ -102,6 +104,7 @@ const loadInlineScript = source => {
       params,
       expect,
       request,
+      mfa,
     ]);
   });
 };