spec/functional/shardable_spec.rb in mongo_mapper-0.15.5 vs spec/functional/shardable_spec.rb in mongo_mapper-0.15.6

@@ -1,66 +1,58 @@
 require 'spec_helper'
 
 describe 'Shardable' do
-  let(:sharded_model) {
-    Doc do
-      key :first_name, String
-      key :last_name, String
-      shard_key :first_name, :last_name
-    end
-  }
-
   describe 'shard_key_fields' do
     it 'returns declared field names' do
-      sharded_model.shard_key_fields.should == ['first_name', 'last_name']
+      ShardedModel.shard_key_fields.should == ['first_name']
     end
   end
 
   describe 'shard_key_filter' do
     context 'new record' do
-      let(:document) { sharded_model.new(first_name: 'John', last_name: 'Smith') }
+      let(:document) { ShardedModel.new(first_name: 'John', last_name: 'Smith') }
 
       it 'returns current values' do
-        document.shard_key_filter.should == { 'first_name' => 'John', 'last_name' => 'Smith' }
+        document.shard_key_filter.should == { 'first_name' => 'John' }
       end
     end
 
     context 'persisted record' do
-      let(:document) { sharded_model.create!(first_name: 'John', last_name: 'Smith') }
+      let(:document) { ShardedModel.create!(first_name: 'John', last_name: 'Smith') }
 
       before do
         document.first_name = 'William'
       end
 
       it 'returns persisted values' do
-        document.shard_key_filter.should == { 'first_name' => 'John', 'last_name' => 'Smith' }
+        document.shard_key_filter.should == { 'first_name' => 'John' }
       end
     end
   end
 
   context 'creating new document' do
-    let(:document) { sharded_model.new(first_name: 'John', last_name: 'Smith') }
+    let(:document) { ShardedModel.new(first_name: 'John', last_name: 'Smith') }
 
     it 'inserts new document' do
-      lambda { document.save! }.should change { sharded_model.count }.by(1)
+      lambda { document.save! }.should change { ShardedModel.count }.by(1)
 
-      persisted = sharded_model.find(document.id)
+      persisted = ShardedModel.find(document.id)
       persisted.first_name.should == 'John'
       persisted.last_name.should == 'Smith'
     end
   end
 
   context 'updating persisted document' do
-    let(:document) { sharded_model.create!(first_name: 'John', last_name: 'Smith') }
+    let(:document) { ShardedModel.create!(first_name: 'John', last_name: 'Smith') }
 
     before do
       document.first_name = 'William'
     end
 
     it 'updates persisted document' do
-      lambda { document.save! }.should_not change { sharded_model.count }
+      lambda { document.save! }.should_not change { ShardedModel.count }
 
-      persisted = sharded_model.find(document.id)
+      persisted = ShardedModel.find(document.id)
       persisted.first_name.should == 'William'
       persisted.last_name.should == 'Smith'
     end
   end
 end