MFA (TOTP) 認証が必要なサイトへのSyntheticテスト（Scripted Browser）のスクリプト

var assert = require('assert');
var DefaultTimeout = 30000;
var otpauth = require('otpauth');


// TOTPの設定
let totp = new otpauth.TOTP({
  algorithm: 'SHA1',
  digits: 6,
  period: 30,
  secret: $secure.MFA_LOGIN_PRIVATE_KEY
});

/**
 * 有効期限に余裕のあるTOTP Tokenを取得する関数
 * @param {number} minRemainingSeconds - 最低限必要な残り時間（秒）
 * @returns {Promise<string>} Token
 */
async function getSafeToken(minRemainingSeconds = 5) {
  // 現在のタイムスタンプ（ミリ秒）から、現在の30秒周期の残り時間を計算
  const periodMs = totp.period * 1000;
  let remainingMs = periodMs - (Date.now() % periodMs);
  let remainingSeconds = remainingMs / 1000;

  // 残り時間が指定秒数未満（例: 5秒未満）の場合は、次の周期まで待機
  if (remainingSeconds < minRemainingSeconds) {
    // 次のトークンに切り替わるまでの時間（ミリ秒）待機する
    // 完全に切り替わるよう、少しだけバッファ（+100ms）を足しています
    const waitTime = remainingMs + 100; 
    console.log(`Tokenの残り時間が ${remainingSeconds.toFixed(1)}秒 のため、次のTokenまで ${waitTime / 1000}秒 待機します...`);
    
    await new Promise(resolve => setTimeout(resolve, waitTime));
  }

  // 余裕がある状態（または切り替わった直後）でTokenを生成して返す
  return totp.generate();
}


console.log('Step 1');
console.log('Navigating to "https://authenticationtest.com/totpChallenge/"');

$webDriver.get("https://authenticationtest.com/totpChallenge/")
.then(function() {
    console.log('Step 2');
    console.log('ユーザ名入力');
    $webDriver.findElement($selenium.By.id("email")).sendKeys($secure.MFA_LOGIN_USER_NAME);
})
.then(function() {
    console.log('Step 3');
    console.log('パスワード入力');
    $webDriver.findElement($selenium.By.id("password")).sendKeys($secure.MFA_LOGIN_PASSWORD);
})
.then(async function() {
    console.log('Step 4');
    console.log('Tokenの所得及び入力');
    let token = await getSafeToken(5);
    $webDriver.findElement($selenium.By.id("totpmfa")).sendKeys(token);
})
.then(function() {
    console.log('Step 5');
    console.log('Submitのクリック');
    $webDriver.findElement($selenium.By.css('input[type="submit"]')).click();
})
.then(function() {
    console.log('Step 6: ページロードが完了するのを待つ');
    
    // ブラウザの document.readyState が "complete" になるまで最大10秒待機
    return $webDriver.wait(function() {
        return $webDriver.executeScript('return document.readyState').then(function(readyState) {
            return readyState === 'complete';
        });
    }, 10000);
})
.then(function() {
    console.log('Step 7: H1要素が出現するのを待ってテキストを取得');
    // ページロードが完了した状態で、h1要素を探してテキストを取得
    return $webDriver.findElement($selenium.By.tagName('h1')).getText();
})
.then(function(headerText) {
    // 実際のテキストが「Login Success」かを検証
    assert.equal(headerText, 'Login Success');
    console.log('Verification passed: Login Success!');
})
.then(function() {
    //　スクリーンショットを取得
    $webDriver.takeScreenshot();
});

/**
 * 有効期限に余裕のあるTOTP Tokenを取得する関数
 * @param {number} minRemainingSeconds - 最低限必要な残り時間（秒）
 * @returns {Promise<string>} Token
 */
async function getSafeToken(minRemainingSeconds = 5) {
  // 現在のタイムスタンプ（ミリ秒）から、現在の30秒周期の残り時間を計算
  const periodMs = totp.period * 1000;
  let remainingMs = periodMs - (Date.now() % periodMs);
  let remainingSeconds = remainingMs / 1000;

  // 残り時間が指定秒数未満（例: 5秒未満）の場合は、次の周期まで待機
  if (remainingSeconds < minRemainingSeconds) {
    // 次のトークンに切り替わるまでの時間（ミリ秒）待機する
    // 完全に切り替わるよう、少しだけバッファ（+100ms）を足しています
    const waitTime = remainingMs + 100; 
    console.log(`Tokenの残り時間が ${remainingSeconds.toFixed(1)}秒 のため、次のTokenまで ${waitTime / 1000}秒 待機します...`);
    
    await new Promise(resolve => setTimeout(resolve, waitTime));
  }

  // 余裕がある状態（または切り替わった直後）でTokenを生成して返す
  return totp.generate();
}

.then(async function() {
    console.log('Step 4');
    console.log('Tokenの所得及び入力');
    let token = await getSafeToken(5);
    $webDriver.findElement($selenium.By.id("totpmfa")).sendKeys(token);
})