refactor: pass request body io as parameter to put

1 files changed, 5 insertions, 6 deletions

diff --git a/server-rack.rb b/server-rack.rb
index 3d4642b..9c768e5 100644
--- a/server-rack.rb
+++ b/server-rack.rb
@@ -30,8 +30,8 @@ class DataStorageServer
     end
   end
 
-  def put(env)
-    data = env['rack.input'].read(MAX_BYTES)
+  def put(io)
+    data = io.read(MAX_BYTES)
     oid = Digest::SHA1.hexdigest(data)
     @storage[oid] = data
     ['201', {}, [JSON.generate({ size: data.size, oid: oid })]]
@@ -48,14 +48,13 @@ class DataStorageServer
   end
 
   def call(env)
-    path = env['PATH_INFO']
     case env['REQUEST_METHOD']
     when 'GET'
-      get(path)
+      get(env['PATH_INFO'])
     when 'PUT'
-      put(env)
+      put(env['rack.input'])
     when 'DELETE'
-      destroy(path)
+      destroy(env['PATH_INFO'])
     else
       raise env.inspect
     end