capybara-screenshot 1.0.15 → 1.0.16

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 17a6c4cc3fd164bb2b6f74448ef7b9ddf83b771c
-  data.tar.gz: fc743fe14f52cb0481a3aa545e9b5290420a4cd1
+  metadata.gz: 9afdd9d33462cf19d2cd005844e6790171b23073
+  data.tar.gz: ac0e7b147e924204f72657cbdfd92b96e2b912cf
 SHA512:
-  metadata.gz: 366222eb34fb0bdfa5bf53c5b27a7e00902c1e8300b6767a377ff6fa207eb1a5104d6cc868d7f69c1829f370ff90ebbde1940e8529b8ab694fb544b9ab5fc38d
-  data.tar.gz: 0c04abede37b6e6fc39a1f1d22a2d7429eada67a0d46bbb8ef59bf1ba53d35763d1d4b4c7361d8c21ffa52dc90e58e7de9421a2864effa8e71dae5c939e9b944
+  metadata.gz: 8966719fe9c36701a98e20c72f92d00be96df39374e871ceb18de0aaf91284c40a9c1bedf714186e0c6967da75df8e477f247bfd0201229c1fd32ce4f57489c4
+  data.tar.gz: 936a36d7efd3fb8e50a203e3130e0575eb0ce5b99c24ae85e56eede5f19faf5b32cb01a42aa997151ebf00971fe85bed1b78734fed5d27116748e1ca9ab7dc10

data/.travis.yml

@@ -1,10 +1,11 @@
 language: ruby
-script: "rake travis:ci"
+script: "bundle exec rake travis:ci"
 rvm:
   - 2.0.0
   - 2.1.10
