File: connection_spec.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (61 lines) | stat: -rw-r--r-- 1,434 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe ClickHouse::Connection, click_house: :without_migrations, feature_category: :database do
  let(:connection) { described_class.new(:main) }

  describe '#database_name' do
    it 'returns the configured database name' do
      name = ClickHouse::Client.configuration.databases[:main].database
      expect(connection.database_name).to eq(name)
    end
  end

  describe '#select' do
    it 'proxies select to client' do
      expect(
        connection.select('SELECT 1')
      ).to eq([{ '1' => 1 }])
    end
  end

  describe '#execute' do
    it 'proxies execute to client' do
      create_test_table

      connection.execute(
        <<~SQL
          INSERT INTO test_table VALUES (1), (2), (3)
        SQL
      )

      expect(connection.select('SELECT id FROM test_table')).to eq(
        [{ 'id' => 1 }, { 'id' => 2 }, { 'id' => 3 }]
      )
    end
  end

  describe '#table_exists?' do
    it "return false when table doesn't exist" do
      expect(connection.table_exists?('test_table')).to eq(false)
    end

    it 'returns true when table exists' do
      create_test_table

      expect(connection.table_exists?('test_table')).to eq(true)
    end
  end

  def create_test_table
    connection.execute(
      <<~SQL
        CREATE TABLE test_table (
          id   UInt64
        ) ENGINE = MergeTree
        PRIMARY KEY(id)
      SQL
    )
  end
end