pgtk 0.18.1 → 0.18.2

data/test/test_retry.rb

@@ -1,232 +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/retry'
-
-# Retry test.
-# Author:: Yegor Bugayenko (yegor256@gmail.com)
-# Copyright:: Copyright (c) 2017-2025 Yegor Bugayenko
-# License:: MIT
-class TestRetry < Pgtk::Test
-  def test_takes_version
-    fake_pool do |pool|
-      v = Pgtk::Retry.new(pool).version
-      refute_nil(v)
-    end
-  end
-
-  def test_executes_select_without_error
-    fake_pool do |pool|
-      retry_pool = Pgtk::Retry.new(pool, attempts: 3)
-      result = retry_pool.exec('SELECT 1 as value')
-      assert_equal('1', result.first['value'])
-    end
-  end
-
-  def test_retries_select_on_failure
-    fake_pool do |pool|
-      counter = 0
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |sql, *args|
-        counter += 1
-        raise PG::Error, 'Connection lost' if counter < 3
-        pool.exec(sql, *args)
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 3)
-      result = retry_pool.exec('SELECT 2 as num')
-      assert_equal('2', result.first['num'])
-      assert_equal(3, counter)
-    end
-  end
-
-  def test_fails_after_max_attempts
-    fake_pool do |_pool|
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |_sql, *_args|
-        raise PG::Error, 'Persistent failure'
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 2)
-      assert_raises(PG::Error) do
-        retry_pool.exec('SELECT * FROM users')
-      end
-    end
-  end
-
-  def test_does_not_retry_insert
-    fake_pool do |_pool|
-      counter = 0
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |_sql, *_args|
-        counter += 1
-        raise PG::Error, 'Insert failed'
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 3)
-      assert_raises(PG::Error) do
-        retry_pool.exec('INSERT INTO book (title) VALUES ($1)', ['Test Book'])
-      end
-      assert_equal(1, counter)
-    end
-  end
-
-  def test_does_not_retry_update
-    fake_pool do |_pool|
-      counter = 0
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |_sql, *_args|
-        counter += 1
-        raise PG::Error, 'Update failed'
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 3)
-      assert_raises(PG::Error) do
-        retry_pool.exec('UPDATE book SET title = $1 WHERE id = $2', ['New Title', 1])
-      end
-      assert_equal(1, counter)
-    end
-  end
-
-  def test_does_not_retry_delete
-    fake_pool do |_pool|
-      counter = 0
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |_sql, *_args|
-        counter += 1
-        raise PG::Error, 'Delete failed'
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 3)
-      assert_raises(PG::Error) do
-        retry_pool.exec('DELETE FROM book WHERE id = $1', [1])
-      end
-      assert_equal(1, counter)
-    end
-  end
-
-  def test_handles_select_with_leading_whitespace
-    fake_pool do |pool|
-      counter = 0
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |sql, *args|
-        counter += 1
-        raise PG::Error, 'Connection lost' if counter < 2
-        pool.exec(sql, *args)
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 3)
-      result = retry_pool.exec('  SELECT 3 as value')
-      assert_equal('3', result.first['value'])
-      assert_equal(2, counter)
-    end
-  end
-
-  def test_handles_select_case_insensitive
-    fake_pool do |pool|
-      counter = 0
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |sql, *args|
-        counter += 1
-        raise PG::Error, 'Connection lost' if counter < 2
-        pool.exec(sql, *args)
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 3)
-      result = retry_pool.exec('select 4 as value')
-      assert_equal('4', result.first['value'])
-      assert_equal(2, counter)
-    end
-  end
-
-  def test_handles_array_sql
-    fake_pool do |pool|
-      counter = 0
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |sql, *args|
-        counter += 1
-        raise PG::Error, 'Connection lost' if counter < 2
-        pool.exec(sql, *args)
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 3)
-      result = retry_pool.exec(%w[SELECT 5 as value])
-      assert_equal('5', result.first['value'])
-      assert_equal(2, counter)
-    end
-  end
-
-  def test_transaction_passes_through
-    fake_pool do |pool|
-      retry_pool = Pgtk::Retry.new(pool)
-      retry_pool.transaction do |t|
-        id = t.exec(
-          'INSERT INTO book (title) VALUES ($1) RETURNING id',
-          ['Transaction Book']
-        ).first['id'].to_i
-        assert_predicate(id, :positive?)
-      end
-    end
-  end
-
-  def test_preserves_original_error_type
-    fake_pool do |_pool|
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |_sql, *_args|
-        raise ArgumentError, 'Invalid argument'
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 2)
-      assert_raises(ArgumentError) do
-        retry_pool.exec('SELECT * FROM table')
-      end
-    end
-  end
-
-  def test_retries_with_unicode_query
-    fake_pool do |pool|
-      counter = 0
-      stub_pool = Object.new
-      def stub_pool.version
-        'stub'
-      end
-      stub_pool.define_singleton_method(:exec) do |sql, *args|
-        counter += 1
-        raise PG::Error, 'Connection lost' if counter < 2
-        pool.exec(sql, *args)
-      end
-      retry_pool = Pgtk::Retry.new(stub_pool, attempts: 3)
-      result = retry_pool.exec('SELECT \'привет\' as greeting')
-      assert_equal('привет', result.first['greeting'])
-      assert_equal(2, counter)
-    end
-  end
-end

data/test/test_stash.rb

