Initial commit

spec/lib/influxdb_spec.rb

@@ -0,0 +1,19 @@
+# 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