couchrest_model_thought 1.0.0.beta8

data/spec/couchrest/attachment_spec.rb

@@ -0,0 +1,148 @@
+require File.expand_path('../../spec_helper', __FILE__)
+
+describe "Model attachments" do
+  
+  describe "#has_attachment?" do
+    before(:each) do
+      reset_test_db!
+      @obj = Basic.new
+      @obj.save.should be_true
+      @file = File.open(FIXTURE_PATH + '/attachments/test.html')
+      @attachment_name = 'my_attachment'
+      @obj.create_attachment(:file => @file, :name => @attachment_name)
+    end
+  
+    it 'should return false if there is no attachment' do
+      @obj.has_attachment?('bogus').should be_false
+    end
+  
+    it 'should return true if there is an attachment' do
+      @obj.has_attachment?(@attachment_name).should be_true
+    end
+  
+    it 'should return true if an object with an attachment is reloaded' do
+      @obj.save.should be_true
+      reloaded_obj = Basic.get(@obj.id)
+      reloaded_obj.has_attachment?(@attachment_name).should be_true
+    end
+  
+    it 'should return false if an attachment has been removed' do
+      @obj.delete_attachment(@attachment_name)
+      @obj.has_attachment?(@attachment_name).should be_false
+    end
+  end
+
+  describe "creating an attachment" do
+    before(:each) do
+      @obj = Basic.new
+      @obj.save.should be_true
+      @file_ext = File.open(FIXTURE_PATH + '/attachments/test.html')
+      @file_no_ext = File.open(FIXTURE_PATH + '/attachments/README')
+      @attachment_name = 'my_attachment'
+      @content_type = 'media/mp3'
+    end
+  
+    it "should create an attachment from file with an extension" do
+      @obj.create_attachment(:file => @file_ext, :name => @attachment_name)
+      @obj.save.should be_true
+      reloaded_obj = Basic.get(@obj.id)
+      reloaded_obj['_attachments'][@attachment_name].should_not be_nil
+    end
+  
+    it "should create an attachment from file without an extension" do
+      @obj.create_attachment(:file => @file_no_ext, :name => @attachment_name)
+      @obj.save.should be_true
+      reloaded_obj = Basic.get(@obj.id)
+      reloaded_obj['_attachments'][@attachment_name].should_not be_nil
+    end
+  
+    it 'should raise ArgumentError if :file is missing' do
+      lambda{ @obj.create_attachment(:name => @attachment_name) }.should raise_error
+    end
+  
+    it 'should raise ArgumentError if :name is missing' do
+      lambda{ @obj.create_attachment(:file => @file_ext) }.should raise_error
+    end
+  
+    it 'should set the content-type if passed' do
+      @obj.create_attachment(:file => @file_ext, :name => @attachment_name, :content_type => @content_type)
+      @obj['_attachments'][@attachment_name]['content_type'].should == @content_type
+    end
+
+    it "should detect the content-type automatically" do
+      @obj.create_attachment(:file => File.open(FIXTURE_PATH + '/attachments/couchdb.png'), :name => "couchdb.png")
+      @obj['_attachments']['couchdb.png']['content_type'].should == "image/png" 
+    end
+
+    it "should use name to detect the content-type automatically if no file" do
+      file = File.open(FIXTURE_PATH + '/attachments/couchdb.png')
+      file.stub!(:path).and_return("badfilname")
+      @obj.create_attachment(:file => File.open(FIXTURE_PATH + '/attachments/couchdb.png'), :name => "couchdb.png")
+      @obj['_attachments']['couchdb.png']['content_type'].should == "image/png" 
+    end
+
+  end
+
+  describe 'reading, updating, and deleting an attachment' do
+    before(:each) do
+      @obj = Basic.new
+      @file = File.open(FIXTURE_PATH + '/attachments/test.html')
+      @attachment_name = 'my_attachment'
+      @obj.create_attachment(:file => @file, :name => @attachment_name)
+      @obj.save.should be_true
+      @file.rewind
+      @content_type = 'media/mp3'
+    end
+  
+    it 'should read an attachment that exists' do
+      @obj.read_attachment(@attachment_name).should == @file.read
+    end
+  
+    it 'should update an attachment that exists' do
+      file = File.open(FIXTURE_PATH + '/attachments/README')
+      @file.should_not == file
+      @obj.update_attachment(:file => file, :name => @attachment_name)
+      @obj.save
+      reloaded_obj = Basic.get(@obj.id)
+      file.rewind
+      reloaded_obj.read_attachment(@attachment_name).should_not == @file.read
+      reloaded_obj.read_attachment(@attachment_name).should == file.read
+    end
+  
+    it 'should set the content-type if passed' do
+      file = File.open(FIXTURE_PATH + '/attachments/README')
+      @file.should_not == file
+      @obj.update_attachment(:file => file, :name => @attachment_name, :content_type => @content_type)
+      @obj['_attachments'][@attachment_name]['content_type'].should == @content_type
+    end
+  
+    it 'should delete an attachment that exists' do
+      @obj.delete_attachment(@attachment_name)
+      @obj.save
+      lambda{Basic.get(@obj.id).read_attachment(@attachment_name)}.should raise_error
+    end
+  end
+
+  describe "#attachment_url" do
+    before(:each) do
+      @obj = Basic.new
+      @file = File.open(FIXTURE_PATH + '/attachments/test.html')
+      @attachment_name = 'my_attachment'
+      @obj.create_attachment(:file => @file, :name => @attachment_name)
+      @obj.save.should be_true
+    end
+  
+    it 'should return nil if attachment does not exist' do
+      @obj.attachment_url('bogus').should be_nil
+    end
+  
+    it 'should return the attachment URL as specified by CouchDB HttpDocumentApi' do
+      @obj.attachment_url(@attachment_name).should == "#{Basic.database}/#{@obj.id}/#{@attachment_name}"
+    end
+    
+    it 'should return the attachment URI' do
+      @obj.attachment_uri(@attachment_name).should == "#{Basic.database.uri}/#{@obj.id}/#{@attachment_name}"
+    end
+    
+  end
+end

