Feature/17/add kind inplit to trading

app/models/trading.rb

@@ -5,7 +5,7 @@ class Trading < ApplicationRecord
   belongs_to :stock
   belongs_to :asset
 
-  enum kind: { buy: 0, sale: 1 }
+  enum kind: { buy: 0, sale: 1, inplit: 2 }
 
   before_validation :set_stock, if: -> { stock.nil? && stock_code.present? }
   before_validation :set_asset

app/services/assets/updater.rb

@@ -12,10 +12,13 @@ def self.call(asset, trading)
     end
 
     def call
-      if trading.kind == 'buy'
+      case trading.kind
+      when 'buy'
         calculate_buy
-      else
+      when 'sale'
         calculate_sale
+      else
+        calculate_inplit
       end
 
       asset.save
@@ -41,5 +44,10 @@ def calculate_sale
         asset.total_invested = (asset.average_price * asset.amount).round(2)
       end
     end
+
+    def calculate_inplit
+      asset.amount = asset.amount / trading.amount
+      asset.average_price = asset.average_price * trading.amount
+    end
   end
 end

config/locales/activerecord/pt-BR.yml

@@ -19,7 +19,7 @@ pt-BR:
       trading:
         amount: 'Quant.'
         date: 'Data'
-        kind: 'C / V'
+        kind: 'Tipo Mov.'
         total_value: 'Valor total'
         value_unit: 'Valor unit.'
         asset: 'Ativo'
@@ -29,6 +29,7 @@ pt-BR:
         kind_enum:
           buy: 'Compra'
           sale: 'Venda'
+          inplit: 'Grupamento'
 
     errors:
       messages:

spec/helpers/tradings_helper_spec.rb

@@ -4,5 +4,6 @@
   context '.trading_locale_kind' do
     it { expect(helper.trading_locale_kind('buy')).to eq('Compra') }
     it { expect(helper.trading_locale_kind('sale')).to eq('Venda') }
+    it { expect(helper.trading_locale_kind('inplit')).to eq('Grupamento') }
   end
 end

spec/services/assets/updater_spec.rb

@@ -42,4 +42,16 @@
       expect(trading.asset.average_price).to eq(0)
     end
   end
+
+  describe 'trading kind inplit' do
+    let(:trading) { create(:trading, kind: :inplit, amount: 10, stock_code: 'TEST99', user:) }
+
+    before :each do
+      create(:trading, amount: 100, value_unit: 2, total_value: 200, stock_code: 'TEST99', user:)
+    end
+
+    it { expect(trading.asset.amount).to eq(10) }
+    it { expect(trading.asset.average_price).to eq(20) }
+    it { expect(trading.asset.total_invested).to eq(200) }
+  end
 end