homebrew-core/Formula/wal2json.rb

class Wal2json < Formula
  desc "Convert PostgreSQL changesets to JSON format"
  homepage "https://github.com/eulerto/wal2json"
  url "https://github.com/eulerto/wal2json/archive/wal2json_1_0.tar.gz"
  sha256 "d8c1b774bc2d41747ae4266280f798b26e8e72e8ea46a58b71764edb98cf2ef4"
  revision 1

  bottle do
    cellar :any_skip_relocation
    sha256 "e3b1a7520d70bac53b3f299785ba60dfc278ef48cd9d032b857eb6209fb59b2c" => :catalina
    sha256 "700660cc4ed295257d890d2ab1b533592283efe27aac65c532a6e959bf7690d7" => :mojave
    sha256 "25684fa793b6c9e4aaec3bf43775cddbb3db894fded3beae35a6560487809d71" => :high_sierra
  end

  depends_on "postgresql"

  def install
    mkdir "stage"
    system "make", "install", "USE_PGXS=1", "DESTDIR=#{buildpath}/stage"
    lib.install Dir["stage/#{HOMEBREW_PREFIX}/lib/*"]
  end

  test do
    system "initdb", testpath/"datadir"
    mkdir testpath/"socket"
    File.open(testpath/"datadir"/"postgresql.conf", "a") do |f|
      f << "wal_level = logical\n"
      f << "listen_addresses = ''\n"
      f << "unix_socket_directories = '#{testpath}/socket'\n"
      f << "dynamic_library_path = '$libdir:#{lib}/postgresql'\n"
    end
    pid = Process.fork { exec "postgres", "-D", testpath/"datadir" }
    sleep 2
    begin
      system "createdb", "-h", testpath/"socket", "test"

      input_sql = <<~EOS
        CREATE TABLE table2_with_pk (a SERIAL, b VARCHAR(30), c TIMESTAMP NOT NULL, PRIMARY KEY(a, c));
        CREATE TABLE table2_without_pk (a SERIAL, b NUMERIC(5,2), c TEXT);

        SELECT 'init' FROM pg_create_logical_replication_slot('test_slot', 'wal2json');

        BEGIN;
        INSERT INTO table2_with_pk (b, c) VALUES('Backup and Restore', '2019-10-08 12:00:00');
        INSERT INTO table2_with_pk (b, c) VALUES('Tuning', '2019-10-08 12:00:00');
        INSERT INTO table2_with_pk (b, c) VALUES('Replication', '2019-10-08 12:00:00');
        DELETE FROM table2_with_pk WHERE a < 3;

        INSERT INTO table2_without_pk (b, c) VALUES(2.34, 'Tapir');
        -- it is not added to stream because there isn't a pk or a replica identity
        UPDATE table2_without_pk SET c = 'Anta' WHERE c = 'Tapir';
        COMMIT;

        SELECT data FROM pg_logical_slot_get_changes('test_slot', NULL, NULL, 'pretty-print', '1');
        SELECT 'stop' FROM pg_drop_replication_slot('test_slot');
      EOS

      File.open(testpath/"input.sql", "w") do |f|
        f.write(input_sql)
      end

      system "psql", "-h", testpath/"socket", "-f", testpath/"input.sql", "-o", testpath/"output.txt", "-Atq", "test"
      actual_output = File.read(testpath/"output.txt")

      expected_output = <<~EOS
        init
        {
		"change": [
			{
				"kind": "insert",
				"schema": "public",
				"table": "table2_with_pk",
				"columnnames": ["a", "b", "c"],
				"columntypes": ["integer", "character varying(30)", "timestamp without time zone"],
				"columnvalues": [1, "Backup and Restore", "2019-10-08 12:00:00"]
			}
			,{
				"kind": "insert",
				"schema": "public",
				"table": "table2_with_pk",
				"columnnames": ["a", "b", "c"],
				"columntypes": ["integer", "character varying(30)", "timestamp without time zone"],
				"columnvalues": [2, "Tuning", "2019-10-08 12:00:00"]
			}
			,{
				"kind": "insert",
				"schema": "public",
				"table": "table2_with_pk",
				"columnnames": ["a", "b", "c"],
				"columntypes": ["integer", "character varying(30)", "timestamp without time zone"],
				"columnvalues": [3, "Replication", "2019-10-08 12:00:00"]
			}
			,{
				"kind": "delete",
				"schema": "public",
				"table": "table2_with_pk",
				"oldkeys": {
					"keynames": ["a", "c"],
					"keytypes": ["integer", "timestamp without time zone"],
					"keyvalues": [1, "2019-10-08 12:00:00"]
				}
			}
			,{
				"kind": "delete",
				"schema": "public",
				"table": "table2_with_pk",
				"oldkeys": {
					"keynames": ["a", "c"],
					"keytypes": ["integer", "timestamp without time zone"],
					"keyvalues": [2, "2019-10-08 12:00:00"]
				}
			}
			,{
				"kind": "insert",
				"schema": "public",
				"table": "table2_without_pk",
				"columnnames": ["a", "b", "c"],
				"columntypes": ["integer", "numeric(5,2)", "text"],
				"columnvalues": [1, 2.34, "Tapir"]
			}
		]
        }
        stop
      EOS

      assert_equal(expected_output.strip, actual_output.strip)
    ensure
      Process.kill 9, pid
      Process.wait pid
    end
  end
end