data/spec/couchrest/attribute_protection_spec.rb

@@ -0,0 +1,153 @@
+require File.expand_path("../../spec_helper", __FILE__)
+
+describe "Model Attributes" do
+
+  describe "no declarations" do
+    class NoProtection < CouchRest::Model::Base
+      use_database TEST_SERVER.default_database
+      property :name
+      property :phone
+    end
+
+    it "should not protect anything through new" do
+      user = NoProtection.new(:name => "will", :phone => "555-5555")
+
+      user.name.should == "will"
+      user.phone.should == "555-5555"
+    end
+
+    it "should not protect anything through attributes=" do
+      user = NoProtection.new
+      user.attributes = {:name => "will", :phone => "555-5555"}
+
+      user.name.should == "will"
+      user.phone.should == "555-5555"
+    end
+    
+    it "should recreate from the database properly" do
+      user = NoProtection.new
+      user.name = "will"
+      user.phone = "555-5555"
+      user.save!
+      
+      user = NoProtection.get(user.id)
+      user.name.should == "will"
+      user.phone.should == "555-5555"
+    end
+  end
+
+  describe "Model Base", "accessible flag" do
+    class WithAccessible < CouchRest::Model::Base
+      use_database TEST_SERVER.default_database
+      property :name, :accessible => true
+      property :admin, :default => false
+    end
+
+    it "should recognize accessible properties" do
+      props = WithAccessible.accessible_properties.map { |prop| prop.name}
+      props.should include("name")
+      props.should_not include("admin")
+    end
+
+    it "should protect non-accessible properties set through new" do
+      user = WithAccessible.new(:name => "will", :admin => true) 
+
+      user.name.should == "will"
+      user.admin.should == false
+    end
+
+    it "should protect non-accessible properties set through attributes=" do
+      user = WithAccessible.new
+      user.attributes = {:name => "will", :admin => true}
+
+      user.name.should == "will"
+      user.admin.should == false
+    end
+  end
+
+  describe "Model Base", "protected flag" do
+    class WithProtected < CouchRest::Model::Base
+      use_database TEST_SERVER.default_database
+      property :name
+      property :admin, :default => false, :protected => true
+    end
+
+    it "should recognize protected properties" do
+      props = WithProtected.protected_properties.map { |prop| prop.name}
+      props.should_not include("name")
+      props.should include("admin")
+    end
+
+    it "should protect non-accessible properties set through new" do
+      user = WithProtected.new(:name => "will", :admin => true) 
+
+      user.name.should == "will"
+      user.admin.should == false
+    end
+
+    it "should protect non-accessible properties set through attributes=" do
+      user = WithProtected.new
+      user.attributes = {:name => "will", :admin => true}
+
+      user.name.should == "will"
+      user.admin.should == false
+    end
+  end
+
+  describe "protected flag" do
+    class WithBoth < CouchRest::Model::Base
+      use_database TEST_SERVER.default_database
+      property :name, :accessible => true
+      property :admin, :default => false, :protected => true
+    end
+
+    it "should raise an error when both are set" do
+      lambda { WithBoth.new }.should raise_error 
+    end
+  end
+
+  describe "from database" do
+    class WithProtected < CouchRest::Model::Base
+      use_database TEST_SERVER.default_database
+      property :name
+      property :admin, :default => false, :protected => true
+      view_by :name
+    end
+    
+    before(:each) do
+      @user = WithProtected.new
+      @user.name = "will"
+      @user.admin = true
+      @user.save!
+    end
+
+    def verify_attrs(user)
+      user.name.should  == "will"
+      user.admin.should == true
+    end
+
+    it "Base#get should not strip protected attributes" do
+      reloaded = WithProtected.get( @user.id )
+      verify_attrs reloaded 
+    end
+
+    it "Base#get! should not strip protected attributes" do
+      reloaded = WithProtected.get!( @user.id )
+      verify_attrs reloaded 
+    end
+
+    it "Base#all should not strip protected attributes" do
+      # all creates a CollectionProxy
+      docs = WithProtected.all(:key => @user.id)
+      docs.size.should == 1
+      reloaded = docs.first
+      verify_attrs reloaded 
+    end
+
+    it "views should not strip protected attributes" do
+      docs = WithProtected.by_name(:startkey => "will", :endkey => "will")
+      reloaded = docs.first
+      verify_attrs reloaded 
+    end
+  end
+end