# frozen_string_literal: true

describe Peatio::InfluxDB do
  context 'host sharding' do
    before do
      Peatio::InfluxDB.instance_variable_set(:@clients, {})
      Peatio::InfluxDB.stubs(:config).returns({ host: %w[inflxudb-0 inflxudb-1] })
    end

    after do
      Peatio::InfluxDB.instance_variable_set(:@clients, {})
    end

    it do
      expect(Peatio::InfluxDB.client(keyshard: 'btcusd').config.hosts).to eq(['inflxudb-1'])
      expect(Peatio::InfluxDB.client(keyshard: 'ethusd').config.hosts).to eq(['inflxudb-0'])
    end
  end
end