seomoz-riak-client 1.0.0.pre

data/seomoz-riak-client.gemspec

@@ -0,0 +1,55 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{seomoz-riak-client}
+  s.version = "1.0.0.pre"
+
+  s.required_rubygems_version = Gem::Requirement.new("> 1.3.1") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Sean Cribbs"]
+  s.date = %q{2011-11-18}
+  s.description = %q{riak-client is a rich client for Riak, the distributed database by Basho. It supports the full HTTP interface including storage operations, bucket configuration, link-walking and map-reduce.}
+  s.email = %q{sean@basho.com}
+  s.files = ["erl_src/riak_kv_test_backend.beam", "erl_src/riak_kv_test_backend.erl", "erl_src/riak_search_test_backend.beam", "erl_src/riak_search_test_backend.erl", "Gemfile", "Guardfile", "lib/active_support/cache/riak_store.rb", "lib/riak/bucket.rb", "lib/riak/cache_store.rb", "lib/riak/client/beefcake/messages.rb", "lib/riak/client/beefcake/object_methods.rb", "lib/riak/client/beefcake_protobuffs_backend.rb", "lib/riak/client/excon_backend.rb", "lib/riak/client/http_backend/configuration.rb", "lib/riak/client/http_backend/key_streamer.rb", "lib/riak/client/http_backend/object_methods.rb", "lib/riak/client/http_backend/request_headers.rb", "lib/riak/client/http_backend/transport_methods.rb", "lib/riak/client/http_backend.rb", "lib/riak/client/net_http_backend.rb", "lib/riak/client/protobuffs_backend.rb", "lib/riak/client/pump.rb", "lib/riak/client/search.rb", "lib/riak/client.rb", "lib/riak/core_ext/blank.rb", "lib/riak/core_ext/extract_options.rb", "lib/riak/core_ext/json.rb", "lib/riak/core_ext/slice.rb", "lib/riak/core_ext/stringify_keys.rb", "lib/riak/core_ext/symbolize_keys.rb", "lib/riak/core_ext/to_param.rb", "lib/riak/core_ext.rb", "lib/riak/encoding.rb", "lib/riak/failed_request.rb", "lib/riak/i18n.rb", "lib/riak/json.rb", "lib/riak/link.rb", "lib/riak/locale/en.yml", "lib/riak/map_reduce/filter_builder.rb", "lib/riak/map_reduce/phase.rb", "lib/riak/map_reduce.rb", "lib/riak/map_reduce_error.rb", "lib/riak/robject.rb", "lib/riak/search.rb", "lib/riak/serializers.rb", "lib/riak/stamp.rb", "lib/riak/test_server.rb", "lib/riak/util/escape.rb", "lib/riak/util/fiber1.8.rb", "lib/riak/util/headers.rb", "lib/riak/util/multipart/stream_parser.rb", "lib/riak/util/multipart.rb", "lib/riak/util/tcp_socket_extensions.rb", "lib/riak/util/translation.rb", "lib/riak/walk_spec.rb", "lib/riak.rb", "Rakefile", "riak-client.gemspec", "seomoz-riak-client.gemspec", "spec/fixtures/cat.jpg", "spec/fixtures/multipart-blank.txt", "spec/fixtures/multipart-mapreduce.txt", "spec/fixtures/multipart-with-body.txt", "spec/fixtures/server.cert.crt", "spec/fixtures/server.cert.key", "spec/fixtures/test.pem", "spec/integration/riak/cache_store_spec.rb", "spec/integration/riak/http_backends_spec.rb", "spec/integration/riak/protobuffs_backends_spec.rb", "spec/integration/riak/test_server_spec.rb", "spec/riak/beefcake_protobuffs_backend_spec.rb", "spec/riak/bucket_spec.rb", "spec/riak/client_spec.rb", "spec/riak/core_ext/to_param_spec.rb", "spec/riak/escape_spec.rb", "spec/riak/excon_backend_spec.rb", "spec/riak/headers_spec.rb", "spec/riak/http_backend/configuration_spec.rb", "spec/riak/http_backend/object_methods_spec.rb", "spec/riak/http_backend/transport_methods_spec.rb", "spec/riak/http_backend_spec.rb", "spec/riak/link_spec.rb", "spec/riak/map_reduce/filter_builder_spec.rb", "spec/riak/map_reduce/phase_spec.rb", "spec/riak/map_reduce_spec.rb", "spec/riak/multipart_spec.rb", "spec/riak/net_http_backend_spec.rb", "spec/riak/robject_spec.rb", "spec/riak/search_spec.rb", "spec/riak/serializers_spec.rb", "spec/riak/stamp_spec.rb", "spec/riak/stream_parser_spec.rb", "spec/riak/walk_spec_spec.rb", "spec/spec_helper.rb", "spec/support/drb_mock_server.rb", "spec/support/http_backend_implementation_examples.rb", "spec/support/integration_setup.rb", "spec/support/mock_server.rb", "spec/support/mocks.rb", "spec/support/test_server.yml.example", "spec/support/unified_backend_examples.rb"]
+  s.homepage = %q{http://seancribbs.github.com/ripple}
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{riak-client is a rich client for Riak, the distributed database by Basho.}
+  s.test_files = ["lib/riak/walk_spec.rb", "spec/integration/riak/cache_store_spec.rb", "spec/integration/riak/http_backends_spec.rb", "spec/integration/riak/protobuffs_backends_spec.rb", "spec/integration/riak/test_server_spec.rb", "spec/riak/beefcake_protobuffs_backend_spec.rb", "spec/riak/bucket_spec.rb", "spec/riak/client_spec.rb", "spec/riak/core_ext/to_param_spec.rb", "spec/riak/escape_spec.rb", "spec/riak/excon_backend_spec.rb", "spec/riak/headers_spec.rb", "spec/riak/http_backend/configuration_spec.rb", "spec/riak/http_backend/object_methods_spec.rb", "spec/riak/http_backend/transport_methods_spec.rb", "spec/riak/http_backend_spec.rb", "spec/riak/link_spec.rb", "spec/riak/map_reduce/filter_builder_spec.rb", "spec/riak/map_reduce/phase_spec.rb", "spec/riak/map_reduce_spec.rb", "spec/riak/multipart_spec.rb", "spec/riak/net_http_backend_spec.rb", "spec/riak/robject_spec.rb", "spec/riak/search_spec.rb", "spec/riak/serializers_spec.rb", "spec/riak/stamp_spec.rb", "spec/riak/stream_parser_spec.rb", "spec/riak/walk_spec_spec.rb"]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<rspec>, ["~> 2.6.0"])
+      s.add_development_dependency(%q<fakeweb>, [">= 1.2"])
+      s.add_development_dependency(%q<rack>, [">= 1.0"])
+      s.add_development_dependency(%q<excon>, ["~> 0.6.1"])
+      s.add_development_dependency(%q<rake>, ["~> 0.8.7"])
+      s.add_runtime_dependency(%q<i18n>, [">= 0.4.0"])
+      s.add_runtime_dependency(%q<builder>, [">= 2.1.2"])
+      s.add_runtime_dependency(%q<beefcake>, ["= 0.3.2"])
+      s.add_runtime_dependency(%q<multi_json>, ["~> 1.0.0"])
+    else
+      s.add_dependency(%q<rspec>, ["~> 2.6.0"])
+      s.add_dependency(%q<fakeweb>, [">= 1.2"])
+      s.add_dependency(%q<rack>, [">= 1.0"])
+      s.add_dependency(%q<excon>, ["~> 0.6.1"])
+      s.add_dependency(%q<rake>, ["~> 0.8.7"])
+      s.add_dependency(%q<i18n>, [">= 0.4.0"])
+      s.add_dependency(%q<builder>, [">= 2.1.2"])
+      s.add_dependency(%q<beefcake>, ["= 0.3.2"])
+      s.add_dependency(%q<multi_json>, ["~> 1.0.0"])
+    end
+  else
+    s.add_dependency(%q<rspec>, ["~> 2.6.0"])
+    s.add_dependency(%q<fakeweb>, [">= 1.2"])
+    s.add_dependency(%q<rack>, [">= 1.0"])
+    s.add_dependency(%q<excon>, ["~> 0.6.1"])
+    s.add_dependency(%q<rake>, ["~> 0.8.7"])
+    s.add_dependency(%q<i18n>, [">= 0.4.0"])
+    s.add_dependency(%q<builder>, [">= 2.1.2"])
+    s.add_dependency(%q<beefcake>, ["= 0.3.2"])
+    s.add_dependency(%q<multi_json>, ["~> 1.0.0"])
+  end
+end

