pgtk 0.18.1 → 0.18.2

data/renovate.json

@@ -1,6 +0,0 @@
-{
-  "$schema": "https://docs.renovatebot.com/renovate-schema.json",
-  "extends": [
-    "config:base"
-  ]
-}

data/test/test__helper.rb

@@ -1,76 +0,0 @@
-# frozen_string_literal: true
-
-# SPDX-FileCopyrightText: Copyright (c) 2019-2025 Yegor Bugayenko
-# SPDX-License-Identifier: MIT
-
-$stdout.sync = true
-
-require 'simplecov'
-require 'simplecov-cobertura'
-unless SimpleCov.running || ENV['PICKS']
-  SimpleCov.command_name('test')
-  SimpleCov.formatter = SimpleCov::Formatter::MultiFormatter.new(
-    [
-      SimpleCov::Formatter::HTMLFormatter,
-      SimpleCov::Formatter::CoberturaFormatter
-    ]
-  )
-  SimpleCov.minimum_coverage 90
-  SimpleCov.minimum_coverage_by_file 90
-  SimpleCov.start do
-    add_filter 'test/'
-    add_filter 'vendor/'
-    add_filter 'target/'
-    track_files 'lib/**/*.rb'
-    track_files '*.rb'
-  end
-end
-
-require 'minitest/autorun'
-require 'minitest/reporters'
-Minitest::Reporters.use! [Minitest::Reporters::SpecReporter.new]
-
-require 'logger'
-require 'loog'
-require 'rake'
-require 'rake/tasklib'
-require_relative '../lib/pgtk'
-require_relative '../lib/pgtk/liquibase_task'
-require_relative '../lib/pgtk/pgsql_task'
-
-class Pgtk::Test < Minitest::Test
-  def fake_config
-    Dir.mktmpdir do |dir|
-      id = (Time.now.to_f * 1_000_000).to_i % 1_000_000
-      f = File.join(dir, 'cfg.yml')
-      Pgtk::PgsqlTask.new("pgsql#{id}") do |t|
-        t.dir = File.join(dir, 'pgsql')
-        t.user = 'hello'
-        t.password = 'A B C привет ! & | !'
-        t.dbname = 'test'
-        t.yaml = f
-        t.quiet = true
-      end
-      Rake::Task["pgsql#{id}"].invoke
-      Pgtk::LiquibaseTask.new("liquibase#{id}") do |t|
-        t.master = File.join(__dir__, '../test-resources/master.xml')
-        t.yaml = f
-        t.quiet = true
-      end
-      Rake::Task["liquibase#{id}"].invoke
-      assert_path_exists(f)
-      yield f
-    end
-  end
-
-  def fake_pool(log: Loog::NULL)
-    fake_config do |f|
-      pool = Pgtk::Pool.new(
-        Pgtk::Wire::Yaml.new(f),
-        log: log
-      )
-      pool.start(1)
-      yield pool
-    end
-  end
-end

data/test/test_impatient.rb

@@ -1,79 +0,0 @@
-# frozen_string_literal: true
-
-# SPDX-FileCopyrightText: Copyright (c) 2019-2025 Yegor Bugayenko
-# SPDX-License-Identifier: MIT
-
-require 'loog'
-require 'pg'
-require 'qbash'
-require 'rake'
-require 'tmpdir'
-require 'yaml'
-require_relative 'test__helper'
-require_relative '../lib/pgtk/pool'
-require_relative '../lib/pgtk/impatient'
-
-# Pool test.
-# Author:: Yegor Bugayenko (yegor256@gmail.com)
-# Copyright:: Copyright (c) 2017-2025 Yegor Bugayenko
-# License:: MIT
-class TestImpatient < Pgtk::Test
-  def test_takes_version
-    fake_pool do |pool|
-      v = Pgtk::Impatient.new(pool, 1).version
-      refute_nil(v)
-    end
-  end
-
-  def test_interrupts
-    fake_pool do |pool|
-      assert_raises(Pgtk::Impatient::TooSlow) do
-        Pgtk::Impatient.new(pool, 0.01).exec(
-          'SELECT COUNT(*) FROM generate_series(1, 1000000) AS a'
-        )
-      end
-    end
-  end
-
-  def test_skips_by_regex
-    fake_pool do |pool|
-      Pgtk::Impatient.new(pool, 0.01, /^SELECT.*$/).exec(
-        'SELECT COUNT(*) FROM generate_series(1, 1000000) AS a'
-      )
-    end
-  end
-
-  def test_doesnt_shadow_larger_timeout
-    fake_pool do |pool|
-      assert_raises(Timeout::Error) do
-        Timeout.timeout(0.1) do
-          Pgtk::Impatient.new(pool, 999).exec(
-            'SELECT COUNT(*) FROM generate_series(1, 100000000) AS a'
-          )
-        end
-      end
-    end
-  end
-
-  def test_doesnt_interrupt
-    fake_pool do |pool|
-      id = Pgtk::Impatient.new(pool, 1).exec(
-        'INSERT INTO book (title) VALUES ($1) RETURNING id',
-        ['1984']
-      ).first['id'].to_i
-      assert_predicate(id, :positive?)
-    end
-  end
-
-  def test_doesnt_interrupt_in_transaction
-    fake_pool do |pool|
-      Pgtk::Impatient.new(pool, 1).transaction do |t|
-        id = t.exec(
-          'INSERT INTO book (title) VALUES ($1) RETURNING id',
-          ['1984']
-        ).first['id'].to_i
-        assert_predicate(id, :positive?)
-      end
-    end
-  end
-end

