reader_spec.rb in heirloom-0.11.0.beta.1

- old
+ new

@@ -2,21 +2,21 @@
 
 describe Heirloom do
 
   before do
     @sdb_mock = mock 'sdb'
-    @config_mock = double 'config'
+    @config_mock = mock 'config'
     @logger_stub = stub :debug => true
-    @config_mock.should_receive(:logger).and_return @logger_stub
-    Heirloom::AWS::SimpleDB.should_receive(:new).and_return @sdb_mock
+    @config_mock.stub :logger => @logger_stub
     @reader = Heirloom::Reader.new :config => @config_mock,
                                    :name   => 'tim',
                                    :id     => '123'
   end
 
   context "domain does exist" do
     before do
+      Heirloom::AWS::SimpleDB.stub :new => @sdb_mock
       @sdb_mock.stub :domain_exists? => true
     end
 
     it "should show the item record" do
       @sdb_mock.should_receive(:select).
@@ -46,43 +46,41 @@
       @reader.exists?.should == false
     end
 
     it "should return the bucket if it exists" do
       @sdb_mock.should_receive(:select).
-                exactly(3).times.
+                at_least(:once).
                 with("select * from `heirloom_tim` where itemName() = '123'").
-                and_return( { '123' => 
-                              { 'us-west-1-s3-url' => 
-                                [ 's3://the-bucket/the-name/123.tar.gz' ]  
+                and_return( { '123' =>
+                              { 'us-west-1-s3-url' =>
+                                [ 's3://the-bucket/the-name/123.tar.gz' ]
                               }
                             } )
       @reader.get_bucket(:region => 'us-west-1').should == 'the-bucket'
     end
 
     it "should return nil if the key does not exist" do
       @sdb_mock.should_receive(:select).
-                exactly(1).times.
                 with("select * from `heirloom_tim` where itemName() = '123'").
                 and_return( { } )
       @reader.get_key(:region => 'us-west-1').should == nil
     end
 
     it "should return nil if the bucket does not exist" do
       @sdb_mock.should_receive(:select).
-                exactly(1).times.
                 with("select * from `heirloom_tim` where itemName() = '123'").
                 and_return( { } )
       @reader.get_bucket(:region => 'us-west-1').should == nil
     end
 
     it "should return the key if it exists" do
       @sdb_mock.should_receive(:select).
-                exactly(6).times.
+                at_least(:once).
                 with("select * from `heirloom_tim` where itemName() = '123'").
-                and_return( { '123' => 
-                              { 'us-west-1-s3-url' => 
-                                ['s3://the-url/the-bucket/123.tar.gz'] 
+                and_return( { '123' =>
+                              { 'us-west-1-s3-url' =>
+                                ['s3://the-url/the-bucket/123.tar.gz']
                               }
                             } )
       @reader.get_key(:region => 'us-west-1').should == 'the-bucket/123.tar.gz'
     end
 
@@ -100,32 +98,61 @@
       @reader.key_name.should == '123.tar.gz'
     end
 
     it "should return the regions the archive has been uploaded to" do
       @sdb_mock.should_receive(:select).
-                exactly(1).times.
                 with("select * from `heirloom_tim` where itemName() = '123'").
-                and_return( { '123' => 
-                              { 'us-west-1-s3-url' => 
+                and_return( { '123' =>
+                              { 'us-west-1-s3-url' =>
                                 ['s3://the-url-us-west-1/the-bucket/123.tar.gz'],
-                                'build_by' => 
-                                ['user'], 
-                                'us-east-1-s3-url' => 
-                                ['s3://the-url-us-east-1/the-bucket/123.tar.gz'] 
+                                'build_by' =>
+                                ['user'],
+                                'us-east-1-s3-url' =>
+                                ['s3://the-url-us-east-1/the-bucket/123.tar.gz']
                               }
                             } )
       @reader.regions.should == ['us-west-1', 'us-east-1']
     end
 
   end
 
   context "domain does not exist" do
     before do
+      Heirloom::AWS::SimpleDB.stub :new => @sdb_mock
       @sdb_mock.stub :domain_exists? => false
     end
 
     it "should return false if the simpledb domain does not exist" do
       @reader.exists?.should == false
+    end
+  end
+
+  context "object_acl verify" do
+    it "should get object_acls" do
+      regions = ['us-west-1', 'us-west-2']
+      @config_mock.stub :access_key => 'the-key',
+                        :secret_key => 'the-secret'
+      @reader.stub :regions    => regions,
+                   :key_name   => 'mockvalue',
+                   :get_bucket => 'mockvalue'
+      data = { "Owner" => { "ID" => "123", "DisplayName" => "lc" },
+               "AccessControlList" => [
+                { "Grantee" => { "ID" => "321", "DisplayName" => "rickybobby" },
+                                 "Permission" => "READ" },
+                { "Grantee" => { "ID" => "123", "DisplayName" => "lc" },
+                                 "Permission" => "FULL_CONTROL" }]
+             }
+      s3_stub = stub 's3', :get_object_acl => data
+
+      regions.each do |region|
+        Heirloom::AWS::S3.should_receive(:new).
+                          with(:config => @config_mock,
+                               :region => region).
+                          and_return s3_stub
+      end
+
+      @reader.object_acls.should == { 'us-west-1-permissions' => 'rickybobby:read, lc:full_control',
+                                      'us-west-2-permissions' => 'rickybobby:read, lc:full_control' }
     end
   end
 
 end