Add get_usage_allocation method

Author: waits

lib/dropbox/account.rb

@@ -34,4 +34,14 @@ def initialize(attrs={})
       super(attrs)
     end
   end
+
+  class SpaceUsage
+    attr_reader :used, :allocation, :allocated
+
+    def initialize(attrs={})
+      @used = attrs['used'] # Space used in bytes
+      @allocation = attrs['allocation']['.tag'] # The type of allocation
+      @allocated = attrs['allocation']['allocated'] # Space allocated in bytes
+    end
+  end
 end

lib/dropbox/client.rb

@@ -37,6 +37,7 @@ def get_account(id)
       parse_tagged_response(resp, 'basic_account')
     end
 
+    # Takes an array of account IDs and returns an array of BasicAccounts
     def get_account_batch(ids)
       resp = request('/users/get_account_batch', account_ids: ids)
       resp.map { |a| parse_tagged_response(a, 'basic_account') }
@@ -57,6 +58,11 @@ def get_preview(path)
       return parse_tagged_response(resp, 'file'), body
     end
 
+    def get_space_usage
+      resp = request('/users/get_space_usage')
+      SpaceUsage.new(resp)
+    end
+
     def get_temporary_link(path)
       resp = request('/files/get_temporary_link', path: path)
       return parse_tagged_response(resp['metadata'], 'file'), resp['link']

test/test_account.rb

@@ -35,4 +35,15 @@ def test_full_account_initialize
     assert_equal false, account.disabled
     assert_equal true, account.is_paired
   end
+
+  def test_space_usage_initialize
+    usage = Dropbox::SpaceUsage.new({
+      'used' => 1,
+      'allocation' => {'.tag' => 'team', 'allocated' => 2}
+    })
+
+    assert_equal 1, usage.used
+    assert_equal 'team', usage.allocation
+    assert_equal 2, usage.allocated
+  end
 end

test/test_users.rb

@@ -45,4 +45,12 @@ def test_get_current_account
     assert_equal false, account.disabled
   end
 
+  def test_get_space_usage
+    usage = @client.get_space_usage
+
+    assert usage.used.is_a?(Integer)
+    assert_equal 'individual', usage.allocation
+    assert usage.allocated.is_a?(Integer)
+  end
+
 end