rapflag 0.0.2 → 0.0.3

data/lib/rapflag/poloniex.rb

@@ -0,0 +1,325 @@
+require 'csv'
+require 'open-uri'
+require 'faraday'
+require 'fileutils'
+require 'rapflag/history'
+require 'poloniex'
+require 'json'
+require 'pp'
+
+module RAPFLAG
+
+  class Poloniex < History
+    attr_reader :complete_balances, :active_loans, :lending_history, :deposits, :withdrawals, :deposit_addresses,
+        :trade_history, :available_account_balances, :open_orders, :tradable_balances
+
+    @@btc_to_usd = {}
+    @@bfx_to_usd = {}
+
+    def get_usd_exchange(date_time = Time.now, from='BTC')
+      return 1.0 if from == 'USD'
+      daily  = ::Poloniex.get_all_daily_exchange_rates('BTC_GNT')
+
+      key = date_time.strftime(DATE_FORMAT)
+      return @@btc_to_usd[key] if from.eql?('BTC') && @@btc_to_usd.size > 0
+      return @@bfx_to_usd[key] if from.eql?('BFX') && @@bfx_to_usd.size > 0
+      ms =  (date_time.is_a?(Date) ? date_time.to_time : date_time).to_i*1000
+      ms_next_date = ms + (3*24*3600)*1000
+      # this does not work
+      # url = "https://api.bitfinex.com/v2/candles/trade:1D:t#{from}USD/hist?start:#{ms}?end:#{ms_next_date}"
+      url = "https://api.bitfinex.com/v2/candles/trade:1D:t#{from}USD/hist?start:#{ms}?end:#{ms_next_date}"
+      # therefore we just return the most uptodate
+      url = "https://api.bitfinex.com/v2/candles/trade:1D:t#{from}USD/hist?start:#{ms}"
+      puts "Fetching #{date_time}: #{url} #{@@btc_to_usd.size} BTC #{@@bfx_to_usd.size} BFX" if $VERBOSE
+      response = Faraday.get(url)
+      items = eval(response.body)
+      rates = {}
+      items.each do |item|
+        if item.first.eql?(:error)
+          puts "Fetching returned #{item}. Aborting"
+          exit(1)
+        end
+        timestamp = Time.at(item.first/1000).strftime(DATE_FORMAT)
+        rates[timestamp] = item[2]
+      end;
+      from.eql?('BTC') ? @@btc_to_usd = rates.clone : @@bfx_to_usd = rates.clone
+      rates[key] ? rates[key] : nil
+    rescue => err
+      puts "Err #{err}"
+    end
+
+    def dump_history
+      load_history_info
+      FileUtils.makedirs('output') unless File.directory?('output')
+      CSV.open('output/trade_history.csv','w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => [ 'currency_pair'] + @trade_history.values.first.first.to_h.keys
+        ) do |csv|
+        @trade_history.each do |currency_pair, trades|
+          trades.each do |trace|
+            csv << [ currency_pair] + trace.to_h.values
+          end
+        end
+      end
+      CSV.open('output/lending_history.csv','w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => @lending_history.first.to_h.keys
+        ) do |csv|
+        @lending_history.each do |info|
+          csv << info.to_h.values
+        end
+      end
+      CSV.open('output/tradable_balances.csv','w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => [ 'from_currency', 'to_from_currency', ]
+        ) do |csv|
+        @tradable_balances.each do |currency_pair, balance|
+          balance.each do |info|
+            csv << [ currency_pair] + info
+          end
+        end
+      end
+      CSV.open('output/complete_balances.csv', 'w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => [ 'currency', 'available', 'onOrders', 'btcValue' ]
+        ) do |csv|
+        @complete_balances.each do |balance|
+          csv << [balance[0]] + balance[1].values
+        end
+      end
+      CSV.open('output/active_loans.csv', 'w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => [ 'key', 'id', 'currency', 'rate', 'amount', 'duration', 'autoRenew', 'date', 'fees', ]
+        ) do |csv|
+        @active_loans.each do |key, loans|
+          loans.each do | loan |
+            csv << [key] + loan.values
+          end
+        end
+      end
+
+      CSV.open('output/available_account_balances.csv', 'w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => [ 'key', 'currency', 'balance']
+        ) do |csv|
+        @available_account_balances.each do |key, balances|
+          balances.each do |currency, balance|
+            csv << [key, currency, balance]
+          end
+        end
+      end
+      CSV.open('output/deposit_addresses.csv', 'w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => [ 'currency', 'id']
+        ) do |csv|
+        @deposit_addresses.each do |currency, id|
+          csv << [currency, id]
+        end
+      end
+      CSV.open('output/withdrawals.csv','w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => @deposits.first.to_h.keys
+        ) do |csv|
+        @deposits.each do |info|
+          csv << info.to_h.values
+        end
+      end
+      CSV.open('output/deposits.csv','w+',
+               :col_sep => ';',
+               :write_headers=> true,
+               :headers => @withdrawals.first.to_h.keys
+        ) do |csv|
+        @withdrawals.each do |info|
+          csv << info.to_h.values
+        end
+      end
+    end
+
+    def find_per_currency_and_day(items, currency, date)
+      puts "Searching for #{currency} in day #{date}" if $VERBOSE
+      start_time  = date.to_time.to_i
+      end_time    = (date + 1).to_time.to_i
+      found = items.find_all{|item| item.currency.eql?(currency) && item.timestamp >= start_time && item.timestamp < end_time}
+      puts "find_per_currency_and_day: found #{found.size} #{found.first} #{found.last}" if $VERBOSE
+      found
+    end
+
+    def find_lending_info_day(currency, date)
+      puts "Searching lending (close) for #{currency} in day #{date}" if $VERBOSE
+      found = @lending_history.find_all { |item| item.currency.eql?(currency) && date.eql?(Date.parse(item.close)) }
+      puts "find_lending_info_day: found #{found.size} #{found.first} #{found.last}" if $VERBOSE
+      found
+    end
+
+    # type is either buy or sell
+    def find_day_trade(currency, date, type)
+      found = []
+      @trade_history.each do |currency_pair, trades|
+        next unless /^#{currency}/.match(currency_pair)
+        trades.each do |trade|
+          if trade.type.eql?(type) && date.eql?(Date.parse(trade.date))
+            trade.cucurrency_pair = currency_pair
+            found << trade
+          end
+        end
+      end
+      puts "find_day_trade: #{date} #{currency} found #{found.size}" if $VERBOSE
+      found
+    end
+    def create_csv_file
+      binding.pry
+      puts "create_csv_file: already done"
+    end
+
+    def fetch_csv_history
+      load_history_info
+      @trade_history.values.collect{|values| values.collect{|x| x.date} }.max
+
+      max_time = [ @deposits.collect{|x| x.timestamp}.max,
+                   Time.parse(@trade_history.values.collect{|values| values.collect{|x| x.date} }.max.max).to_i,
+                   Time.parse(@lending_history.collect{|x| x.close}.max).to_i,
+                   @withdrawals.collect{|x| x.timestamp}.max,].max
+      min_time = [ @deposits.collect{|x| x.timestamp}.min,
+                   Time.parse(@trade_history.values.collect{|values| values.collect{|x| x.date} }.min.min).to_i,
+                   Time.parse(@lending_history.collect{|x| x.close}.min).to_i,
+                   @withdrawals.collect{|x| x.timestamp}.min,].min
+
+      min_date = Time.at(min_time).to_date
+      max_date = Time.at(max_time).to_date
+
+      puts "We start using the available_account_balances"
+      pp @available_account_balances
+      puts
+      puts "Should we better start using the complete_balances"
+      pp @complete_balances
+      puts "How should I handle the following currency_pair"
+      pp @trade_history.keys
+      @available_account_balances.each do |key, balances|
+        balances.each do |currency, balance|
+          puts "Calculation history for #{key} #{currency} with current balance of #{balance}"
+          out_name = "output/poloniex/#{key}_#{currency}.csv"
+          FileUtils.makedirs(File.dirname(out_name)) unless File.exists?(File.dirname(out_name))
+          @history = []
+          current_day =  max_date
+          current_balance  = balance.to_f
+          while (current_day >= min_date)
+            entry = OpenStruct.new
+            entry.current_day = current_day
+            entry.current_balance = current_balance
+
+            deposits = find_per_currency_and_day(@deposits, currency,current_day)
+            sum_deposits = 0.0; deposits.each{ |x| sum_deposits += x.amount.to_f }
+            entry.deposits = sum_deposits
+
+            withdrawals = find_per_currency_and_day(@withdrawals, currency, current_day)
+            sum_withdrawals = 0.0; withdrawals.each{ |x| sum_withdrawals += x.amount.to_f }
+            entry.withdrawals = sum_withdrawals
+
+            lendings = find_lending_info_day(currency, current_day)
+            earned = 0.0;  sum_fee = 0.0; lendings.each{ |x| earned += x.earned.to_f; sum_fee += x.fee.to_f }
+            entry.earned = earned
+            entry.fees = sum_fee
+
+            # End_of_Day_Balance = End_of_Day_Balance(-1) + Deposits - Withdrawals + Lending_Income - Trading_Fees + Purchases - Sales
+            sales = find_day_trade(currency, current_day, 'sell')
+            sum_sales = 0.0; sales.each{ |sale| sum_sales += sale.amount.to_f }
+            entry.sales = sum_sales
+
+            purchases = find_day_trade(currency, current_day, 'buy')
+            sum_purchase = 0.0; purchases.each{ |purchase| sum_purchase += purchase.amount.to_f }
+            entry.purchases = sum_purchase
+            entry.day_difference = sum_deposits - sum_withdrawals + sum_purchase - sum_sales + sum_fee
+            @history << entry
+            current_day -= 1
+            # balance for previous day
+            current_balance = entry.current_balance - entry.day_difference
+          end
+          next unless @history.size > 0
+          CSV.open(out_name,'w+',
+                  :col_sep => ';',
+                  :write_headers=> true,
+                  :headers => @history.first.to_h.keys
+            ) do |csv|
+            @history.each do |info|
+              csv << info.to_h.values
+            end
+          end
+        end
+      end
+    end
+    private
+    def check_config
+      @spec_data = File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'spec', 'data'))
+      ['poloniex_api_key',
+       'poloniex_secret',
+       ].each do |item|
+        raise "Must define #{item} in config.yml" unless Config[item]
+      end
+      ::Poloniex.setup do | config |
+          config.key    = Config['poloniex_api_key']
+          config.secret = Config['poloniex_secret']
+      end
+      nil
+    end
+    private
+    def load_or_save_json(name, param = nil)
+      json_file = File.join(@spec_data, name.to_s + '.json')
+      if File.exist?(json_file) && defined?(RSpec)
+        body = IO.read(json_file)
+      else
+        cmd = param ? "::Poloniex.#{name.to_s}('#{param}').body" : "::Poloniex.#{name.to_s}.body"
+        body = eval(cmd)
+        File.open(json_file, 'w+') { |f| f.write(body)}
+      end
+      eval("@#{name} = JSON.parse(body)")
+    end
+    def load_history_info
+      check_config
+      begin
+        @balances = load_or_save_json(:balances)
+      rescue => error
+        puts "Error was #{error.inspect}"
+        puts "Calling @balances from poloniex failed. Configuration was"
+        pp ::Poloniex.configuration
+        exit 1
+      end
+      @active_loans = load_or_save_json(:active_loans)
+      @available_account_balances = load_or_save_json(:available_account_balances)
+      all = load_or_save_json(:complete_balances)
+      @complete_balances = all.find_all{ | currency, values| values["available"].to_f != 0.0 }
+      @deposit_addresses = load_or_save_json(:deposit_addresses)
+
+      @deposits_withdrawals  = load_or_save_json(:deposits_withdrawls)
+      # deposits and withdrawals have a different structure
+      @deposits = []
+      @deposits_withdrawals['deposits'].each {|x| @deposits << OpenStruct.new(x) };
+      @withdrawals =[]
+      @deposits_withdrawals['withdrawals'].each {|x| @withdrawals << OpenStruct.new(x) };
+      info = load_or_save_json(:lending_history)
+      @lending_history = []
+      info.each {|x| @lending_history << OpenStruct.new(x) };
+
+      @open_orders  = load_or_save_json(:open_orders, 'all')
+      info  = load_or_save_json(:trade_history, 'all')
+      @trade_history = {}
+      info.each do |currency_pair, trades|
+        @trade_history[currency_pair] = []
+        trades.each {|x| @trade_history[currency_pair] << OpenStruct.new(x) };
+        @trade_history[currency_pair].sort!{|x,y| x[:date] <=> y[:date]}.collect{ |x| x[:date]}
+      end
+      # @trade_history.values.first.first.tradeID
+      @tradable_balances  = load_or_save_json(:tradable_balances)
+      @active_loans   # key
+      @provided_loans = []; @active_loans['provided'].each {|x| @provided_loans << OpenStruct.new(x) };
+      @used_loans = []; @active_loans['used'].each {|x| @used_loans << OpenStruct.new(x) };
+    end
+  end
+end