@@ -1,108 +0,0 @@
-# frozen_string_literal: true
-
-# SPDX-FileCopyrightText: Copyright (c) 2019-2025 Yegor Bugayenko
-# SPDX-License-Identifier: MIT
-
-require_relative 'test__helper'
-require_relative '../lib/pgtk/pool'
-require_relative '../lib/pgtk/stash'
-
-# Pool test.
-# Author:: Yegor Bugayenko (yegor256@gmail.com)
-# Copyright:: Copyright (c) 2017-2025 Yegor Bugayenko
-# License:: MIT
-class TestStash < Pgtk::Test
-  def test_simple_insert
-    fake_pool do |pool|
-      id = Pgtk::Stash.new(pool).exec(
-        'INSERT INTO book (title) VALUES ($1) RETURNING id',
-        ['Elegant Objects']
-      )[0]['id'].to_i
-      assert_predicate(id, :positive?)
-    end
-  end
-
-  def test_non_trivial_queries
-    fake_pool do |pool|
-      pg = Pgtk::Stash.new(pool)
-      [
-        'VACUUM FULL',
-        'START TRANSACTION',
-        'REINDEX TABLE book',
-        'TRUNCATE book',
-        'CREATE TABLE tmp (id INT)',
-        'ALTER TABLE tmp ADD COLUMN foo INT',
-        'DROP TABLE tmp',
-        'SET client_min_messages TO WARNING',
-        "SET TIME ZONE 'America/Los_Angeles'"
-      ].each do |q|
-        pg.exec(q)
-      end
-    end
-  end
-
-  def test_caching
-    fake_pool do |pool|
-      stash = Pgtk::Stash.new(pool)
-      query = 'SELECT count(*) FROM book'
-      first_result = stash.exec(query)
-      second_result = stash.exec(query)
-      assert_equal(first_result.to_a, second_result.to_a)
-      assert_same(first_result, second_result)
-    end
-  end
-
-  def test_cache_invalidation
-    fake_pool do |pool|
-      stash = Pgtk::Stash.new(pool)
-      query = 'SELECT count(*) FROM book'
-      first_result = stash.exec(query)
-      stash.exec('INSERT INTO book (title) VALUES ($1)', ['New Book'])
-      second_result = stash.exec(query)
-      refute_same(first_result, second_result)
-    end
-  end
-
-  def test_caching_with_params
-    fake_pool do |pool|
-      stash = Pgtk::Stash.new(pool)
-      query = 'SELECT * FROM book WHERE title = $1'
-      first_result = stash.exec(query, ['Elegant Objects'])
-      second_result = stash.exec(query, ['Elegant Objects'])
-      assert_equal(first_result.to_a, second_result.to_a)
-      assert_same(first_result, second_result)
-      different_param_result = stash.exec(query, ['Different Title'])
-      refute_same(first_result, different_param_result)
-    end
-  end
-
-  def test_version
-    fake_pool do |pool|
-      stash = Pgtk::Stash.new(pool)
-      assert_match(/^\d+\.\d+/, stash.version)
-    end
-  end
-
-  def test_transaction
-    fake_pool do |pool|
-      stash = Pgtk::Stash.new(pool)
-      stash.exec('INSERT INTO book (title) VALUES ($1)', ['Transaction Test'])
-      stash.transaction do |tx|
-        result = tx.exec('SELECT title FROM book WHERE title = $1', ['Transaction Test'])
-        assert_equal('Transaction Test', result[0]['title'])
-        true
-      end
-    end
-  end
-
-  def test_start
-    fake_pool do |pool|
-      stash = Pgtk::Stash.new(pool)
-      stash.exec('INSERT INTO book (title) VALUES ($1)', ['Start Test'])
-      new_stash = stash.start(1)
-      assert_instance_of(Pgtk::Stash, new_stash)
-      result = new_stash.exec('SELECT title FROM book WHERE title = $1', ['Start Test'])
-      assert_equal('Start Test', result[0]['title'])
-    end
-  end
-end

data/test/test_wire.rb

@@ -1,37 +0,0 @@
-# frozen_string_literal: true
-
-# SPDX-FileCopyrightText: Copyright (c) 2019-2025 Yegor Bugayenko
-# SPDX-License-Identifier: MIT
-
-require 'cgi'
-require 'yaml'
-require_relative 'test__helper'
-require_relative '../lib/pgtk/wire'
-
-# Wire test.
-# Author:: Yegor Bugayenko (yegor256@gmail.com)
-# Copyright:: Copyright (c) 2017-2025 Yegor Bugayenko
-# License:: MIT
-class TestWire < Pgtk::Test
-  def test_connects
-    fake_config do |f|
-      wire = Pgtk::Wire::Yaml.new(f)
-      c = wire.connection
-      refute_nil(c)
-    end
-  end
-
-  def test_connects_via_env_variable
-    fake_config do |f|
-      c = YAML.load_file(f)['pgsql']
-      v = 'DATABASE_URL'
-      ENV[v] = [
-        "postgres://#{CGI.escape(c['user'])}:#{CGI.escape(c['password'])}",
-        "@#{CGI.escape(c['host'])}:#{CGI.escape(c['port'].to_s)}/#{CGI.escape(c['dbname'])}"
-      ].join
-      wire = Pgtk::Wire::Env.new(v)
-      c = wire.connection
-      refute_nil(c)
-    end
-  end
-end