Add from/until sorting to balance API

app/controllers/repp/v1/accounts_controller.rb

@@ -2,10 +2,9 @@ module Repp
   module V1
     class AccountsController < BaseController
       def balance
-        return activity if params[:detailed] == 'true'
-
         resp = { balance: current_user.registrar.cash_account.balance,
                  currency: current_user.registrar.cash_account.currency }
+        resp[:activities] = activities if params[:detailed] == 'true'
         render_success(data: resp)
       end
 
@@ -17,22 +16,21 @@ module Repp
       end
 
       def activities
-        bal = current_user.registrar.cash_account.balance
-        act = []
         activities = current_user.registrar.cash_account.activities.order(created_at: :desc)
+        activities = activities.where('created_at >= ?', params[:from]) if params[:from]
+        activities = activities.where('created_at <= ?', params[:until]) if params[:until]
+        arr = []
         activities.each do |a|
-          act << {
+          arr << {
             created_at: a.created_at,
             description: a.description,
             type: a.activity_type == 'add_credit' ? 'credit' : 'debit',
             sum: a.sum,
-            balance: bal,
+            balance: a.new_balance,
           }
-
-          bal = a.activity_type == 'add_credit' ? bal = bal + a.sum : bal - a.sum
         end
 
-        act
+        arr
       end
     end
   end

test/integration/repp/v1/accounts/balance_test.rb

@@ -13,10 +13,6 @@ class ReppV1BalanceTest < ActionDispatch::IntegrationTest
     get '/repp/v1/accounts/balance', headers: @auth_headers
     json = JSON.parse(response.body, symbolize_names: true)
 
-    puts response.body
-    puts @registrar.registrar.cash_account.balance
-    puts @registrar.registrar.cash_account.currency
-
     assert_response :ok
     assert_equal 1000, json[:code]
     assert_equal 'Command completed successfully', json[:message]
@@ -28,7 +24,7 @@ class ReppV1BalanceTest < ActionDispatch::IntegrationTest
     started_from = "2020-01-01"
     end_to = DateTime.current.to_date.to_s(:db)
 
-    get "/repp/v1/accounts/balance?detailed=true&started_from=#{started_from}&end_to=#{end_to}", headers: @auth_headers
+    get "/repp/v1/accounts/balance?detailed=true&from=#{started_from}&until=#{end_to}", headers: @auth_headers
     json = JSON.parse(response.body, symbolize_names: true)
 
     assert_response :ok