rhoconnect 4.0.4 → 5.1.1

data/spec/server/stats_spec.rb

@@ -8,18 +8,18 @@ STATS_RECORD_SIZE = 8 unless defined? STATS_RECORD_SIZE
 include Rhoconnect
 
 describe "Middleware" do
-  
+
   before(:each) do
     @now = 10.0
     Store.flush_all
-    app = mock('app')
-    app.stub!(:call)
+    app = double('app')
+    app.stub(:call)
     Rhoconnect.stats = true
     Rhoconnect::Server.enable :stats
     @middleware_new_routes = Rhoconnect::Middleware::Stats.new(app)
-    Store.stub!(:lock).and_yield
+    Store.stub(:lock).and_yield
   end
-  
+
   after(:each) do
     Rhoconnect.stats = false
     Rhoconnect::Server.disable :stats
@@ -27,7 +27,7 @@ describe "Middleware" do
 
   it "should compute http average" do
     @incr = 0
-    Time.stub!(:now).and_return do
+    Time.stub(:now).and_return do
       if @incr > 0
         @now += 0.3
         @incr -= 1
@@ -45,13 +45,13 @@ describe "Middleware" do
     metric = 'http:GET:/api/application/query:SampleAdapter'
     Rhoconnect::Stats::Record.key(metric).should == "stat:#{metric}"
 
-    # The conversion algorithm (float to string) currently checks two precisions. 
-    # it tries 16 digits and if that's not enough it then uses 17.       
+    # The conversion algorithm (float to string) currently checks two precisions.
+    # it tries 16 digits and if that's not enough it then uses 17.
     Rhoconnect::Stats::Record.range(metric, 0, -1).should == [
-      "2.0,0.6000000000000014:12", 
-      "2.0,0.6000000000000014:14", 
-      "2.0,0.6000000000000014:16", 
+      "2.0,0.6000000000000014:12",
+      "2.0,0.6000000000000014:14",
+      "2.0,0.6000000000000014:16",
       "2.0,0.6000000000000014:18"
     ]
-  end  
+  end
 end

data/spec/source_adapter_spec.rb

@@ -1,154 +1,153 @@
 require File.join(File.dirname(__FILE__),'spec_helper')
 
 describe "loading Model" do
-  it_behaves_like "SharedRhoconnectHelper", :rhoconnect_data => true do
-    it "should load Model instance from user name" do
-      username = 'testuser'
-      model_instance = SimpleAdapter.load(username)
-      model_instance.class.name.should == 'SimpleAdapter'
-      model_instance.source.name.should == 'SimpleAdapter'
-    end
+  include_examples "SharedRhoconnectHelper", :rhoconnect_data => true
+
+  it "should load Model instance from user name" do
+    username = 'testuser'
+    model_instance = SimpleAdapter.load(username)
+    model_instance.class.name.should == 'SimpleAdapter'
+    model_instance.source.name.should == 'SimpleAdapter'
   end
 end
 
 describe "Model" do
-  it_behaves_like "SharedRhoconnectHelper", :rhoconnect_data => true do
-    before(:each) do
-      @s = Source.load('SimpleAdapter',@s_params)
-      @sa = Rhoconnect::Model::Base.create(@s)
-    end
+  include_examples "SharedRhoconnectHelper", :rhoconnect_data => true
 
-    def setup_adapter(name)
-      fields = {
-        :name => name,
-        :url => 'http://example.com',
-        :login => 'testuser',
-        :password => 'testpass',
-      }
-      Source.create(fields,@s_params)
-    end
+  before(:each) do
+    @s = Source.load('SimpleAdapter',@s_params)
+    @sa = Rhoconnect::Model::Base.create(@s)
+  end
 
-    it "should create model with source" do
-      @sa.class.name.should == @s.name
-    end
-    
-    it "should create all existing pre-defined models" do
-      Rhoconnect.predefined_sources.keys.sort.should == ['RhoInternalBenchmarkAdapter'].sort
-      Rhoconnect.predefined_sources.each do |adapter_name, filename|
-        pas = Source.load(adapter_name, @s_params)
-        pas.should_not == nil
-        pa = Rhoconnect::Model::Base.create(pas)
-        pa.class.name.should == "Rhoconnect::#{adapter_name}"
-      end
-    end
-    
-    it "should create DynamicAdapterModel" do
-      @sa1 = Rhoconnect::Model::Base.create(@s2)
-      @sa1.class.name.should == 'Rhoconnect::Model::DynamicAdapterModel'
-    end
-    
-    it "should capture exception in create" do
-      Rhoconnect::Model::DynamicAdapterModel.should_receive(:new).once.and_raise(Exception)
-      lambda { @sa1 = Rhoconnect::Model::Base.create(@s2) }.should raise_error(Exception)
-    end
+  def setup_adapter(name)
+    fields = {
+      :name => name,
+      :url => 'http://example.com',
+      :login => 'testuser',
+      :password => 'testpass',
+    }
+    Source.create(fields,@s_params)
+  end
 
-    it "should fail to create Bro-ken model" do
-      broken_source = setup_adapter('Bro-ken')    
-      lambda { Rhoconnect::Model::Base.create(broken_source) }.should raise_error(Exception)
-      broken_source.delete
-    end
+  it "should create model with source" do
+    @sa.class.name.should == @s.name
+  end
 
-    it "should create model with trailing spaces" do
-      s = setup_adapter('SimpleAdapter ')
-      Rhoconnect::Model::Base.create(s).is_a?(SimpleAdapter).should be_true
+  it "should create all existing pre-defined models" do
+    Rhoconnect.predefined_sources.keys.sort.should == ['RhoInternalBenchmarkAdapter'].sort
+    Rhoconnect.predefined_sources.each do |adapter_name, filename|
+      pas = Source.load(adapter_name, @s_params)
+      pas.should_not == nil
+      pa = Rhoconnect::Model::Base.create(pas)
+      pa.class.name.should == "Rhoconnect::#{adapter_name}"
     end
+  end
 
-    describe "model methods" do
-      it "should execute Model login method with source vars" do
-        @sa.login.should == true
-      end
+  it "should create DynamicAdapterModel" do
+    @sa1 = Rhoconnect::Model::Base.create(@s2)
+    @sa1.class.name.should == 'Rhoconnect::Model::DynamicAdapterModel'
+  end
 
-      it "should get Model data (:md)" do
-        expected = {'1'=>@product1,'2'=>@product2}
-        @s.put_data(:md, expected)
-        @sa.get_data.should == expected  
-      end
+  it "should capture exception in create" do
+    Rhoconnect::Model::DynamicAdapterModel.should_receive(:new).once.and_raise(Exception)
+    lambda { @sa1 = Rhoconnect::Model::Base.create(@s2) }.should raise_error(Exception)
+  end
 
-      it "should get Model data from specific document" do
-        expected = {'4'=>@product1,'5'=>@product2}
-        Source.define_valid_doctypes([:md_custom])
-        @s.put_data(:md_custom, expected)
-        @sa.get_data(:md_custom).should == expected  
-        Source.valid_doctypes.delete(:md_custom)
-      end
+  it "should fail to create Bro-ken model" do
+    broken_source = setup_adapter('Bro-ken')
+    lambda { Rhoconnect::Model::Base.create(broken_source) }.should raise_error(Exception)
+    broken_source.delete
+  end
 
-      it "should execute Model query method" do
-        expected = {'1'=>@product1,'2'=>@product2}
-        @sa.inject_result expected
-        @sa.query.should == expected
-      end
+  it "should create model with trailing spaces" do
+    s = setup_adapter('SimpleAdapter ')
+    Rhoconnect::Model::Base.create(s).is_a?(SimpleAdapter).should be_true
+  end
 
-      it "should execute Model search method and modify params" do
-        params = {:hello => 'world'}
-        expected = {'1'=>@product1,'2'=>@product2}
-        @sa.inject_result expected
-        @sa.search(params).should == expected
-        params.should == {:hello => 'world', :foo => 'bar'}
-      end
+  describe "model methods" do
+    it "should execute Model login method with source vars" do
+      @sa.login.should == true
+    end
 
-      it "should execute Model login with current_user" do
-        @sa.should_receive(:current_user).with(no_args()).and_return(@u)
-        @sa.login
-      end
+    it "should get Model data (:md)" do
+      expected = {'1'=>@product1,'2'=>@product2}
+      @s.put_data(:md, expected)
+      @sa.get_data.should == expected
+    end
 
-      it "should execute Model sync method" do
-        expected = {'1'=>@product1,'2'=>@product2}
-        @sa.inject_result expected
-        @sa.do_query
-        Store.get_data(@s.docname(:md)).should == expected
-        Store.get_value(@s.docname(:md_size)).to_i.should == 2
-      end
+    it "should get Model data from specific document" do
+      expected = {'4'=>@product1,'5'=>@product2}
+      Source.define_valid_doctypes([:md_custom])
+      @s.put_data(:md_custom, expected)
+      @sa.get_data(:md_custom).should == expected
+      Source.valid_doctypes.delete(:md_custom)
+    end
 