data/lib/rapflag/version.rb

@@ -1,4 +1,4 @@
 module RAPFLAG
   Wallets = ['trading', 'exchange', 'deposit']
-  VERSION='0.0.2'
+  VERSION='0.0.3'
 end

data/rapflag.gemspec

@@ -17,6 +17,7 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
+  spec.add_dependency 'poloniex'
   spec.add_dependency 'bitfinex-rb'
   spec.add_dependency 'trollop'
 

data/spec/bitfinex_spec.rb

@@ -0,0 +1,128 @@
+# encoding: utf-8
+require 'spec_helper'
+require 'rapflag/bitfinex'
+require 'vcr'
+require 'date'
+
+VCR.configure do |config|
+  config.cassette_library_dir = "fixtures/vcr_cassettes"
+  config.hook_into :faraday # or :fakeweb
+end
+
+VCR.eject_cassette # we use insert/eject around each example
+describe RAPFLAG::Bitfinex do
+  OUTPUT_ROOT = File.expand_path(File.join(__FILE__, '..', '..', 'output', 'bitfinex'))
+  Bitfinex_CSV_Test_File             = OUTPUT_ROOT + '/BTC_exchange.csv'
+  BITFINEX_SUMMARY_EXCHANGE_BTC_File = OUTPUT_ROOT + '/BTC_exchange_summary.csv'
+  BITFINEX_SUMMARY_DEPOSIT_BFX_File  = OUTPUT_ROOT + '/BFX_deposit_summary.csv'
+
+  # include ServerMockHelper
+  before(:all) do
+  end
+  after(:all) do
+  end
+  context 'bitfinex' do
+    before(:all) do
+      VCR.use_cassette("rapflag", :record => :new_episodes) do
+        FileUtils.rm_f(Bitfinex_CSV_Test_File) if File.exist?(Bitfinex_CSV_Test_File)
+        FileUtils.rm_f(BITFINEX_SUMMARY_DEPOSIT_BFX_File) if File.exist?(BITFINEX_SUMMARY_DEPOSIT_BFX_File)
+        FileUtils.rm_f(BITFINEX_SUMMARY_EXCHANGE_BTC_File) if File.exist?(Bitfinex_CSV_Test_File)
+        expect(File.exist?(Bitfinex_CSV_Test_File)).to eql(false)
+        @rap = RAPFLAG::Bitfinex.new('exchange', 'BTC')
+        @rap.fetch_csv_history
+        @rap.create_csv_file
+      end
+    end
+    context 'history' do
+      it 'should have correct currency' do
+        expect(@rap.currency).to eql('BTC')
+      end
+      it 'should have correct size' do
+        expect(@rap.history.size).to eql(206)
+      end
+    end
+  end
+  context 'bitfinex CSV' do
+    context 'csv' do
+      it 'should have generated a correct CSV file' do
+        expect(File.exist?(Bitfinex_CSV_Test_File)).to eql(true)
+        lines = IO.readlines(Bitfinex_CSV_Test_File)
+        expect(lines.first.chomp).to eql('currency;amount;balance;description;date_time')
+        expect(lines[1].chomp).to eql(
+          'BTC;-0.00000005;0.0;Transfer of 0.0 BTC from wallet Exchange to Deposit on wallet Exchange;2016.12.03 21:20:47')
+      end
+    end
+  end
+  context 'exchange option --clean' do
+    before(:all) do
+      @date_bfx_1 = Date.new(2017,1,10)
+      @date_btx_1 = Date.new(2017,1,21)
+      @date_btx_2 = Date.new(2017,1,10)
+      VCR.use_cassette("rapflag", :record => :new_episodes) do
+        FileUtils.rm_f(BITFINEX_SUMMARY_EXCHANGE_BTC_File) if File.exist?(Bitfinex_CSV_Test_File)
+        expect(File.exist?(BITFINEX_SUMMARY_EXCHANGE_BTC_File)).to eql(false)
+        @exchange = RAPFLAG::Bitfinex.new('exchange', 'BTC')
+        @exchange.fetch_csv_history
+        @exchange.create_summary
+        @bfx   = @exchange.get_usd_exchange(@date_bfx_1, 'BFX')
+        @btx_1 = @exchange.get_usd_exchange(@date_btx_1, 'BTC')
+        @btx_2 = @exchange.get_usd_exchange(@date_btx_2, 'BTC')
+      end
+    end
+    it 'should have generated a correct summary CSV file' do
+      expect(File.exist?(BITFINEX_SUMMARY_EXCHANGE_BTC_File)).to eql(true)
+      lines = IO.readlines(BITFINEX_SUMMARY_EXCHANGE_BTC_File)
+      expect(lines.first.chomp).to eql('currency;date;income;balance;rate;balance_in_usd')
+      expect(lines[1].chomp).to eql('BTC;2016.01.15;0.0;8.99788147;;')
+      expect(lines[-1].chomp).to eql('BTC;2016.12.03;0.0;0.0;765.46;0.0')
+    end
+    it 'should have a balance for for each day' do
+      expect(File.exist?(BITFINEX_SUMMARY_EXCHANGE_BTC_File)).to eql(true)
+      lines = IO.readlines(BITFINEX_SUMMARY_EXCHANGE_BTC_File)
+      first_date = Date.strptime(lines[1].chomp.split(';')[1], '%Y.%m.%d')
+      last_date = Date.strptime(lines[-1].chomp.split(';')[1], '%Y.%m.%d')
+      (last_date >  first_date).should be true
+      nr_dates = 323
+      (last_date -  first_date).to_i.should eql nr_dates
+      (1..nr_dates).each do |j|
+        wish_date = (first_date + j).strftime('%Y.%m.%d')
+        binding.pry unless lines.find_all{|line| line.index(wish_date)}.size == 1
+        lines.find_all{|line| line.index(wish_date)}.size.should eql 1
+      end
+    end
+    it 'should have NOT have generated a correct summary deposit BFX CSV file' do
+      expect(File.exist?(BITFINEX_SUMMARY_DEPOSIT_BFX_File)).to eql(false)
+    end
+    it 'should have the correct BTC -> USD rate' do
+      expect(@btx_1).to eql 924.02
+      expect(@btx_2).to eql 905.76
+    end
+    it 'should have the correct BFX -> USD rate' do
+      expect(@bfx).to eql 0.5697
+    end
+  end
+  context 'deposit option --clean' do
+    before(:all) do
+      FileUtils.rm_f(BITFINEX_SUMMARY_EXCHANGE_BTC_File) if File.exist?(BITFINEX_SUMMARY_EXCHANGE_BTC_File)
+      FileUtils.rm_f(BITFINEX_SUMMARY_DEPOSIT_BFX_File) if File.exist?(Bitfinex_CSV_Test_File)
+      @date_bfx_1 = Date.new(2017,1,10)
+      @date_btx_1 = Date.new(2017,1,21)
+      @date_btx_2 = Date.new(2017,1,10)
+      VCR.use_cassette("rapflag", :record => :new_episodes) do
+        expect(File.exist?(BITFINEX_SUMMARY_DEPOSIT_BFX_File)).to eql(false)
+        @deposit = RAPFLAG::Bitfinex.new('deposit', 'BFX')
+        @deposit.fetch_csv_history
+        @deposit.create_summary
+      end
+    end
+    it 'should have NOT generated a exchange BTC summary CSV file' do
+      expect(File.exist?(BITFINEX_SUMMARY_EXCHANGE_BTC_File)).to eql(false)
+    end
+    it 'should have NOT generated a correct summary CSV file' do
+      expect(File.exist?(BITFINEX_SUMMARY_DEPOSIT_BFX_File)).to eql(true)
+      lines = IO.readlines(BITFINEX_SUMMARY_DEPOSIT_BFX_File)
+      expect(lines.first.chomp).to eql('currency;date;income;balance;rate;balance_in_usd')
+      expect(lines[1].chomp).to eql('BFX;2016.01.15;0.0;8.99788147;;')
+    end
+  end
+end