data/test/test_liquibase_task.rb

@@ -1,74 +0,0 @@
-# frozen_string_literal: true
-
-# SPDX-FileCopyrightText: Copyright (c) 2019-2025 Yegor Bugayenko
-# SPDX-License-Identifier: MIT
-
-require 'tmpdir'
-require 'rake'
-require 'yaml'
-require_relative 'test__helper'
-require_relative '../lib/pgtk/pgsql_task'
-require_relative '../lib/pgtk/liquibase_task'
-
-# Liquibase rake task test.
-# Author:: Yegor Bugayenko (yegor256@gmail.com)
-# Copyright:: Copyright (c) 2017-2025 Yegor Bugayenko
-# License:: MIT
-class TestLiquibaseTask < Pgtk::Test
-  def test_basic
-    Dir.mktmpdir do |dir|
-      Pgtk::PgsqlTask.new(:pgsql2) do |t|
-        t.dir = File.join(dir, 'pgsql')
-        t.user = 'hello'
-        t.password = 'A B C привет ! & | !'
-        t.dbname = 'test'
-        t.yaml = File.join(dir, 'cfg.yml')
-        t.quiet = true
-      end
-      Rake::Task['pgsql2'].invoke
-      Pgtk::LiquibaseTask.new(:liquibase2) do |t|
-        t.master = File.join(__dir__, '../test-resources/master.xml')
-        t.yaml = ['file-is-absent', File.join(dir, 'cfg.yml'), 'another']
-        t.quiet = true
-        t.postgresql_version = '42.7.0'
-        t.liquibase_version = '3.2.2'
-        t.contexts = '!test'
-      end
-      Rake::Task['liquibase2'].invoke
-    end
-  end
-
-  def test_latest_version
-    Dir.mktmpdir do |dir|
-      Pgtk::PgsqlTask.new(:pgsql) do |t|
-        t.dir = File.join(dir, 'pgsql')
-        t.user = 'xxx'
-        t.password = 'xxx'
-        t.dbname = 'xxx'
-        t.yaml = File.join(dir, 'xxx.yml')
-        t.quiet = true
-      end
-      Rake::Task['pgsql'].invoke
-      Pgtk::LiquibaseTask.new(:liquibase) do |t|
-        t.master = File.join(__dir__, '../test-resources/master.xml')
-        t.yaml = File.join(dir, 'xxx.yml')
-        t.postgresql_version = '42.7.1'
-        t.liquibase_version = '4.25.1'
-        t.quiet = true
-      end
-      Rake::Task['liquibase'].invoke
-    end
-  end
-
-  def test_with_invalid_master_file
-    Pgtk::LiquibaseTask.new(:lb) do |t|
-      t.master = 'the-file-doesnt-exist.xml'
-      t.yaml = { 'pgsql' => {} }
-      t.quiet = true
-    end
-    ex = assert_raises(StandardError) do
-      Rake::Task['lb'].invoke
-    end
-    assert_includes(ex.message, 'the-file-doesnt-exist.xml')
-  end
-end

data/test/test_pgsql_task.rb