-      it "should execute Model sync method with nil result" do
-        @sa.inject_result nil
-        @sa.do_query
-        Store.get_data(@s.docname(:md)).should == {}
-      end
+    it "should execute Model query method" do
+      expected = {'1'=>@product1,'2'=>@product2}
+      @sa.inject_result expected
+      @sa.query.should == expected
+    end
 
-      it "should fail gracefully if @result is missing" do
-        @sa.inject_result nil
-        lambda { @sa.query }.should_not raise_error
-      end
+    it "should execute Model search method and modify params" do
+      params = {:hello => 'world'}
+      expected = {'1'=>@product1,'2'=>@product2}
+      @sa.inject_result expected
+      @sa.search(params).should == expected
+      params.should == {:hello => 'world', :foo => 'bar'}
+    end
 
-      it "should reset count if @result is empty" do
-        @sa.inject_result({'1'=>@product1,'2'=>@product2})
-        @sa.do_query
-        Store.get_value(@s.docname(:md_size)).to_i.should == 2
-        @sa.inject_result({})
-        @sa.do_query
-        Store.get_value(@s.docname(:md_size)).to_i.should == 0
-      end
+    it "should execute Model login with current_user" do
+      @sa.should_receive(:current_user).with(no_args()).and_return(@u)
+      @sa.login
+    end
 
-      it "should execute Model create method" do
-        @sa.create(@product4).should == 'obj4'
-      end
+    it "should execute Model sync method" do
+      expected = {'1'=>@product1,'2'=>@product2}
+      @sa.inject_result expected
+      @sa.do_query
+      Store.get_data(@s.docname(:md)).should == expected
+      Store.get_value(@s.docname(:md_size)).to_i.should == 2
+    end
 
-      it "should stash @result in store and set it to nil" do
-        expected = {'1'=>@product1,'2'=>@product2}
-        Source.define_valid_doctypes(['tmpdoc'.to_sym])
-        @sa.inject_result(expected)
-        @sa.inject_tmpdoc('tmpdoc')
-        @sa.stash_result
-        @s.get_data('tmpdoc').should == expected
-        Source.valid_doctypes.delete('tmpdoc'.to_sym)
-      end
+    it "should execute Model sync method with nil result" do
+      @sa.inject_result nil
+      @sa.do_query
+      Store.get_data(@s.docname(:md)).should == {}
+    end
+
+    it "should fail gracefully if @result is missing" do
+      @sa.inject_result nil
+      lambda { @sa.query }.should_not raise_error
+    end
 
-      describe "Model metadata method" do
+    it "should reset count if @result is empty" do
+      @sa.inject_result({'1'=>@product1,'2'=>@product2})
+      @sa.do_query
+      Store.get_value(@s.docname(:md_size)).to_i.should == 2
+      @sa.inject_result({})
+      @sa.do_query
+      Store.get_value(@s.docname(:md_size)).to_i.should == 0
+    end
+
+    it "should execute Model create method" do
+      @sa.create(@product4).should == 'obj4'
+    end
+
+    it "should stash @result in store and set it to nil" do
+      expected = {'1'=>@product1,'2'=>@product2}
+      Source.define_valid_doctypes(['tmpdoc'.to_sym])
+      @sa.inject_result(expected)
+      @sa.inject_tmpdoc('tmpdoc')
+      @sa.stash_result
+      @s.get_data('tmpdoc').should == expected
+      Source.valid_doctypes.delete('tmpdoc'.to_sym)
+    end
 
-        it "should execute Model metadata method" do
-          mock_metadata_method([SimpleAdapter]) do
-            @sa.metadata.should == "{\"foo\":\"bar\"}"
-          end
+    describe "Model metadata method" do
+      it "should execute Model metadata method" do
+        mock_metadata_method([SimpleAdapter]) do
+          @sa.metadata.should == "{\"foo\":\"bar\"}"
         end
       end
     end

data/spec/source_spec.rb

@@ -1,157 +1,156 @@
 require File.join(File.dirname(__FILE__),'spec_helper')
 
 describe "Source" do
-  it_behaves_like "SharedRhoconnectHelper", :rhoconnect_data => false do
-  
-    before(:each) do
-      @s = Source.load(@s_fields[:name],@s_params)
-    end
-  
-    it "should create and load source with @s_fields and @s_params" do
-      @s.name.should == @s_fields[:name]
-      @s.url.should == @s_fields[:url]
-      @s.login.should == @s_fields[:login]
-      @s.app.name.should == @a_fields[:name]
-      @s.priority.should == 1
-      @s.callback_url.should be_nil
-      @s.queue.should be_nil
-      @s.query_queue.should be_nil
-      @s.cud_queue.should be_nil
-      @s.app_id.should == @s_params[:app_id]
-      @s.user_id.should == @s_params[:user_id]
-      @s.sync_type.should == :incremental
-      @s.partition_type.should == :user
-      @s.poll_interval.should == 300
-  
-      @s1 = Source.load(@s.id,@s_params)
-      @s1.name.should == @s_fields[:name]
-      @s1.url.should == @s_fields[:url]
-      @s1.login.should == @s_fields[:login]
-      @s1.app.name.should == @a_fields[:name]
-      @s1.priority.should == 1
-      @s1.callback_url.should be_nil
-      @s1.poll_interval.should == 300
-      @s1.app_id.should == @s_params[:app_id]
-      @s1.user_id.should == @s_params[:user_id]
-    end
-  
-    it "should create source with user" do
-      @s.user.login.should == @u_fields[:login]
-    end
-  
-    it "should create source with app and document" do
-      @s.app.name.should == @a_fields[:name]
-      @s.docname(:md).should == "source:#{@s.app.id}:#{@u.id}:#{@s_fields[:name]}:md"
-    end
-  
-    it 'should return values that set by setter method' do
-      @s.login = "shurab"
-      @s.login.should == "shurab"
-      @s.poll_interval = 350
-      @s.poll_interval.should == 350
-      @s.poll_interval = nil
-      @s.poll_interval.should == nil
-      @s.url = nil
-      @s.url.should be_nil  
-    end
-    
-    it "should delete source" do
-      @s.delete
-      Source.is_exist?(@s_fields[:name]).should == false
-      @s = Source.create(@s_fields,@s_params)
-    end
-  
-    it "should delete master and all documents associated with source" do
-      key = @s.docname('*')
-      Source.define_valid_doctypes([:md_source])
-      set_doc_state(@s, {:md_source => @data})
-      @s.delete
-      i = 0
-      while i < Store.num_stores do
-        Store.get_store(i).keys(key).should == []
-        i += 1
-      end
-      Source.valid_doctypes.delete(:md_source)
-      @s = Source.create(@s_fields,@s_params)
-    end
-  
-    it "should create source with default partition user" do
-      @s1 = Source.load(@s_fields[:name],{:app_id => @a.id,:user_id => '*'})
-      @s1.partition.should == :user
-    end
-  
-    it "should create correct docname based on partition scheme" do
-      @s.partition = :app
-      @s.docname(:md).should == "source:#{@s.app.id}:__shared__:#{@s_fields[:name]}:md"
-    end
-  
-    it "should create source with default read/write queue" do
-      @s.delete
-      @s_fields[:queue] = :default
-      @s_fields[:query_queue] = :query
-      @s_fields[:cud_queue] = :cud
-      @s = Source.create(@s_fields,@s_params)
-      s = Source.load(@s_fields[:name],@s_params)
-      s.queue.should == :default
-      s.query_queue.should == :query
-      s.cud_queue.should == :cud
-    end
-  
-    it "should add associations based on belongs_to field for a source" do
-      @s2 = Source.create({:name => 'SimpleAdapter'}, @s_params)
-      @s2.belongs_to = [{'product_id' => 'SampleAdapter'}].to_json
-      sf = Source.load(@s1.name,{:app_id => @a.id,:user_id => '*'})
-      Source.update_associations([@s.name,@s1.name, @s2.name])
-      s = Source.load(@s.name,{:app_id => @a.id,:user_id => '*'})
-      s.has_many.should == "#{@s1.name},brand,#{@s2.name},product_id"
-    end
-  
-    it "should log warning about incorrect belongs_to format for a source" do
-      @s2 = Source.create({:name => 'SimpleAdapter'}, @s_params)
-      @s2.belongs_to = {'product_id' => 'SampleAdapter'}.to_json
-      Source.should_receive(:log).once.with(
-        "WARNING: Incorrect belongs_to format for SimpleAdapter, belongs_to should be an array."
-      )
-      Source.update_associations([@s.name,@s1.name, @s2.name])
-    end
+  include_examples "SharedRhoconnectHelper", :rhoconnect_data => false
 
