expose mfa for inline tests

Author: vigneshshanmugam

__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('FLIIOLP3IR3W');
+          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 () => {}})`)

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,
     ]);
   });
 };