shield 0.0.2 → 0.0.3

data/lib/shield.rb

@@ -1,5 +1,5 @@
 module Shield
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 
   autoload :Password,  "shield/password"
   autoload :Helpers,   "shield/helpers"

data/lib/shield/helpers.rb

@@ -1,10 +1,13 @@
 module Shield
   module Helpers
     def ensure_authenticated(model)
-      return if authenticated(model)
-
-      session[:return_to] = request.fullpath
-      redirect_to_login
+      if authenticated(model)
+        return true
+      else
+        session[:return_to] = request.fullpath
+        redirect_to_login
+        return false
+      end
     end
 
     def authenticated(model)
@@ -37,5 +40,9 @@ module Shield
 
       @_authenticated.delete(model) if defined?(@_authenticated)
     end
+
+    def authenticate(user)
+      session[user.class.to_s] = user.id
+    end
   end
 end

data/test/shield_test.rb

@@ -48,7 +48,7 @@ end
 
 test "ensure_authenticated when logged in" do |context|
   context.session["User"] = 1
-  assert nil == context.ensure_authenticated(User)
+  assert true == context.ensure_authenticated(User)
   assert nil == context.redirect
   assert nil == context.session[:return_to]
 end
@@ -111,4 +111,10 @@ test "logout" do |context|
   assert nil == context.session["User"]
   assert nil == context.session[:return_to]
   assert nil == context.authenticated(User)
+end
+
+test "authenticate" do |context|
+  context.authenticate(User[1001])
+
+  assert User[1] == context.authenticated(User)
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Michel Martens
@@ -16,7 +16,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-12-19 00:00:00 +08:00
+date: 2011-02-16 00:00:00 +08:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency