two_factor_authentication/spec/support/authenticated_model_helper.rb

module AuthenticatedModelHelper
  def build_guest_user
    GuestUser.new
  end

  def create_user(type = 'encrypted', attributes = {})
    create_table_for_nonencrypted_user if type == 'not_encrypted'

    User.create!(valid_attributes(attributes))
  end

  def create_admin
    Admin.create!(valid_attributes.except(:nickname))
  end

  def valid_attributes(attributes={})
    {
      nickname: 'Marissa',
      email: generate_unique_email,
      password: 'password',
      password_confirmation: 'password'
    }.merge(attributes)
  end

  def generate_unique_email
    @@email_count ||= 0
    @@email_count += 1
    "user#{@@email_count}@example.com"
  end

  def create_table_for_nonencrypted_user
    silence_stream(STDOUT) do
      ActiveRecord::Schema.define(version: 1) do
        create_table 'users', force: :cascade do |t|
          t.string   'email', default: '', null: false
          t.string   'encrypted_password', default: '', null: false
          t.string   'reset_password_token'
          t.datetime 'reset_password_sent_at'
          t.datetime 'remember_created_at'
          t.integer  'sign_in_count', default: 0,  null: false
          t.datetime 'current_sign_in_at'
          t.datetime 'last_sign_in_at'
          t.string   'current_sign_in_ip'
          t.string   'last_sign_in_ip'
          t.datetime 'created_at', null: false
          t.datetime 'updated_at', null: false
          t.integer  'second_factor_attempts_count', default: 0
          t.string   'nickname', limit: 64
          t.string   'otp_secret_key'
          t.string   'direct_otp'
          t.datetime 'direct_otp_sent_at'
        end
      end
    end
  end
end

RSpec.configuration.send(:include, AuthenticatedModelHelper)