data/spec/poloniex_spec.rb

@@ -0,0 +1,115 @@
+# encoding: utf-8
+require 'spec_helper'
+require 'rapflag/poloniex'
+require 'vcr'
+
+VCR.configure do |config|
+  config.cassette_library_dir = "fixtures/vcr_cassettes"
+  config.hook_into :faraday
+end
+
+VCR.eject_cassette # we use insert/eject around each example
+
+describe RAPFLAG::Poloniex do
+  TEST_OUTPUT_ROOT = File.expand_path(File.join(__FILE__, '..', '..', 'output/poloniex'))
+  Poloniex_CSV_Test_File             = TEST_OUTPUT_ROOT + '/BTC_exchange.csv'
+  Poloniex_SUMMARY_EXCHANGE_BTC_File = TEST_OUTPUT_ROOT + '/BTC_exchange_summary.csv'
+  Poloniex_SUMMARY_DEPOSIT_BFX_File  = TEST_OUTPUT_ROOT + '/BFX_deposit_summary.csv'
+
+  context 'poloniex' do
+    before(:all) do
+      # skip("We cannot yet use VCR here. Therefore omitting all spec tests")
+      # VCR.use_cassette("poloniex", :record => :new_episodes) do
+        FileUtils.rm_f(Poloniex_CSV_Test_File) if File.exist?(Poloniex_CSV_Test_File)
+        FileUtils.rm_f(Poloniex_SUMMARY_DEPOSIT_BFX_File) if File.exist?(Poloniex_SUMMARY_DEPOSIT_BFX_File)
+        FileUtils.rm_f(Poloniex_SUMMARY_EXCHANGE_BTC_File) if File.exist?(Poloniex_CSV_Test_File)
+        expect(File.exist?(Poloniex_CSV_Test_File)).to eql(false)
+        @rap = RAPFLAG::Poloniex.new('exchange', 'BTC')
+        @rap.fetch_csv_history
+        # @rap.create_csv_file
+       # end
+    end
+    context 'history' do
+      it 'should have correct currency' do
+        expect(@rap.currency).to eql('BTC')
+      end
+      it 'should have correct size' do
+        expect(@rap.history.size).to eql(368)
+      end
+    end
+  end
+  context 'poloniex CSV' do
+    before(:all) do
+      skip("We cannot yet use VCR here. Therefore omitting all spec tests")
+    end
+    context 'csv' do
+      it 'should have generated a correct CSV file' do
+        expect(File.exist?(Poloniex_CSV_Test_File)).to eql(true)
+        lines = IO.readlines(Poloniex_CSV_Test_File)
+        expect(lines.first.chomp).to eql('currency,amount,balance,description,date_time')
+        expect(lines[1].chomp).to eql(
+          'BTC,-0.00000005,0.0,Transfer of 0.0 BTC from wallet Exchange to Deposit on wallet Exchange,2016.12.03 21:20:47')
+      end
+    end
+  end
+  context 'exchange option --clean' do
+    before(:all) do
+      skip("We cannot yet use VCR here. Therefore omitting all spec tests")
+      @date_bfx_1 = Date.new(2017,1,10)
+      @date_btx_1 = Date.new(2017,1,21)
+      @date_btx_2 = Date.new(2017,1,10)
+      VCR.use_cassette("poloniex", :record => :new_episodes) do
+        FileUtils.rm_f(Poloniex_SUMMARY_EXCHANGE_BTC_File) if File.exist?(Poloniex_CSV_Test_File)
+        expect(File.exist?(Poloniex_SUMMARY_EXCHANGE_BTC_File)).to eql(false)
+        @exchange = RAPFLAG::Poloniex.new('exchange', 'BTC')
+        @exchange.fetch_csv_history
+        @exchange.create_summary
+        @bfx   = @exchange.get_usd_exchange(@date_bfx_1, 'BFX')
+        @btx_1 = @exchange.get_usd_exchange(@date_btx_1, 'BTC')
+        @btx_2 = @exchange.get_usd_exchange(@date_btx_2, 'BTC')
+      end
+    end
+    it 'should have generated a correct summary CSV file' do
+      expect(File.exist?(Poloniex_SUMMARY_EXCHANGE_BTC_File)).to eql(true)
+      lines = IO.readlines(Poloniex_SUMMARY_EXCHANGE_BTC_File)
+      expect(lines.first.chomp).to eql('currency,date,income,balance,rate,balance_in_usd')
+      expect(lines[1].chomp).to eql('BTC,2016.01.15,0.0,8.99788147,,')
+      expect(lines[-1].chomp).to eql('BTC,2016.12.03,0.0,0.0,765.46,0.0')
+    end
+    it 'should have NOT have generated a correct summary deposit BFX CSV file' do
+      expect(File.exist?(Poloniex_SUMMARY_DEPOSIT_BFX_File)).to eql(false)
+    end
+    it 'should have the correct BTC -> USD rate' do
+      expect(@btx_1).to eql 924.02
+      expect(@btx_2).to eql 905.76
+    end
+    it 'should have the correct BFX -> USD rate' do
+      expect(@bfx).to eql 0.5697
+    end
+  end
+  context 'deposit option --clean' do
+    before(:all) do
+      skip("We cannot yet use VCR here. Therefore omitting all spec tests")
+      FileUtils.rm_f(Poloniex_SUMMARY_EXCHANGE_BTC_File) if File.exist?(Poloniex_SUMMARY_EXCHANGE_BTC_File)
+      FileUtils.rm_f(Poloniex_SUMMARY_DEPOSIT_BFX_File) if File.exist?(Poloniex_CSV_Test_File)
+      @date_bfx_1 = Date.new(2017,1,10)
+      @date_btx_1 = Date.new(2017,1,21)
+      @date_btx_2 = Date.new(2017,1,10)
+      VCR.use_cassette("rapflag", :record => :new_episodes) do
+        expect(File.exist?(Poloniex_SUMMARY_DEPOSIT_BFX_File)).to eql(false)
+        @deposit = RAPFLAG::Poloniex.new('deposit', 'BFX')
+        @deposit.fetch_csv_history
+        @deposit.create_summary
+      end
+    end
+    it 'should have NOT generated a exchange BTC summary CSV file' do
+      expect(File.exist?(Poloniex_SUMMARY_EXCHANGE_BTC_File)).to eql(false)
+    end
+    it 'should have NOT generated a correct summary CSV file' do
+      expect(File.exist?(Poloniex_SUMMARY_DEPOSIT_BFX_File)).to eql(true)
+      lines = IO.readlines(Poloniex_SUMMARY_DEPOSIT_BFX_File)
+      expect(lines.first.chomp).to eql('currency,date,income,balance,rate,balance_in_usd')
+      expect(lines[1].chomp).to eql('BFX,2016.01.15,0.0,8.99788147,,')
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -4,7 +4,9 @@ rescue LOAD_ERROR
 end
 
 require 'rapflag/version'
