spec/riak/bucket_spec.rb in riak-client-1.0.0.beta vs spec/riak/bucket_spec.rb in riak-client-1.0.0

@@ -2,10 +2,11 @@
 
 describe Riak::Bucket do
   before :each do
     @client = Riak::Client.new
     @backend = mock("Backend")
-    @client.stub!(:backend).and_return(@backend)
+    @client.stub!(:backend).and_yield(@backend)
+    @client.stub!(:http).and_yield(@backend)
     @bucket = Riak::Bucket.new(@client, "foo")
   end
 
   describe "when initializing" do
     it "should require a client and a name" do