Introducing first spec to test otp code.

lib/two_factor_authentication.rb

@@ -2,6 +2,7 @@ require 'two_factor_authentication/version'
 require 'devise'
 require 'active_support/concern'
 require "active_model"
+require "active_record"
 require "active_support/core_ext/class/attribute_accessors"
 require "cgi"
 require "rotp"

spec/lib/two_factor_authentication/models/two_factor_authenticatable_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+include AuthenticatedModelHelper
+
+
+describe Devise::Models::TwoFactorAuthenticatable, '#otp_code' do
+  let(:instance) { AuthenticatedModelHelper.create_new_user }
+  subject { instance.otp_code(time) }
+  let(:time) { 1392852456 }
+
+  it "should return an error if no secret is set" do
+    expect {
+      subject
+    }.to raise_error
+  end
+
+  context "secret is set" do
+    before :each do
+      instance.otp_secret_key = "2z6hxkdwi3uvrnpn"
+    end
+
+    it "should not return an error" do
+      subject
+    end
+
+    context "with a known time" do
+      let(:time) { 1392852756 }
+
+      it "should return a known result" do
+        expect(subject).to eq(562202)
+      end
+    end
+  end
+end

spec/spec_helper.rb

@@ -3,6 +3,7 @@ require "bundler/setup"
 
 require 'two_factor_authentication'
 
+
 Dir["#{Dir.pwd}/spec/support/**/*.rb"].each {|f| require f}
 
 

spec/support/authenticated_model_helper.rb

@@ -0,0 +1,18 @@
+module AuthenticatedModelHelper
+
+  class User
+    extend ActiveModel::Callbacks
+    include ActiveModel::Validations
+    include Devise::Models::TwoFactorAuthenticatable
+
+    define_model_callbacks :create
+    attr_accessor :otp_secret_key, :email
+
+    has_one_time_password
+  end
+
+  def create_new_user
+    User.new
+  end
+
+end