1 files changed, 6 insertions, 2 deletions
diff --git a/test/authentication/methods/test_publickey.rb b/test/authentication/methods/test_publickey.rb
index 33d1394..f4f544f 100644
--- a/test/authentication/methods/test_publickey.rb
+++ b/test/authentication/methods/test_publickey.rb
@@ -110,7 +110,11 @@ module Authentication; module Methods
       end
 
       def key_manager(options={})
-        @key_manager ||= stub("key_manager", :identities => options[:keys] || keys)
+        @key_manager ||= begin
+          manager = stub("key_manager")
+          manager.stubs(:each_identity).multiple_yields(*(options[:keys] || keys))
+          manager
+        end
       end
 
       def subject(options={})
@@ -120,4 +124,4 @@ module Authentication; module Methods
 
   end
 
-end; end
-\ No newline at end of file
+end; end