test_joint.rb in joint-0.6.1

- old
+ new

@@ -54,21 +54,25 @@
   end
 
   context "Using Joint plugin" do
     should "add each attachment to attachment_names" do
       Asset.attachment_names.should == Set.new([:image, :file])
+      EmbeddedAsset.attachment_names.should == Set.new([:image, :file])
     end
 
     should "add keys for each attachment" do
       key_names.each do |key|
         Asset.keys.should include("image_#{key}")
         Asset.keys.should include("file_#{key}")
+        EmbeddedAsset.keys.should include("image_#{key}")
+        EmbeddedAsset.keys.should include("file_#{key}")
       end
     end
 
     should "add memoized accessors module" do
       Asset.attachment_accessor_module.should be_instance_of(Module)
+      EmbeddedAsset.attachment_accessor_module.should be_instance_of(Module)
     end
 
     context "with inheritance" do
       should "add attachment to attachment_names" do
         BaseModel.attachment_names.should == Set.new([:file])
@@ -151,15 +155,85 @@
     should "know that attachment exists" do
       subject.image?.should be(true)
       subject.file?.should be(true)
     end
 
-    should "respond with false when asked if the attachment is nil?" do
-      subject.image.nil?.should be(false)
-      subject.file.nil?.should be(false)
+    should "respond with false when asked if the attachment is blank?" do
+      subject.image.blank?.should be(false)
+      subject.file.blank?.should be(false)
     end
 
+    should "clear assigned attachments so they don't get uploaded twice" do
+      Mongo::Grid.any_instance.expects(:put).never
+      subject.save
+    end
+  end
+
+  context "Assigning new attachments to embedded document" do
+    setup do
+      @asset = Asset.new
+      @doc = @asset.embedded_assets.build(:image => @image, :file => @file)
+      @asset.save!
+      rewind_files
+    end
+    subject { @doc }
+
+    should "assign GridFS content_type" do
+      grid.get(subject.image_id).content_type.should == 'image/jpeg'
+      grid.get(subject.file_id).content_type.should == 'application/pdf'
+    end
+
+    should "assign joint keys" do
+      subject.image_size.should  == 13661
+      subject.file_size.should   == 68926
+
+      subject.image_type.should  == "image/jpeg"
+      subject.file_type.should   == "application/pdf"
+
+      subject.image_id.should_not be_nil
+      subject.file_id.should_not be_nil
+
+      subject.image_id.should be_instance_of(BSON::ObjectId)
+      subject.file_id.should be_instance_of(BSON::ObjectId)
+    end
+
+    should "allow accessing keys through attachment proxy" do
+      subject.image.size.should  == 13661
+      subject.file.size.should   == 68926
+
+      subject.image.type.should  == "image/jpeg"
+      subject.file.type.should   == "application/pdf"
+
+      subject.image.id.should_not be_nil
+      subject.file.id.should_not be_nil
+
+      subject.image.id.should be_instance_of(BSON::ObjectId)
+      subject.file.id.should be_instance_of(BSON::ObjectId)
+    end
+
+    should "proxy unknown methods to GridIO object" do
+      subject.image.files_id.should      == subject.image_id
+      subject.image.content_type.should  == 'image/jpeg'
+      subject.image.filename.should      == 'mr_t.jpg'
+      subject.image.file_length.should   == 13661
+    end
+
+    should "assign file name from path if original file name not available" do
+      subject.image_name.should  == 'mr_t.jpg'
+      subject.file_name.should   == 'unixref.pdf'
+    end
+
+    should "save attachment contents correctly" do
+      subject.file.read.should   == @file.read
+      subject.image.read.should  == @image.read
+    end
+
+    should "know that attachment exists" do
+      subject.image?.should be(true)
+      subject.file?.should be(true)
+    end
+
     should "respond with false when asked if the attachment is blank?" do
       subject.image.blank?.should be(false)
       subject.file.blank?.should be(false)
     end
 
@@ -196,10 +270,37 @@
       grid.get(subject.file_id).file_length.should  == 5
       grid.get(subject.file_id).read.should         == @test2.read
     end
   end
 
