Merge pull request #1 from Root-App/zack-update-fake-methods

Added a way to fake uploading and downloading files

Author: martinza

lib/amazon/fake_communicator.rb

@@ -1,12 +1,18 @@
 module Amazon
   class FakeCommunicator
+    def self.data
+      @data ||= {}
+    end
     def initialize(access_key_id, secret_access_key, region)
       @access_key_id = access_key_id
       @secret_access_key = secret_access_key
       @region = region
+      @bucket_files
     end
 
     def find_bucket(bucket_name)
+      FakeCommunicator.data[bucket_name] ||= {}
+      @bucket_files = FakeCommunicator.data[bucket_name]
       _fake_bucket(bucket_name)
     end
 
@@ -77,6 +83,7 @@ def _fake_bucket(bucket_name)
       bucket = Object.new
       bucket.instance_variable_set(:@bucket_name, bucket_name)
       bucket.instance_variable_set(:@region, @region)
+      bucket.instance_variable_set(:@bucket_files, @bucket_files)
 
       def bucket.get_folders(folder: "")
         case @bucket_name
@@ -112,11 +119,14 @@ def bucket.find_files(file)
       end
 
       def bucket.download_file(object_name, local_file_name)
-        local_file_name
+        Tempfile.open(local_file_name) do |file|
+          file.write(@bucket_files[object_name])
+          file
+        end
       end
 
       def bucket.upload_file(object_name, local_file_name, content_type: nil)
-        true
+        @bucket_files[object_name] = File.read(local_file_name)
       end
 
       def bucket.region

spec/support/shared_examples.rb

@@ -22,6 +22,18 @@
       expect(link).to be_present
       expect(link).to start_with("https://")
     end
+
+    it "should upload and download a file to s3" do
+      file = Tempfile.open("out.txt") do |f|
+        f.write("hello world")
+        f
+      end
+      bucket.upload_file("sample.file", file.path)
+
+      downloaded_file = bucket.download_file("sample.file", "out.txt")
+      expect(downloaded_file.class).to be(Tempfile)
+      expect(downloaded_file.path).to include("out.txt")
+    end
   end
 
   context "cannot find valid bucket" do