@@ -1,58 +0,0 @@
-# frozen_string_literal: true
-
-# SPDX-FileCopyrightText: Copyright (c) 2019-2025 Yegor Bugayenko
-# SPDX-License-Identifier: MIT
-
-require 'rake'
-require 'tmpdir'
-require 'yaml'
-require_relative 'test__helper'
-require_relative '../lib/pgtk/pgsql_task'
-
-# Pgsql rake task test.
-# Author:: Yegor Bugayenko (yegor256@gmail.com)
-# Copyright:: Copyright (c) 2017-2025 Yegor Bugayenko
-# License:: MIT
-class TestPgsqlTask < Pgtk::Test
-  def test_basic
-    Dir.mktmpdir do |dir|
-      Pgtk::PgsqlTask.new(:p2) do |t|
-        t.dir = File.join(dir, 'pgsql')
-        t.user = 'hello'
-        t.password = 'A B C привет ! & | !'
-        t.dbname = 'test'
-        t.yaml = File.join(dir, 'cfg.yml')
-        t.quiet = true
-        t.config = {
-          log_min_error_statement: 'ERROR'
-        }
-      end
-      Rake::Task['p2'].invoke
-      yaml = YAML.load_file(File.join(dir, 'cfg.yml'))
-      assert(yaml['pgsql']['url'].start_with?('jdbc:postgresql://localhost'))
-    end
-  end
-
-  def test_not_quiet
-    Dir.mktmpdir do |dir|
-      Pgtk::PgsqlTask.new(:p3) do |t|
-        t.dir = File.join(dir, 'pgsql')
-        t.user = 'hello'
-        t.password = 'the password'
-        t.dbname = 'test'
-        t.yaml = File.join(dir, 'cfg.yml')
-        t.quiet = true
-        t.config = {
-          log_directory: dir,
-          logging_collector: 'on',
-          log_statement: 'all',
-          log_filename: 'pgsql.log'
-        }
-      end
-      Rake::Task['p3'].invoke
-      yaml = YAML.load_file(File.join(dir, 'cfg.yml'))
-      assert(yaml['pgsql']['url'].start_with?('jdbc:postgresql://localhost'))
-      assert_path_exists(File.join(dir, 'pgsql.log'))
-    end
-  end
-end

data/test/test_pool.rb