+  context "Updating existing attachment in embedded document" do
+    setup do
+      @asset = Asset.new
+      @doc = @asset.embedded_assets.build(:file => @test1)
+      @asset.save!
+      assert_no_grid_difference do
+        @doc.file = @test2
+        @doc.save!
+      end
+      rewind_files
+    end
+    subject { @doc }
+
+    should "update keys" do
+      subject.file_name.should == 'test2.txt'
+      subject.file_type.should == "text/plain"
+      subject.file_size.should == 5
+    end
+
+    should "update GridFS" do
+      grid.get(subject.file_id).filename.should     == 'test2.txt'
+      grid.get(subject.file_id).content_type.should == 'text/plain'
+      grid.get(subject.file_id).file_length.should  == 5
+      grid.get(subject.file_id).read.should         == @test2.read
+    end
+  end
+
   context "Updating document but not attachments" do
     setup do
       @doc = Asset.create(:image => @image)
       @doc.update_attributes(:title => 'Updated')
       @doc.reload
@@ -214,10 +315,30 @@
     should "update document attributes" do
       subject.title.should == 'Updated'
     end
   end
 
+  context "Updating embedded document but not attachments" do
+    setup do
+      @asset = Asset.new
+      @doc = @asset.embedded_assets.build(:image => @image)
+      @doc.update_attributes(:title => 'Updated')
+      @asset.reload
+      @doc = @asset.embedded_assets.first
+      rewind_files
+    end
+    subject { @doc }
+
+    should "not affect attachment" do
+      subject.image.read.should == @image.read
+    end
+
+    should "update document attributes" do
+      subject.title.should == 'Updated'
+    end
+  end
+
   context "Assigning file where file pointer is not at beginning" do
     setup do
       @image.read
       @doc = Asset.create(:image => @image)
       @doc.reload
@@ -276,11 +397,62 @@
       assert_nil subject.image_id
       assert_nil subject.image_name
       assert_nil subject.image_type
       assert_nil subject.image_size
     end
+  end
 
+  context "Setting attachment to nil on embedded document" do
+    setup do
+      @asset = Asset.new
+      @doc = @asset.embedded_assets.build(:image => @image)
+      @asset.save!
+      rewind_files
+    end
+    subject { @doc }
+
+    should "delete attachment after save" do
+      assert_no_grid_difference   { subject.image = nil }
+      assert_grid_difference(-1)  { subject.save }
+    end
+
+    should "know that the attachment has been nullified" do
+      subject.image = nil
+      subject.image?.should be(false)
+    end
+
+    should "respond with true when asked if the attachment is nil?" do
+      subject.image = nil
+      subject.image.nil?.should be(true)
+    end
+
+    should "respond with true when asked if the attachment is blank?" do
+      subject.image = nil
+      subject.image.blank?.should be(true)
+    end
+
+    should "clear nil attachments after save and not attempt to delete again" do
+      Mongo::Grid.any_instance.expects(:delete).once
+      subject.image = nil
+      subject.save
+      Mongo::Grid.any_instance.expects(:delete).never
+      subject.save
+    end
+
+    should "clear id, name, type, size" do
+      subject.image = nil
+      subject.save
+      assert_nil subject.image_id
+      assert_nil subject.image_name
+      assert_nil subject.image_type
+      assert_nil subject.image_size
+      s = subject._root_document.reload.embedded_assets.first
+      assert_nil s.image_id
+      assert_nil s.image_name
+      assert_nil s.image_type
+      assert_nil s.image_size
+    end
   end
 
   context "Retrieving attachment that does not exist" do
     setup do
       @doc = Asset.create
@@ -311,10 +483,26 @@
     should "remove files from grid fs as well" do
       assert_grid_difference(-1) { subject.destroy }
     end
   end
 
+  context "Destroying an embedded document's _root_document" do
+    setup do
+      @asset = Asset.new
+      @doc = @asset.embedded_assets.build(:image => @image)
+      @doc.save!
+      rewind_files
+    end
+    subject { @doc }
+
+    should "remove files from grid fs as well" do
+      assert_grid_difference(-1) { subject._root_document.destroy }
+    end
+  end
+
+  # What about when an embedded document is removed?
+
   context "Assigning file name" do
     should "default to path" do
       Asset.create(:image => @image).image.name.should == 'mr_t.jpg'
     end
 
@@ -356,18 +544,17 @@
   context "Assigning joint io instance" do
     setup do
       io = Joint::IO.new({
         :name    => 'foo.txt',
         :type    => 'plain/text',
-        :content => 'This is my stuff',
-        :size    => 19,
+        :content => 'This is my stuff'
       })
       @asset = Asset.create(:file => io)
     end
 
     should "work" do
       @asset.file_name.should == 'foo.txt'
-      @asset.file_size.should == 19
+      @asset.file_size.should == 16
       @asset.file_type.should == 'plain/text'
       @asset.file.read.should == 'This is my stuff'
     end
   end
 
\ No newline at end of file