data/spec/fixtures/multipart-blank.txt

@@ -0,0 +1,7 @@
+
+--73NmmA8dJxSB5nL2dVerpFIi8ze
+Content-Type: multipart/mixed; boundary=8fPXq9XfV15txMoV1IbA3hovEij
+
+--8fPXq9XfV15txMoV1IbA3hovEij--
+
+--73NmmA8dJxSB5nL2dVerpFIi8ze--

data/spec/fixtures/multipart-mapreduce.txt

@@ -0,0 +1,10 @@
+
+--NT6cqYFYCfbYZsocVt15tNWCpG9
+Content-Type: application/json
+
+{"phase":0,"data":["source :gemcutter\n\ngem 'i18n'\ngem 'builder'\ngem 'rspec', \"~>2.0.0\"\ngem 'fakeweb', \">=1.2\"\ngem 'rack', '>=1.0'\ngem 'rake'\ngem 'bundler'\ngem 'excon', \"~>0.3.4\"\n\nif defined? JRUBY_VERSION\n  gem 'json'\n  gem 'jruby-openssl'\nelse\n  gem 'curb', '>=0.6'\n  gem 'yajl-ruby'\nend\n\ngroup :integration do\n  gem 'activesupport', '~>3.0'\nend\n"]}
+--NT6cqYFYCfbYZsocVt15tNWCpG9
+Content-Type: application/json
+
+{"phase":0,"data":["source \"http://rubygems.org\"\n\ngem 'rake'\ngem 'gollum-site'\ngem 'rdiscount'\ngem 'RedCloth'\ngem 'rspec'\n"]}
+--NT6cqYFYCfbYZsocVt15tNWCpG9--

