Bugfix/store recognisable session on registration

app/controllers/devise_recognisable/registrations_controller.rb

@@ -0,0 +1,17 @@
+
+class DeviseRecognisable::RegistrationsController < Devise::RegistrationsController
+  append_after_action :store_recognisable_details, only: :create
+
+  # After the user has been signed up, we save the ip address in the
+  # DeviseRecognisable::RecognisableSession table.
+  def store_recognisable_details
+    return unless resource.persisted?
+    DeviseRecognisable::RecognisableSession.create!(
+      recognisable: resource_class.find_by(email: params[resource_name][:email]),
+      sign_in_ip: request.location.ip,
+      sign_in_at: Time.now,
+      user_agent: request.user_agent,
+      accept_header: request.headers["HTTP_ACCEPT"]
+    )
+  end
+end

lib/devise-recognisable/mapping.rb

@@ -7,6 +7,7 @@ def default_controllers(options)
       options[:controllers] ||= {}
       # Use our SessionsController instead of Devise's.
       options[:controllers][:sessions] ||= 'devise_recognisable/sessions'
+      options[:controllers][:registrations] ||= 'devise_recognisable/registrations'
       super
     end
 

spec/dummy-app/spec/features/sign_in_spec.rb

@@ -62,7 +62,7 @@
         fill_in 'Email', with: email
         fill_in 'Password', with: password
         click_button 'Log in'
-      }.to change { DeviseRecognisable::RecognisableSession.count }.from(0).to(1)
+      }.to change { DeviseRecognisable::RecognisableSession.count }.from(1).to(2)
     end
 
     it "doesn't create a new DeviseRecognisable::RecognisableSession on unsuccessful sign in" do

spec/dummy-app/spec/features/sign_up_spec.rb

@@ -0,0 +1,23 @@
+require 'rails_helper'
+
+RSpec.feature "Sign in" do
+  let(:email) { Faker::Internet.email }
+  let(:password) { Faker::Internet.password }
+
+  context 'as a new user' do
+    it 'creates a devise recognisable session' do
+      visit '/'
+      expect(page).to have_content 'Welcome to my website'
+      click_link 'Log in'
+      click_link 'Sign up'
+      fill_in 'Email', with: email
+      fill_in 'Password', with: password
+      fill_in 'Password confirmation', with: password
+      expect {
+        click_button 'Sign up'
+      }.to change(DeviseRecognisable::RecognisableSession, :count).by(1)
+      expect(page).to have_content 'Welcome!'
+      expect(page).to have_content('signed up successfully')
+    end
+  end
+end