-    # check custom partitions
-    it "should return custom partition for users if defined in the model" do
-      u1_fields = {:login => 'cus_user1'}
-      u1 = User.create(u1_fields)
-      u2_fields = {:login => 'cus_user2'}
-      u2 = User.create(u2_fields)
-      u3_fields = {:login => 'other_user'}
-      u3 = User.create(u3_fields)
-
-      s = Source.load('SampleAdapter', {:user_id => u1.login, :app_id => APP_NAME})
-      s.user_by_partition.should == 'custom_partition'
-      s.docname(:md).should == "source:#{s.app.id}:custom_partition:SampleAdapter:md"
-      s = Source.load('FixedSchemaAdapter', {:user_id => u1.login, :app_id => APP_NAME})
-      s.user_by_partition.should == 'cus_user1'
-      s = Source.load('SampleAdapter', {:user_id => u2.login, :app_id => APP_NAME})
-      s.user_by_partition.should == 'custom_partition'
-      s = Source.load('SampleAdapter', {:user_id => u3.login, :app_id => APP_NAME})
-      s.user_by_partition.should == 'other_user'
-    end
+  before(:each) do
+    @s = Source.load(@s_fields[:name],@s_params)
+  end
+
+  it "should create and load source with @s_fields and @s_params" do
+    @s.name.should == @s_fields[:name]
+    @s.url.should == @s_fields[:url]
+    @s.login.should == @s_fields[:login]
+    @s.app.name.should == @a_fields[:name]
+    @s.priority.should == 1
+    @s.callback_url.should be_nil
+    @s.queue.should be_nil
+    @s.query_queue.should be_nil
+    @s.cud_queue.should be_nil
+    @s.app_id.should == @s_params[:app_id]
+    @s.user_id.should == @s_params[:user_id]
+    @s.sync_type.should == :incremental
+    @s.partition_type.should == :user
+    @s.poll_interval.should == 300
+
+    @s1 = Source.load(@s.id,@s_params)
+    @s1.name.should == @s_fields[:name]
+    @s1.url.should == @s_fields[:url]
+    @s1.login.should == @s_fields[:login]
+    @s1.app.name.should == @a_fields[:name]
+    @s1.priority.should == 1
+    @s1.callback_url.should be_nil
+    @s1.poll_interval.should == 300
+    @s1.app_id.should == @s_params[:app_id]
+    @s1.user_id.should == @s_params[:user_id]
+  end
+
+  it "should create source with user" do
+    @s.user.login.should == @u_fields[:login]
+  end
+
+  it "should create source with app and document" do
+    @s.app.name.should == @a_fields[:name]
+    @s.docname(:md).should == "source:#{@s.app.id}:#{@u.id}:#{@s_fields[:name]}:md"
+  end
 
-    it "should delete data from custom partition" do
-      u1_fields = {:login => 'cus_user1'}
-      u1 = User.create(u1_fields)
-      data = {'1' => {'Name' => 'Value'}}
-      
-      s = Source.load('SampleAdapter', {:user_id => u1.login, :app_id => APP_NAME})
-      s.user_by_partition.should == 'custom_partition'
-      Source.define_valid_doctypes([:md_source])
-      set_doc_state(s, {:md_source => data})
-      Store.get_data("source:#{s.app.id}:custom_partition:SampleAdapter:md_source").should == data
-      Store.get_data("source:#{s.app.id}:cus_user1:SampleAdapter:md_source").should == {}
-      Store.put_data("source:#{s.app.id}:cus_user1:SampleAdapter:md_source", data)
-
-      s.delete
-      Store.get_data("source:#{s.app.id}:custom_partition:SampleAdapter:md_source").should == {}
-      Store.get_data("source:#{s.app.id}:cus_user1:SampleAdapter:md_source").should == data
-      Source.valid_doctypes.delete(:md_source)
+  it 'should return values that set by setter method' do
+    @s.login = "shurab"
+    @s.login.should == "shurab"
+    @s.poll_interval = 350
+    @s.poll_interval.should == 350
+    @s.poll_interval = nil
+    @s.poll_interval.should == nil
+    @s.url = nil
+    @s.url.should be_nil
+  end
+
+  it "should delete source" do
+    @s.delete
+    Source.is_exist?(@s_fields[:name]).should == false
+    @s = Source.create(@s_fields,@s_params)
+  end
+
+  it "should delete master and all documents associated with source" do
+    key = @s.docname('*')
+    Source.define_valid_doctypes([:md_source])
+    set_doc_state(@s, {:md_source => @data})
+    @s.delete
+    i = 0
+    while i < Store.num_stores do
+      Store.get_store(i).keys(key).should == []
+      i += 1
     end