data/spec/fixtures/multipart-with-body.txt

@@ -0,0 +1,16 @@
+
+--5EiMOjuGavQ2IbXAqsJPLLfJNlA
+Content-Type: multipart/mixed; boundary=7extjTzvYIKVMVHowUiTn0LfvSs
+
+--7extjTzvYIKVMVHowUiTn0LfvSs
+X-Riak-Vclock: a85hYGBgyWDKBVHMr9s3ZzAlMuaxMtyZcPAIH1RYyObHDqiwxIZjcOG1M98chAq3bUQIz7SSFQEKM4FUbwMKZwEA
+Location: /riak/foo/baz
+Content-Type: text/plain
+Link: </riak/foo>; rel="up"
+Etag: 6JdI51eFrvv5lDwY6un7a2
+Last-Modified: Sat, 16 Jan 2010 22:13:44 GMT
+
+SCP sloooow....
+--7extjTzvYIKVMVHowUiTn0LfvSs--
+
+--5EiMOjuGavQ2IbXAqsJPLLfJNlA--

data/spec/fixtures/server.cert.crt

@@ -0,0 +1,15 @@
+-----BEGIN CERTIFICATE-----
+MIICPzCCAagCCQDfZhEdJjSgTDANBgkqhkiG9w0BAQUFADBkMQswCQYDVQQGEwJV
+UzELMAkGA1UECBMCTkMxEjAQBgNVBAcTCUNoYXJsb3R0ZTEOMAwGA1UEChMFQmFz
+aG8xEDAOBgNVBAsTB1N1cHBvcnQxEjAQBgNVBAMTCTEyNy4wLjAuMTAeFw0xMTAz
+MjYxNjMzMzNaFw0zODA4MTAxNjMzMzNaMGQxCzAJBgNVBAYTAlVTMQswCQYDVQQI
+EwJOQzESMBAGA1UEBxMJQ2hhcmxvdHRlMQ4wDAYDVQQKEwVCYXNobzEQMA4GA1UE
+CxMHU3VwcG9ydDESMBAGA1UEAxMJMTI3LjAuMC4xMIGfMA0GCSqGSIb3DQEBAQUA
+A4GNADCBiQKBgQCtfHMCsR86HQEAsO52/WvtFWCEigXll0rxYjXvdqeszEPEMFIy
+Qlg3GlRxH51YrBzp46ReF9Qv5sf0Nh6SR7HGzlWmMEVfPeuAKcz1fzVcjD+IXWHK
+qklAQjpxz+18dvaGxQ7ZJMPsBlq1v64siTLXI0yhjxXOuQPJWrWsvuuXkQIDAQAB
+MA0GCSqGSIb3DQEBBQUAA4GBAIqt1A5Ah7s2oUoYQ8YCKC83fKbXbbNCiLFLwIzy
+TGYXd8j7JTfeY8ettbtitlYgP+ouf23LzonuMo47GRuMgVKRWm4l+ZVMP5Qbkx9t
+uspx+6lHUWnMT9aRdP9/2I7dscyfuhtzs0UxddADLzL9Cif4Y06E1NhR/KK+zo46
+Nep8
+-----END CERTIFICATE-----

