party_cc 0.0.7 → 0.0.8

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    OGEzM2Q2OTUyOTYzM2VkNjFjMzUyZGFiMGI2ZjI4Y2EwNmNmOGIxMA==
+    MjIwY2NjMDRmOGQxYzExYTZiZjk0YzUyNTVhZDFmNWRhOTliZmNhNA==
   data.tar.gz: !binary |-
-    YjhhNWIyOThiMDNiZWNkOTBkZDY2ZGY3OTVjM2UyMzBhM2E0NzhhMw==
+    NjE0N2RjY2MwYzIyNTViZTY5MjA1MWY2ODdmMjgxZmYwODM2M2VhZA==
 SHA512:
   metadata.gz: !binary |-
-    MzUwNjgwMmQ0YzM3MDgxM2E5ZjlmZDJjY2E5MjgwZDU5MWMzNDdmZTM0MTU5
-    ZDcyYWI0NDZhNTMxMjA3Mzk4ZDlkM2MzMGQ4YmIxMDhhYmM3ZmFiYjFhMDg0
-    MzQzNDE2NmM5NTkyZjM5Yjg5NjVhZmU5ZWJlMzdmZWZkMmEwNzQ=
+    NjBiMTY2ZjFjMGUyOWY2MzAwODhmNDNhNzJhNWJhNjQ0MDc2NjI2MTEyMTI5
+    NzZmYmFiOTRmMGUyNzVhOTIzMWZiNGExMTY3MWRkY2E0YWE0ZjVjNjk3YzUx
+    Mzc0MDIyOGQyNDE5ZjdhNzc2OGJmY2FjNDRmNzkxMzNmNTdiZWE=
   data.tar.gz: !binary |-
-    MDk5OGVlNzgxYWJlNzZmYTk0ODNhZDM3MmE0YTM4YjU5YWViYzhmMjYwNGE1
-    NWM5MjBkZTcwMzg0NjA5ZTcyZTM1NzA3YWQ0Njc4NjQ2MGIwY2NjYmYzOGYx
-    OGEwN2E0OGJhMTliOWZhMWRmMGFkMjAyN2YxMDQzODE2OWE2Zjc=
+    Y2MyYTQ2YjMzNzI0MzhjNDg1YjRmMDllNzRmN2IzODBjM2IyMDgyMGVhNDM4
+    OGZkZjE2YWRkYTEzYjIzNmJmNTE3YWY5ZDI1YmNhODRlNDNkZDg1YzYwOTMz
+    NDE3OTUxMmQ0MDM1ODQyYWY2YzA5M2QxNGZkZGU4ZmE5OGQ4Y2I=

data/lib/party_cc/httparty.rb

@@ -1,7 +1,7 @@
 module HTTParty
   class << self
 
-    def generate_stub method, response_filename, url, params
+    def generate_stub(method, response_filename, url, params={})
       %Q|stub_request(:#{method}, '#{url}')
         .with(#{params})
         .to_return(File.new('#{response_filename}'))|
@@ -11,19 +11,22 @@ module HTTParty
       buffer.string
         .gsub('=>', ': ')
         .gsub('nil', 'null')
+    rescue
+      buffer
     end
 
-    def pp_body body
+    def pp_body(body, type)
       buffer = StringIO.new
-
-      jsonified = JSON.load body
-
-      PP.pp(jsonified, buffer)
-
-      buffer
-    rescue
-      xmldoc = REXML::Document.new body
-      xmldoc.write buffer
+      case
+      when type.downcase.include?('json')
+        jsonified = JSON.load body
+        PP.pp(jsonified, buffer)
+      when type.downcase.include?('xml')
+        xmldoc = REXML::Document.new(body)
+        xmldoc.write(buffer)
+      else
+        PP.pp(body, buffer)
+      end
 
       buffer
     end
@@ -52,14 +55,14 @@ module HTTParty
           base_file_name = "#{method}_\#{path}_\#{stmp}.txt"
           response_file_name = "response_\#{base_file_name}"
 
-          prettied_rsp = decode_ruby(pp_body(rsp.response.body))
-          adjusted_headers = adjust_content_length_for_pretty_print prettied_rsp, rsp
+          prettied_rsp = decode_ruby(pp_body(rsp.response.body, rsp.headers['content-type']))
+          adjusted_headers = adjust_content_length_for_pretty_print(prettied_rsp, rsp)
 
-          File.write("stub_\#{base_file_name}", generate_stub(:#{method}, response_file_name, *args))
           File.open(response_file_name, 'w') do |f|
-            f.write adjusted_headers
-            f.write prettied_rsp
+            f.write(adjusted_headers)
+            f.write(prettied_rsp)
           end
+          File.write("stub_\#{base_file_name}", generate_stub(:#{method}, response_file_name, *args))
 
           rsp
         end

data/lib/party_cc/version.rb

@@ -1,3 +1,3 @@
 module PartyCC
-  VERSION = "0.0.7"
+  VERSION = "0.0.8"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: party_cc
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors:
 - barberj
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-01-06 00:00:00.000000000 Z
+date: 2014-03-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler