lib/nsisam/fake_client.rb in nsisam-0.6.4 vs lib/nsisam/fake_client.rb in nsisam-0.7.0

@@ -1,27 +1,38 @@
 require 'json'
 require 'base64'
 require File.dirname(__FILE__) + '/response'
+require 'webmock'
+include WebMock::API
+WebMock.allow_net_connect!
 
+
 module NSISam
   class FakeClient
 
     attr_accessor :expire
+    attr_reader :host, :port
 
-    def initialize
+    def initialize(host="localhost", port="8888")
       @storage = {}
+      @host = host
+      @port = port
     end
 
     def store(data)
-      key = Time.now.to_i.to_s
+      key = Time.now.nsec.to_s
       @storage[key] = JSON.load(data.to_json) unless @expire
+      if data.kind_of?(Hash) and data.has_key?(:file) and data.has_key?(:filename)
+        stub_request(:get, "http://#{@host}:#{@port}/file/#{key}").to_return(body: Base64.decode64(data[:file]))
+      end
       Response.new 'key' => key, 'checksum' => 0
     end
 
-    def store_file(file, type=:file)
+    def store_file(file, filename, type=:file)
       key = Time.now.to_i.to_s
-      @storage[key] = {type.to_s => Base64.encode64(file)}.to_json unless @expire
+      @storage[key] = {type.to_s => Base64.encode64(file), filename: filename}.to_json unless @expire
+      stub_request(:get, "http://#{@host}:#{@port}/file/#{key}").to_return(body: file)
       Response.new "key" => key, "checksum" => 0
     end
 
     def get(key, expected_checksum=nil)
       if @storage.has_key?(key)
@@ -52,14 +63,23 @@
     def update(key, value)
       if @storage.has_key?(key)
         if @expire
           @storage.delete(key)
         else
-          @storage[key] = value 
+          @storage[key] = value
         end
-        Response.new 'key' => key, 'checksum' => 0 
+        Response.new 'key' => key, 'checksum' => 0
       else
         raise NSISam::Errors::Client::KeyNotFoundError
       end
     end
+
+    def update_file(key, file, filename)
+      hash = {file: file, filename: filename}
+      @storage[key] = hash
+      remove_request_stub(:get, "http://#{@host}:#{@port}/file/#{key}")
+      stub_request(:get, "http://#{@host}:#{@port}/file/#{key}").to_return(body: file)
+      Response.new "key" => key, "checksum" => 0
+    end
+
   end
 end