+require 'rapflag/config'
 require 'bitfinex-api-rb'
+require 'poloniex'
 
 module RAPFLAG
   # create dummy config for spec tests
@@ -13,6 +15,10 @@ module RAPFLAG
     conf.secret  = 'SECRECT_KEY'
     conf.websocket_api_endpoint = 'wss://api.bitfinex.com/ws'
   end
+  Poloniex.setup do | config |
+    config.key    = 'P_123467889'
+    config.secret = 'P_SECRECT_KEY'
+  end
 end
 
 # This file was generated by the `rspec --init` command. Conventionally, all

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: rapflag
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Zeno R.R. Davatz, Niklaus Giger
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-03-29 00:00:00.000000000 Z
+date: 2017-04-05 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: poloniex
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bitfinex-rb
   requirement: !ruby/object:Gem::Requirement
@@ -133,16 +147,20 @@ files:
 - ".rspec"
 - ".travis.yml"
 - Gemfile
+- History.md
 - LICENSE
 - README.md
 - Rakefile
 - bin/rapflag
 - fixtures/vcr_cassettes/rapflag.yml
+- lib/rapflag/bitfinex.rb
 - lib/rapflag/config.rb
-- lib/rapflag/fetch.rb
+- lib/rapflag/history.rb
+- lib/rapflag/poloniex.rb
 - lib/rapflag/version.rb
 - rapflag.gemspec
-- spec/rapflag_spec.rb
+- spec/bitfinex_spec.rb
+- spec/poloniex_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/zdavatz/rapflag
 licenses:
@@ -169,5 +187,6 @@ signing_key:
 specification_version: 4
 summary: Bitfinex Exporter for your Taxman from ywesee
 test_files:
-- spec/rapflag_spec.rb
+- spec/bitfinex_spec.rb
+- spec/poloniex_spec.rb
 - spec/spec_helper.rb