1 files changed, 37 insertions, 0 deletions
diff --git a/spec/lib/peek/views/click_house_spec.rb b/spec/lib/peek/views/click_house_spec.rb
new file mode 100644
index 00000000000..9d7d06204fc
--- /dev/null
+++ b/spec/lib/peek/views/click_house_spec.rb
@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Peek::Views::ClickHouse, :click_house, :request_store, feature_category: :database do
+  before do
+    allow(::Gitlab::PerformanceBar).to receive(:enabled_for_request?).and_return(true)
+  end
+
+  describe '#results' do
+    let(:results) { described_class.new.results }
+
+    it 'includes performance details' do
+      ::Gitlab::SafeRequestStore.clear!
+
+      data = ClickHouse::Client.select('SELECT 1 AS value', :main)
+      ClickHouse::Client.execute('INSERT INTO events (id) VALUES (1)', :main)
+
+      expect(data).to eq([{ 'value' => 1 }])
+
+      expect(results[:calls]).to eq(2)
+      expect(results[:duration]).to be_kind_of(String)
+
+      expect(results[:details]).to match_array([
+        a_hash_including({
+          sql: 'SELECT 1 AS value',
+          database: 'database: main'
+        }),
+        a_hash_including({
+          sql: 'INSERT INTO events (id) VALUES (1)',
+          database: 'database: main',
+          statistics: include('written_rows=>"1"')
+        })
+      ])
+    end
+  end
+end