data/spec/fixtures/server.cert.key

@@ -0,0 +1,15 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIICXQIBAAKBgQCtfHMCsR86HQEAsO52/WvtFWCEigXll0rxYjXvdqeszEPEMFIy
+Qlg3GlRxH51YrBzp46ReF9Qv5sf0Nh6SR7HGzlWmMEVfPeuAKcz1fzVcjD+IXWHK
+qklAQjpxz+18dvaGxQ7ZJMPsBlq1v64siTLXI0yhjxXOuQPJWrWsvuuXkQIDAQAB
+AoGBAJIguzdPPgBTId8VKSes+lVupie9oo3qy8NaeBfGGCIixAnisbmHzIpNcUb/
+3CcuggQ4LODcrWvTtiTr2QBZx1FL7E4POBJl/N7zJMaQd+pGjmrJGfv5haSSQN+H
+r74Ix3HCd0RPdSgt5pmlT4KfsqkfRmqsPd6Nw54zbyLFMlTpAkEA4ar3ZJi2+Y+u
+FH3AycXuPdDtVW0tKFtxfvKlS48gshB6gmkd06Ugss5eZkdbSY0voAp88Tr2shOJ
++pXc++Zl6wJBAMTN9K9k728cVCf41pR6mDVxIaaqjJeY4DWppGQFSqw/fmYz/Quu
+PlTvk6pGRYiGN6y9CZFNoL2I/SWcd4ukrXMCQCyfYOHsbKn2Zka5Awki8VQZ3wQ4
+XWiQhGXE1ziUqbNsHL1yyaoTCd8xfWseCwgFOficek49CZD22h7JyXOqAFcCQCn2
+mFPFu9//NFqJjod+VHIgu0IkX3H7oOMQVwMUtcVgjH0SXMRe1N+bbesCrNTdeYWV
+kTKwULPZP9EDOeJGrM0CQQCsX+8VZ15yKTy6ADINrOt26PNpD4ib4552TE6T/1wG
+LKdjn5l0qB5K7ILc22z3LCenNBa0Uxbg5/RSdoX57aHA
+-----END RSA PRIVATE KEY-----

data/spec/fixtures/test.pem

@@ -0,0 +1 @@
+i-am-a-pem

data/spec/integration/riak/cache_store_spec.rb

@@ -0,0 +1,154 @@
+require 'spec_helper'
+require 'riak/cache_store'
+
+describe Riak::CacheStore do
+  before :all do
+    if $test_server
+      @web_port = 9000
+      $test_server.start
+    end
+  end
+
+  before do
+    @web_port ||= 8098
+    @cache = ActiveSupport::Cache.lookup_store(:riak_store, :http_port => @web_port)
+    @cleanup = true
+  end
+
+  after do
+    if @cleanup
+      if $test_server
+        $test_server.recycle
+        Thread.current[:curl_easy_handle] = nil
+      else
+        @cache.bucket.keys(:force => true).each do |k|
+          @cache.bucket.delete(k, :rw => 1) unless k.blank?
+        end
+      end
+    end
+  end
+
+  describe "Riak integration" do
+    before do
+      @cleanup = false
+    end
+
+    it "should have a client" do
+      @cache.should respond_to(:client)
+      @cache.client.should be_kind_of(Riak::Client)
+    end
+
+    it "should have a bucket to store entries in" do
+      @cache.bucket.should be_kind_of(Riak::Bucket)
+    end
+
+    it "should configure the client according to the initialized options" do
+      @cache = ActiveSupport::Cache.lookup_store(:riak_store, :http_port => 10000)
+      @cache.client.http_port.should == 10000
+    end
+
+    it "should choose the bucket according to the initializer option" do
+      @cache = ActiveSupport::Cache.lookup_store(:riak_store, :bucket => "foobar", :http_port => @web_port)
+      @cache.bucket.name.should == "foobar"
+    end
+
+    it "should set the N value to 2 by default" do
+      @cache.bucket.n_value.should == 2
+    end
+
+    it "should set the N value to the specified value" do
+      @cache = ActiveSupport::Cache.lookup_store(:riak_store, :n_value => 1, :http_port => @web_port)
+      @cache.bucket.n_value.should == 1
+    end
+
+    it "should set the bucket R value to 1 by default" do
+      @cache.bucket.r.should == 1
+    end
+
+    it "should set the bucket R default to the specified value" do
+      @cache = ActiveSupport::Cache.lookup_store(:riak_store, :r => "quorum", :http_port => @web_port)
+      @cache.bucket.r.should == "quorum"
+    end
+
+    it "should set the bucket W value to 1 by default" do
+      @cache.bucket.w.should == 1
+    end
+
+    it "should set the bucket W default to the specified value" do
+      @cache = ActiveSupport::Cache.lookup_store(:riak_store, :w => "all", :http_port => @web_port)
+      @cache.bucket.w.should == "all"
+    end
+
+    it "should set the bucket DW value to 0 by default" do
+      @cache.bucket.dw.should == 0
+    end
+
+    it "should set the bucket DW default to the specified value" do
+      @cache = ActiveSupport::Cache.lookup_store(:riak_store, :dw => "quorum", :http_port => @web_port)
+      @cache.bucket.dw.should == "quorum"
+    end
+
+    it "should set the bucket RW value to quorum by default" do
+      @cache.bucket.rw.should == "quorum"
+    end
+
+    it "should set the bucket RW default to the specified value" do
+      @cache = ActiveSupport::Cache.lookup_store(:riak_store, :rw => "all", :http_port => @web_port)
+      @cache.bucket.rw.should == "all"
+    end
+  end
+
+
+  it "should read and write strings" do
+    @cache.write('foo', 'bar')
+    @cache.read('foo').should == 'bar'
+  end
+
+  it "should read and write hashes" do
+    @cache.write('foo', {:a => "b"})
+    @cache.read('foo').should == {:a => "b"}
+  end
+
+  it "should read and write integers" do
+    @cache.write('foo', 1)
+    @cache.read('foo').should == 1
+  end
+
+  it "should read and write nil" do
+    @cache.write('foo', nil)
+    @cache.read('foo').should be_nil
+  end
+
+  it "should return the stored value when fetching on hit" do
+    @cache.write('foo', 'bar')
+    @cache.fetch('foo'){'baz'}.should == 'bar'
+  end
+
+  it "should return the default value when fetching on miss" do
+    @cache.fetch('foo'){ 'baz' }.should == 'baz'
+  end
+
+  it "should return the default value when forcing a miss" do
+    @cache.fetch('foo', :force => true){'bar'}.should == 'bar'
+  end
+
+  it "should detect if a value exists in the cache" do
+    @cache.write('foo', 'bar')
+    @cache.exist?('foo').should be_true
+  end
+
+  it "should delete matching keys from the cache" do
+    @cache.write('foo', 'bar')
+    @cache.write('green', 'thumb')
+    @cache.delete_matched(/foo/)
+    @cache.read('foo').should be_nil
+    @cache.read('green').should == 'thumb'
+  end
+
+  it "should delete a single key from the cache" do
+    @cache.write('foo', 'bar')
+    @cache.read('foo').should == 'bar'
+    @cache.delete('foo')
+    @cache.read('foo').should be_nil
+  end
+end

