associate user sessions to locations.

1 files changed, 9 insertions, 0 deletions

diff --git a/spec/models/user_session_spec.rb b/spec/models/user_session_spec.rb
index df0941e1..10cc706d 100644
--- a/spec/models/user_session_spec.rb
+++ b/spec/models/user_session_spec.rb
@@ -12,8 +12,13 @@ describe UserSession do
 
   describe "#access" do
     let(:request) { double(ip: '192.168.1.1', user_agent: 'blah') }
+    let(:location) { build(:location) }
     let!(:because) { subject.access(request) }
 
+    before :each do
+      Location.stub(:build_from_ip).with('192.168.1.1').and_return(location)
+    end
+
     it "records the time the session was accessed" do
       expect(subject.accessed_at).to_not be_nil
     end
@@ -26,6 +31,10 @@ describe UserSession do
       expect(subject.user_agent).to eql(request.user_agent)
     end
 
+    it 'records a location for the session' do
+      expect(subject.location).to_not be_nil
+    end
+
     it "returns a hash to store in the cookie" do
       expect(because[:value]).to eql(subject.key)
       expect(because[:httponly]).to be_truthy