lib/eroi/response.rb in eroi-0.0.1 vs lib/eroi/response.rb in eroi-0.1.0

@@ -10,13 +10,19 @@
       end
     end
 
     class Get < Base
       def success?
-        !@data['ErrorCode']
+        (@data['ErrorCode'] || @data['xml']) ? false : true
       end
 
+      def contact
+        @data['Retrieve']['Record']
+      end
+
+      alias :contacts :contact
+
       def error_message
         case @data['ErrorCode'].to_i
         when 1
           'Invalid username/password was provided.'
         when 2
@@ -27,16 +33,16 @@
       end
     end
 
     class Post < Base
       def success?
-        @data['Compiled'] == 'Yes' &&
-        @data['DBConnect'] == 'OK' &&
-        @data['XMLUpload'] == 'Complete'
+        @data['Response']['Compiled'] == 'Yes' &&
+        @data['Response']['DBConnect'] == 'OK' &&
+        @data['Response']['XMLUpload'] == 'Complete'
       end
 
       def number_of_records
-        @data['ImportRecords'].to_i
+        @data['Response']['ImportRecords'].to_i
       end
     end
   end
 end