data/spec/integration/riak/http_backends_spec.rb

@@ -0,0 +1,58 @@
+require 'spec_helper'
+
+describe "HTTP" do
+  before :all do
+    if $test_server
+      @web_port = 9000
+      $test_server.start
+    end
+  end
+
+  before do
+    @web_port ||= 8098
+    @client = Riak::Client.new(:http_port => @web_port)
+  end
+
+  after do
+    $test_server.recycle if $test_server.started?
+  end
+
+  [:ExconBackend, :NetHTTPBackend].each do |klass|
+    bklass = Riak::Client.const_get(klass)
+    if bklass.configured?
+      describe klass.to_s do
+        before do
+          @backend = bklass.new(@client)
+        end
+
+        it_should_behave_like "Unified backend API"
+      end
+    end
+  end
+
+  class Reader < Array
+    def read(*args)
+      shift
+    end
+
+    def size
+      join.size
+    end
+  end
+
+  class SizelessReader < Reader
+    undef :size
+  end
+
+  describe 'NetHTTPBackend' do
+    subject { Riak::Client::NetHTTPBackend.new(@client) }
+    let(:file) { File.open(__FILE__) }
+    let(:sized) { Reader.new(["foo", "bar", "baz"]) }
+    let(:sizeless) { SizelessReader.new(["foo", "bar", "baz"]) }
+    it "should set the content-length or transfer-encoding properly on IO uploads" do
+      lambda { subject.put(204, "/riak/nethttp", "test-file", file, {"Content-Type" => "text/plain"}) }.should_not raise_error
+      lambda { subject.put(204, "/riak/nethttp", "test-sized", sized, {"Content-Type" => "text/plain"}) }.should_not raise_error
+      lambda { subject.put(204, "/riak/nethttp", "test-file", sizeless, {"Content-Type" => "text/plain"}) }.should_not raise_error
+    end
+  end
+end

data/spec/integration/riak/protobuffs_backends_spec.rb

@@ -0,0 +1,32 @@
+require 'spec_helper'
+
+describe "Protocol Buffers" do
+  before :all do
+    if $test_server
+      @pbc_port = 9002
+      $test_server.start
+    end
+  end
+
+  before do
+    @pbc_port ||= 8087
+    @client = Riak::Client.new(:pb_port => @pbc_port, :protocol => "pbc")
+  end
+
+  after do
+    $test_server.recycle if $test_server.started?
+  end
+
+  [:BeefcakeProtobuffsBackend].each do |klass|
+    bklass = Riak::Client.const_get(klass)
+    if bklass.configured?
+      describe klass.to_s do
+        before do
+          @backend = bklass.new(@client)
+        end
+
+        it_should_behave_like "Unified backend API"
+      end
+    end
+  end
+end

data/spec/integration/riak/test_server_spec.rb

