tfa_test.totp.inc in Two-factor Authentication (TFA) 7.2

Tests for the TfaValidationPluginInterface.

File

tests/includes/tfa_test.totp.inc

<?php

/**
 * @file
 * Tests for the TfaValidationPluginInterface.
 */

/**
 * Class TfaTestTotp.
 */
class TfaTestTotp extends TfaBasePlugin implements TfaValidationPluginInterface {

  /**
   * Data store for testing encryption.
   *
   * @var string
   */
  protected $store;

  /**
   * {@inheritdoc}
   */
  public function ready() {
    return variable_get('tfa_test_is_ready', TRUE);
  }

  /**
   * {@inheritdoc}
   */
  public function getForm(array $form, array &$form_state) {
    $form['code'] = array(
      '#type' => 'textfield',
      '#title' => t('TOTP code'),
      '#required' => TRUE,
    );
    $form['login'] = array(
      '#type' => 'submit',
      '#value' => t('Submit'),
    );
    return $form;
  }

  /**
   * {@inheritdoc}
   */
  public function validateForm(array $form, array &$form_state) {
    $this->code = variable_get('tfa_totp', 123456);
    if (!parent::validate($form_state['values']['code'])) {
      $this->errorMessages['code'] = t('Invalid TOTP code');
      return FALSE;
    }
    else {
      return TRUE;
    }
  }

  /**
   * One-off methods that allow for testing base plugin encryption.
   */
  public function setInStore($data) {
    $this->encryptionKey = drupal_random_bytes(32);
    $this->store = $this
      ->encrypt($data);
  }

  /**
   * One-off methods that allow for testing base plugin encryption.
   */
  public function readFromStore() {
    return $this
      ->decrypt($this->store);
  }

}

Classes

Name	Description
TfaTestTotp	Class TfaTestTotp.