@@ -1,224 +0,0 @@
-# frozen_string_literal: true
-
-# SPDX-FileCopyrightText: Copyright (c) 2019-2025 Yegor Bugayenko
-# SPDX-License-Identifier: MIT
-
-require 'loog'
-require 'pg'
-require 'qbash'
-require 'rake'
-require 'tmpdir'
-require 'yaml'
-require_relative 'test__helper'
-require_relative '../lib/pgtk/liquibase_task'
-require_relative '../lib/pgtk/pgsql_task'
-require_relative '../lib/pgtk/pool'
-require_relative '../lib/pgtk/spy'
-
-# Pool test.
-# Author:: Yegor Bugayenko (yegor256@gmail.com)
-# Copyright:: Copyright (c) 2017-2025 Yegor Bugayenko
-# License:: MIT
-class TestPool < Pgtk::Test
-  def test_reads_version
-    fake_pool do |pool|
-      ver = pool.version
-      assert(ver.start_with?('1'))
-      refute_includes(ver, ' ')
-    end
-  end
-
-  def test_basic
-    fake_pool do |pool|
-      id = pool.exec(
-        'INSERT INTO book (title) VALUES ($1) RETURNING id',
-        ['Elegant Objects']
-      )[0]['id'].to_i
-      assert_predicate(id, :positive?)
-    end
-  end
-
-  def test_logs_pgsql_errors_to_logger
-    buf = Loog::Buffer.new
-    fake_pool(log: buf) do |pool|
-      pool.exec(
-        "
-        CREATE FUNCTION intentional_failure() RETURNS trigger AS
-        'BEGIN
-          IF NEW.title = ''War and War'' THEN
-            RAISE EXCEPTION ''The title of the book is bad'';
-          END IF;
-          RETURN NEW;
-        END' LANGUAGE PLPGSQL
-        "
-      )
-      pool.exec(
-        "
-        CREATE TRIGGER check_book_title
-        BEFORE INSERT ON book
-        FOR EACH ROW EXECUTE PROCEDURE intentional_failure();
-        "
-      )
-      assert_raises(PG::RaiseException) do
-        pool.exec('INSERT INTO book (title) VALUES ($1)', ['War and War'])
-      end
-      assert_includes(buf.to_s, 'The title of the book is bad')
-      assert_includes(buf.to_s, 'function intentional_failure() line 3')
-    end
-  end
-
-  def test_queries_with_block
-    fake_pool do |pool|
-      pool.exec('INSERT INTO book (title) VALUES ($1)', ['1984'])
-      rows = []
-      pool.exec('SELECT * FROM book') do |row|
-        rows.append(row)
-      end
-      assert_equal(1, rows.size)
-    end
-  end
-
-  def test_with_spy
-    queries = []
-    fake_pool do |pool|
-      pool = Pgtk::Spy.new(pool) { |sql| queries.append(sql) }
-      pool.exec(
-        ['INSERT INTO book', '(title) VALUES ($1)'],
-        ['Elegant Objects']
-      )
-    end
-    assert_equal(1, queries.size)
-    assert_equal('INSERT INTO book (title) VALUES ($1)', queries.first)
-  end
-
-  def test_complex_query
-    fake_pool do |pool|
-      pool.exec(
-        "
-        INSERT INTO book (title) VALUES ('one');
-        INSERT INTO book (title) VALUES ('two');
-        "
-      )
-    end
-  end
-
-  def test_logs_sql
-    log = Loog::Buffer.new
-    fake_pool(log: log) do |pool|
-      pool.exec(
-        'INSERT INTO book (title) VALUES ($1)',
-        ['Object Thinking']
-      )
-      assert_includes(log.to_s, 'INSERT INTO book (title) VALUES ($1)')
-    end
-  end
-
-  def test_logs_errors
-    log = Loog::Buffer.new
-    fake_pool(log: log) do |pool|
-      assert_raises(PG::UndefinedTable) do
-        pool.exec('INSERT INTO tableDoesNotExist (a) VALUES (42)')
-      end
-      assert_includes(log.to_s, 'INSERT INTO tableDoesNotExist')
-    end
-  end
-
-  def test_transaction
-    fake_pool do |pool|
-      id = Pgtk::Spy.new(pool).transaction do |t|
-        t.exec('DELETE FROM book')
-        t.exec(
-          [
-            'INSERT INTO book (title)',
-            'VALUES ($1) RETURNING id'
-          ],
-          ['Object Thinking']
-        )[0]['id'].to_i
-      end
-      assert_predicate(id, :positive?)
-    end
-  end
-
-  def test_transaction_with_error
-    fake_pool do |pool|
-      pool.exec('DELETE FROM book')
-      assert_empty(pool.exec('SELECT * FROM book'))
-      assert_raises(StandardError) do
-        pool.transaction do |t|
-          t.exec('INSERT INTO book (title) VALUES ($1)', ['hey'])
-          t.exec('INSERT INTO book (error_here) VALUES ($1)', ['hey'])
-        end
-      end
-      assert_empty(pool.exec('SELECT * FROM book'))
-      pool.exec('INSERT INTO book (title) VALUES ($1)', ['another'])
-      refute_empty(pool.exec('SELECT * FROM book'))
-    end
-  end
-
-  def test_reconnects_on_pg_error
-    fake_pool do |pool|
-      assert_raises PG::UndefinedTable do
-        pool.exec('SELECT * FROM thisiserror')
-      end
-      5.times do
-        pool.exec('SELECT * FROM book')
-      end
-    end
-  end
-
-  def test_reconnects_on_pg_reboot
-    port = RandomPort::Pool::SINGLETON.acquire
-    Dir.mktmpdir do |dir|
-      id = rand(100..999)
-      Pgtk::PgsqlTask.new("pgsql#{id}") do |t|
-        t.dir = File.join(dir, 'pgsql')
-        t.user = 'hello'
-        t.password = 'A B C привет ! & | !'
-        t.dbname = 'test'
-        t.yaml = File.join(dir, 'cfg.yml')
-        t.quiet = true
-        t.fresh_start = true
-        t.port = port
-      end
-      task = Rake::Task["pgsql#{id}"]
-      task.invoke
-      pool = Pgtk::Pool.new(
-        Pgtk::Wire::Yaml.new(File.join(dir, 'cfg.yml')),
-        log: Loog::NULL
-      )
-      pool.start(1)
-      pool.exec('SELECT * FROM pg_catalog.pg_tables')
-      qbash("pg_ctl -D #{Shellwords.escape(File.join(dir, 'pgsql'))} stop", log: nil)
-      cycle = 0
-      loop do
-        begin
-          TCPSocket.new('localhost', port)
-          sleep(0.1)
-          cycle += 1
-          if cycle > 50
-            qbash('ps -ax | grep postgres')
-            raise "Can't stop running postgres at port #{port}, for some reason"
-          end
-        rescue StandardError => e
-          puts e.message
-          break
-        end
-      end
-      assert_raises(PG::UnableToSend, PG::ConnectionBad) do
-        pool.exec('SELECT * FROM pg_catalog.pg_tables')
-      end
-      task.reenable
-      task.invoke
-      loop do
-        begin
-          pool.exec('SELECT * FROM pg_catalog.pg_tables')
-          break
-        rescue StandardError => e
-          puts e.message
-          sleep(0.1)
-          retry
-        end
-      end
-    end
-  end
-end