@@ -0,0 +1,161 @@
+require 'spec_helper'
+
+if $test_server
+  describe Riak::TestServer do
+    before do
+      @server = $test_server
+    end
+
+    after do
+      @server.stop
+      @server.cleanup
+    end
+
+    describe "isolation from and modification of the existing install" do
+      before do
+        @server.prepare!
+        @riak_bin    = "#{@server.temp_dir}/bin/riak"
+        @vm_args     = "#{@server.temp_dir}/etc/vm.args"
+        @app_config  = "#{@server.temp_dir}/etc/app.config"
+      end
+
+      describe "for app.config" do
+        it "should create the app.config file in the temporary directory" do
+          File.should be_exist(File.expand_path(@app_config))
+        end
+
+        it "should be a correct Erlang config" do
+          config = File.read(@app_config)
+          config[-2..-1].should == '].'
+          config[0..0].should == '['
+        end
+
+        it "should set the backend to use the test backend" do
+          File.readlines(@app_config).should be_any do |line|
+            line =~ /\{storage_backend\s*,\s*(.*)\}/ && $1 == "riak_kv_test_backend"
+          end
+        end
+
+        it "should set the default ports to 9000-9002" do
+          config = File.readlines(@app_config)
+          config.should be_any do |line|
+            line =~ /\{web_port\s*,\s*(.*)\}/ && $1 == "9000"
+          end
+          config.should be_any do |line|
+            line =~ /\{handoff_port\s*,\s*(.*)\}/ && $1 == "9001"
+          end
+          config.should be_any do |line|
+            line =~ /\{pb_port\s*,\s*(.*)\}/ && $1 == "9002"
+          end
+        end
+
+        it "should set the ring directory to point to the temporary directory" do
+          config = File.readlines(@app_config)
+          config.should be_any do |line|
+            line =~ /\{ring_state_dir\s*,\s*(.*)\}/ && $1 == File.join(@server.temp_dir, "data", "ring")
+          end
+        end
+      end
+
+      describe "for vm.args" do
+        it "should create the vm.args file in the temporary directory" do
+          File.should be_exist(File.expand_path(@vm_args))
+        end
+
+        it "should set a quasi-random node name" do
+          File.readlines(@vm_args).should be_any do |line|
+            line =~ /^-name (.*)/ && $1 =~ /riaktest\d+@/
+          end
+        end
+
+        it "should set a quasi-random cookie" do
+          File.readlines(@vm_args).should be_any do |line|
+            line =~ /^-setcookie (.*)/ && $1 != "riak"
+          end
+        end
+      end
+
+      describe "for the riak script" do
+        it "should create the script in the temporary directory" do
+          File.should be_exist(File.expand_path(@riak_bin))
+        end
+
+        it "should modify the RUNNER_SCRIPT_DIR to point to the temporary directory" do
+          File.readlines(@riak_bin).should be_any do |line|
+            line =~ /RUNNER_SCRIPT_DIR=(.*)/ && $1 == File.expand_path("#{@server.temp_dir}/bin")
+
+          end
+        end
+
+        it "should modify the RUNNER_ETC_DIR to point to the temporary directory" do
+          File.readlines(@riak_bin).should be_any do |line|
+            line =~ /RUNNER_ETC_DIR=(.*)/ && $1 == File.expand_path("#{@server.temp_dir}/etc")
+          end
+        end
+
+        it "should modify the RUNNER_USER to point to the current user" do
+          File.readlines(@riak_bin).should be_any do |line|
+            line =~ /RUNNER_USER=(.*)/ && $1 == (ENV['USER'] || `whoami`)
+          end
+        end
+
+        it "should modify the RUNNER_LOG_DIR to point to the temporary directory" do
+          File.readlines(@riak_bin).should be_any do |line|
+            line =~ /RUNNER_LOG_DIR=(.*)/ && $1 == File.expand_path("#{@server.temp_dir}/log")
+          end
+        end
+
+        it "should modify the RUNNER_BASE_DIR so that it is not relative" do
+          File.readlines(@riak_bin).should be_any do |line|
+            line =~ /RUNNER_BASE_DIR=(.*)/ && $1.strip != "${RUNNER_SCRIPT_DIR%/*}" && File.directory?($1)
+          end
+        end
+
+        it "should modify the PIPE_DIR to point to the temporary directory" do
+          File.readlines(@riak_bin).should be_any do |line|
+            line =~ /PIPE_DIR=(.*)/ &&  $1 == File.expand_path("#{@server.temp_dir}/pipe") && File.directory?($1)
+          end
+        end
+      end
+    end
+
+    it "should cleanup the existing config" do
+      @server.prepare!
+      @server.cleanup
+      File.should_not be_directory(@server.temp_dir)
+    end
+
+    it "should start Riak in the background" do
+      @server.prepare!
+      @server.start.should be_true
+      @server.should be_started
+    end
+
+    it "should stop a started test server" do
+      @server.prepare!
+      @server.start.should be_true
+      @server.stop
+      @server.should_not be_started
+    end
+
+    it "should recycle the server contents" do
+      begin
+        @server.prepare!
+        @server.start.should be_true
+
+        client = Riak::Client.new(:http_port => 9000)
+        obj = client['test_bucket'].new("test_item")
+        obj.data = {"data" => "testing"}
+        obj.store rescue nil
+
+        @server.recycle
+        @server.should be_started
+        lambda do
+          client['test_bucket']['test_item']
+        end.should raise_error(Riak::FailedRequest)
+      ensure
+        @server.stop
+      end
+    end
+  end
+end