-  - 2.2.1
-  - 2.3.1
+  - 2.2.6
+  - 2.3.3
+  - 2.4.0
 gemfile:
   - gemfiles/cucumber.1.3.gemfile
   - gemfiles/cucumber.2.4.gemfile

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+12 July 2017 - 1.0.15 -> 1.0.16
+
+* [Support s3 key name prefixes](https://github.com/mattheworiordan/capybara-screenshot/pull/202)
+
 12 July 2017 - 1.0.14 -> 1.0.15
 
 * [SVG badges added](https://github.com/mattheworiordan/capybara-screenshot/pull/207)

data/README.md

@@ -223,6 +223,14 @@ Capybara::Screenshot.s3_object_configuration = {
 }
 ```
 
+You may optionally specify a `:key_prefix` when generating the S3 keys, which can be used to create virtual [folders](http://docs.aws.amazon.com/AmazonS3/latest/UG/FolderOperations.html) in S3, e.g.:
+
+```ruby
+Capybara::Screenshot.s3_configuration = {
+  ... # other config here
+  key_prefix: "some/folder/"
+}
+```
 
 Pruning old screenshots automatically
 --------------------------

data/lib/capybara-screenshot/s3_saver.rb

@@ -5,10 +5,11 @@ module Capybara
     class S3Saver
       DEFAULT_REGION = 'us-east-1'
 
-      def initialize(saver, s3_client, bucket_name, object_configuration)
+      def initialize(saver, s3_client, bucket_name, object_configuration, options={})
         @saver = saver
         @s3_client = s3_client
         @bucket_name = bucket_name
+        @key_prefix = options[:key_prefix]
         @object_configuration = object_configuration
       end
 
@@ -24,7 +25,7 @@ module Capybara
         s3_client = Aws::S3::Client.new(s3_client_credentials)
         bucket_name = configuration.fetch(:bucket_name)
 
-        new(saver, s3_client, bucket_name, object_configuration)
+        new(saver, s3_client, bucket_name, object_configuration, configuration)
       rescue KeyError
         raise "Invalid S3 Configuration #{configuration}. Please refer to the documentation for the necessary configurations."
       end
@@ -34,7 +35,7 @@ module Capybara
           File.open(local_file_path) do |file|
             object_payload = {
               bucket: bucket_name,
-              key: File.basename(local_file_path),
+              key: "#{@key_prefix}#{File.basename(local_file_path)}",
               body: file
             }
 
@@ -60,6 +61,7 @@ module Capybara
                   :s3_client,
                   :bucket_name,
                   :object_configuration
+                  :key_prefix
 
       def save_and
         saver.save

data/lib/capybara-screenshot/version.rb

@@ -1,5 +1,5 @@
 module Capybara
   module Screenshot
-    VERSION = '1.0.15'
+    VERSION = '1.0.16'
   end
 end

data/spec/unit/s3_saver_spec.rb

@@ -6,8 +6,10 @@ describe Capybara::Screenshot::S3Saver do
   let(:bucket_name) { double('bucket_name') }
   let(:s3_object_configuration) { {} }
   let(:s3_client) { double('s3_client') }
+  let(:key_prefix){ "some/path/" }
 
-  let(:s3_saver) { Capybara::Screenshot::S3Saver.new(saver, s3_client, bucket_name, s3_object_configuration) }
+  let(:s3_saver) { described_class.new(saver, s3_client, bucket_name, s3_object_configuration) }
+  let(:s3_saver_with_key_prefix) { described_class.new(saver, s3_client, bucket_name, s3_object_configuration, key_prefix: key_prefix) }
 
   describe '.new_with_configuration' do
     let(:access_key_id) { double('access_key_id') }
@@ -24,26 +26,36 @@ describe Capybara::Screenshot::S3Saver do
       s3_client_credentials_using_defaults.merge(region: region)
     }
 
-    it 'destructures the configuration into its components' do
+    before do
       allow(Aws::S3::Client).to receive(:new).and_return(s3_client)
-      allow(Capybara::Screenshot::S3Saver).to receive(:new)
+      allow(described_class).to receive(:new)
+    end
 
-      Capybara::Screenshot::S3Saver.new_with_configuration(saver, {
+    it 'destructures the configuration into its components' do
+      described_class.new_with_configuration(saver, {
         s3_client_credentials: s3_client_credentials,
         bucket_name: bucket_name
       }, s3_object_configuration)
 
       expect(Aws::S3::Client).to have_received(:new).with(s3_client_credentials)
-      expect(Capybara::Screenshot::S3Saver).to have_received(:new).with(saver, s3_client, bucket_name, s3_object_configuration)
+      expect(described_class).to have_received(:new).with(saver, s3_client, bucket_name, s3_object_configuration, hash_including({}))
+    end
+
+    it 'passes key_prefix option if specified' do
+      described_class.new_with_configuration(saver, {
+        s3_client_credentials: s3_client_credentials,
+        bucket_name: bucket_name,
+        key_prefix: key_prefix,
+      }, s3_object_configuration)
+
+      expect(Aws::S3::Client).to have_received(:new).with(s3_client_credentials)
+      expect(described_class).to have_received(:new).with(saver, s3_client, bucket_name, s3_object_configuration, hash_including(key_prefix: key_prefix))
     end
 
     it 'defaults the region to us-east-1' do
       default_region = 'us-east-1'
 
-      allow(Aws::S3::Client).to receive(:new).and_return(s3_client)
-      allow(Capybara::Screenshot::S3Saver).to receive(:new)
-
-      Capybara::Screenshot::S3Saver.new_with_configuration(saver, {
+      described_class.new_with_configuration(saver, {
           s3_client_credentials: s3_client_credentials_using_defaults,
           bucket_name: bucket_name
       }, s3_object_configuration)
@@ -52,23 +64,31 @@ describe Capybara::Screenshot::S3Saver do
         s3_client_credentials.merge(region: default_region)
       )
 
-      expect(Capybara::Screenshot::S3Saver).to have_received(:new).with(saver, s3_client, bucket_name, s3_object_configuration)
+      expect(described_class).to have_received(:new).with(saver, s3_client, bucket_name, s3_object_configuration, hash_including({}))
     end
 
     it 'stores the object configuration when passed' do
       s3_object_configuration = { acl: 'public-read' }
       Capybara::Screenshot.s3_object_configuration = { acl: 'public-read' }
 
-      allow(Aws::S3::Client).to receive(:new).and_return(s3_client)
-      allow(Capybara::Screenshot::S3Saver).to receive(:new)
-
-      Capybara::Screenshot::S3Saver.new_with_configuration(saver, {
+      described_class.new_with_configuration(saver, {
         s3_client_credentials: s3_client_credentials,
         bucket_name: bucket_name
       }, s3_object_configuration)
 
       expect(Aws::S3::Client).to have_received(:new).with(s3_client_credentials)
-      expect(Capybara::Screenshot::S3Saver).to have_received(:new).with(saver, s3_client, bucket_name, s3_object_configuration)
+      expect(described_class).to have_received(:new).with(saver, s3_client, bucket_name, s3_object_configuration, hash_including({}))
+    end
+
+    it 'passes key_prefix option if specified' do
+      described_class.new_with_configuration(saver, {
+        s3_client_credentials: s3_client_credentials,
+        bucket_name: bucket_name,
+        key_prefix: key_prefix,
+      }, s3_object_configuration)
+
+      expect(Aws::S3::Client).to have_received(:new).with(s3_client_credentials)
+      expect(described_class).to have_received(:new).with(saver, s3_client, bucket_name, s3_object_configuration, hash_including(key_prefix: key_prefix))
     end
   end
 
@@ -121,9 +141,9 @@ describe Capybara::Screenshot::S3Saver do
       s3_saver.save
     end
 
-    describe 'save to s3 with object configuration' do
+    context 'with object configuration' do
       let(:s3_object_configuration) { { acl: 'public-read' } }
-      let(:s3_saver) { Capybara::Screenshot::S3Saver.new(saver, s3_client, bucket_name, s3_object_configuration) }
+      let(:s3_saver) { described_class.new(saver, s3_client, bucket_name, s3_object_configuration) }
 
       it 'uploads the html' do
         html_path = '/foo/bar.html'
@@ -163,6 +183,86 @@ describe Capybara::Screenshot::S3Saver do
         s3_saver.save
       end
     end
+
+    context 'with key_prefix specified' do
+      it 'uploads the html with key prefix' do
+        html_path = '/foo/bar.html'
+        expect(saver).to receive(:html_path).and_return(html_path)
+        expect(saver).to receive(:html_saved?).and_return(true)
+
+        html_file = double('html_file')
+
+        expect(File).to receive(:open).with(html_path).and_yield(html_file)
+
+        expect(s3_client).to receive(:put_object).with(
+          bucket: bucket_name,
+          key: 'some/path/bar.html',
+          body: html_file
+        )
+
+        s3_saver_with_key_prefix.save
+      end
+
+      it 'uploads the screenshot with key prefix' do
+        screenshot_path = '/baz/bim.jpg'
+        expect(saver).to receive(:screenshot_path).and_return(screenshot_path)
+        expect(saver).to receive(:screenshot_saved?).and_return(true)
+
+        screenshot_file = double('screenshot_file')
+
+        expect(File).to receive(:open).with(screenshot_path).and_yield(screenshot_file)
+
+        expect(s3_client).to receive(:put_object).with(
+          bucket: bucket_name,
+          key: 'some/path/bim.jpg',
+          body: screenshot_file
+        )
+
+        s3_saver_with_key_prefix.save
+      end
+
+      context 'with object configuration' do
+        let(:s3_object_configuration) { { acl: 'public-read' } }
+
+        it 'uploads the html' do
+          html_path = '/foo/bar.html'
+          expect(saver).to receive(:html_path).and_return(html_path)
+          expect(saver).to receive(:html_saved?).and_return(true)
+
+          html_file = double('html_file')
+
+          expect(File).to receive(:open).with(html_path).and_yield(html_file)
+
+          expect(s3_client).to receive(:put_object).with(
+            bucket: bucket_name,
+            key: 'some/path/bar.html',
+            body: html_file,
+            acl: 'public-read'
+          )
+
+          s3_saver_with_key_prefix.save
+        end
+
+        it 'uploads the screenshot' do
+          screenshot_path = '/baz/bim.jpg'
+          expect(saver).to receive(:screenshot_path).and_return(screenshot_path)
+          expect(saver).to receive(:screenshot_saved?).and_return(true)
+
+          screenshot_file = double('screenshot_file')
+
+          expect(File).to receive(:open).with(screenshot_path).and_yield(screenshot_file)
+
+          expect(s3_client).to receive(:put_object).with(
+            bucket: bucket_name,
+            key: 'some/path/bim.jpg',
+            body: screenshot_file,
+            acl: 'public-read'
+          )
+
+          s3_saver_with_key_prefix.save
+        end
+      end
+    end
   end
 
   # Needed because we cannot depend on Verifying Doubles

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: capybara-screenshot
 version: !ruby/object:Gem::Version
-  version: 1.0.15
+  version: 1.0.16
 platform: ruby
 authors:
 - Matthew O'Riordan