Spec: validate that send_two_factor_authentication_code can be overwritten

lib/two_factor_authentication/models/two_factor_authenticatable.rb

@@ -50,18 +50,18 @@ module Devise
           self.send("#{self.class.otp_column_name}=", attr)
         end
 
-      end
+        def need_two_factor_authentication?(request)
+          true
+        end
 
-      def need_two_factor_authentication?(request)
+        def send_two_factor_authentication_code
-        true
+          raise NotImplementedError.new("No default implementation - please define in your class.")
-      end
+        end
 
-      def send_two_factor_authentication_code(code)
+        def max_login_attempts?
-        p "Code is #{code}"
+          second_factor_attempts_count >= self.class.max_login_attempts
-      end
+        end
 
-      def max_login_attempts?
-        second_factor_attempts_count >= self.class.max_login_attempts
       end
     end
   end

spec/lib/two_factor_authentication/models/two_factor_authenticatable_spec.rb

@@ -52,4 +52,19 @@ describe Devise::Models::TwoFactorAuthenticatable, '#authenticate_otp' do
     code = instance.otp_code(1.minutes.ago.to_i)
     expect(do_invoke(code)).to eq(false)
   end
+end
+
+describe Devise::Models::TwoFactorAuthenticatable, '#send_two_factor_authentication_code' do
+
+  it "should raise an error by default" do
+    instance = AuthenticatedModelHelper.create_new_user
+    expect {
+      instance.send_two_factor_authentication_code
+    }.to raise_error(NotImplementedError)
+  end
+
+  it "should be overrideable" do
+    instance = AuthenticatedModelHelper.create_new_user_with_overrides
+    expect(instance.send_two_factor_authentication_code).to eq("Code sent")
+  end
 end

spec/support/authenticated_model_helper.rb

@@ -11,8 +11,19 @@ module AuthenticatedModelHelper
     has_one_time_password
   end
 
+  class UserWithOverrides < User
+
+    def send_two_factor_authentication_code
+      "Code sent"
+    end
+  end
+
   def create_new_user
     User.new
   end
 
+  def create_new_user_with_overrides
+    UserWithOverrides.new
+  end
+
 end