lib/alma/fine_set.rb in alma-0.2.4 vs lib/alma/fine_set.rb in alma-0.2.5

@@ -1,21 +1,39 @@
 module Alma
-  class FineSet < ResultSet
+  class FineSet
+    extend Forwardable
+    include Enumerable
+    #include Alma::Error
 
-    def top_level_key
-      'fees'
+    attr_reader :response
+    def_delegators :list, :each, :size
+    def_delegators :response, :[], :fetch
+
+    def initialize(response_body_hash)
+      @response = response_body_hash
     end
 
-    def response_records_key
+    def key
       'fee'
     end
 
+    def list
+      fetch(key, [])
+    end
+
     def sum
-      @response[top_level_key].fetch('total_sum', 0)
+      fetch('total_sum', 0)
     end
+    alias :total_sum :sum
 
     def currency
-      @response[top_level_key].fetch('currency', nil)
+      fetch('currency', nil)
     end
 
+    def total_record_count
+      fetch('total_record_count', 0)
+    end
+    alias :total_records :total_record_count
+
+
   end
-end
\ No newline at end of file
+end