+    Source.valid_doctypes.delete(:md_source)
+    @s = Source.create(@s_fields,@s_params)
+  end
+
+  it "should create source with default partition user" do
+    @s1 = Source.load(@s_fields[:name],{:app_id => @a.id,:user_id => '*'})
+    @s1.partition.should == :user
+  end
+
+  it "should create correct docname based on partition scheme" do
+    @s.partition = :app
+    @s.docname(:md).should == "source:#{@s.app.id}:__shared__:#{@s_fields[:name]}:md"
+  end
+
+  it "should create source with default read/write queue" do
+    @s.delete
+    @s_fields[:queue] = :default
+    @s_fields[:query_queue] = :query
+    @s_fields[:cud_queue] = :cud
+    @s = Source.create(@s_fields,@s_params)
+    s = Source.load(@s_fields[:name],@s_params)
+    s.queue.should == :default
+    s.query_queue.should == :query
+    s.cud_queue.should == :cud
+  end
+
+  it "should add associations based on belongs_to field for a source" do
+    @s2 = Source.create({:name => 'SimpleAdapter'}, @s_params)
+    @s2.belongs_to = [{'product_id' => 'SampleAdapter'}].to_json
+    sf = Source.load(@s1.name,{:app_id => @a.id,:user_id => '*'})
+    Source.update_associations([@s.name,@s1.name, @s2.name])
+    s = Source.load(@s.name,{:app_id => @a.id,:user_id => '*'})
+    s.has_many.should == "#{@s1.name},brand,#{@s2.name},product_id"
+  end
+
+  it "should log warning about incorrect belongs_to format for a source" do
+    @s2 = Source.create({:name => 'SimpleAdapter'}, @s_params)
+    @s2.belongs_to = {'product_id' => 'SampleAdapter'}.to_json
+    Source.should_receive(:log).once.with(
+      "WARNING: Incorrect belongs_to format for SimpleAdapter, belongs_to should be an array."
+    )
+    Source.update_associations([@s.name,@s1.name, @s2.name])
+  end
+
+  # check custom partitions
+  it "should return custom partition for users if defined in the model" do
+    u1_fields = {:login => 'cus_user1'}
+    u1 = User.create(u1_fields)
+    u2_fields = {:login => 'cus_user2'}
+    u2 = User.create(u2_fields)
+    u3_fields = {:login => 'other_user'}
+    u3 = User.create(u3_fields)
+
+    s = Source.load('SampleAdapter', {:user_id => u1.login, :app_id => APP_NAME})
+    s.user_by_partition.should == 'custom_partition'
+    s.docname(:md).should == "source:#{s.app.id}:custom_partition:SampleAdapter:md"
+    s = Source.load('FixedSchemaAdapter', {:user_id => u1.login, :app_id => APP_NAME})
+    s.user_by_partition.should == 'cus_user1'
+    s = Source.load('SampleAdapter', {:user_id => u2.login, :app_id => APP_NAME})
+    s.user_by_partition.should == 'custom_partition'
+    s = Source.load('SampleAdapter', {:user_id => u3.login, :app_id => APP_NAME})
+    s.user_by_partition.should == 'other_user'
+  end
+
+  it "should delete data from custom partition" do
+    u1_fields = {:login => 'cus_user1'}
+    u1 = User.create(u1_fields)
+    data = {'1' => {'Name' => 'Value'}}
+
+    s = Source.load('SampleAdapter', {:user_id => u1.login, :app_id => APP_NAME})
+    s.user_by_partition.should == 'custom_partition'
+    Source.define_valid_doctypes([:md_source])
+    set_doc_state(s, {:md_source => data})
+    Store.get_data("source:#{s.app.id}:custom_partition:SampleAdapter:md_source").should == data
+    Store.get_data("source:#{s.app.id}:cus_user1:SampleAdapter:md_source").should == {}
+    Store.put_data("source:#{s.app.id}:cus_user1:SampleAdapter:md_source", data)
+
+    s.delete
+    Store.get_data("source:#{s.app.id}:custom_partition:SampleAdapter:md_source").should == {}
+    Store.get_data("source:#{s.app.id}:cus_user1:SampleAdapter:md_source").should == data
+    Source.valid_doctypes.delete(:md_source)
   end
 end