andyjeffries-rubyrep 1.2.1

data/spec/noisy_connection_spec.rb

@@ -0,0 +1,78 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+include RR
+
+describe NoisyConnection do
+  before(:each) do
+    Initializer.configuration = proxied_config
+    @connection = ProxyConnection.new Initializer.configuration.left
+    @connection.send(:extend, NoisyConnection)
+    @connection.sweeper = TaskSweeper.new(1)
+  end
+
+  it "select_cursor should return correct results" do
+    @connection.sweeper.should_receive(:ping).exactly(4).times
+    @connection.select_record(:table => 'scanner_records').should == {
+      'id' => 1,
+      'name' => 'Alice - exists in both databases'
+    }
+  end
+
+  it "insert_record should write nil values correctly" do
+    @connection.sweeper.should_receive(:ping).exactly(2).times
+    @connection.begin_db_transaction
+    begin
+      @connection.insert_record('extender_combined_key', 'first_id' => 8, 'second_id' => '9', 'name' => nil)
+      @connection.select_one(
+        "select name from extender_combined_key where (first_id, second_id) = (8, 9)"
+      ).should == {"name" => nil}
+    ensure
+      @connection.rollback_db_transaction
+    end
+  end
+
+  it "update_record should update the specified record" do
+    @connection.sweeper.should_receive(:ping).exactly(2).times
+    @connection.begin_db_transaction
+    begin
+      @connection.update_record('scanner_records', 'id' => 1, 'name' => 'update_test')
+      @connection.select_one(
+        "select name from scanner_records where id = 1"
+      ).should == {'name' => 'update_test'}
+    ensure
+      @connection.rollback_db_transaction
+    end
+  end
+
+  it "delete_record should delete the specified record" do
+    @connection.sweeper.should_receive(:ping).exactly(2).times
+    @connection.begin_db_transaction
+    begin
+      @connection.delete_record('extender_combined_key', 'first_id' => 1, 'second_id' => '1', 'name' => 'xy')
+      @connection.select_one(
+        "select first_id, second_id, name
+         from extender_combined_key where (first_id, second_id) = (1, 1)") \
+        .should be_nil
+    ensure
+      @connection.rollback_db_transaction
+    end
+  end
+
+  it "commit_db_transaction should update TaskSweeper" do
+    @connection.begin_db_transaction
+    initializer = ReplicationInitializer.new Session.new(standard_config)
+    begin
+      @connection.execute "insert into scanner_records(id,name) values(99, 'bla')"
+      @connection.sweeper.should_receive(:ping).exactly(2).times
+      @connection.commit_db_transaction
+      initializer.silence_ddl_notices(:left) do # avoid PostgreSQL warning that no transaction is open
+        @connection.rollback_db_transaction
+      end
+      @connection.select_one("select name from scanner_records where id = 99")['name'].
+        should == 'bla'
+    ensure
+      @connection.execute "delete from scanner_records where id = 99"
+    end
+  end
+
+end

data/spec/postgresql_replication_spec.rb

@@ -0,0 +1,48 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+require 'yaml'
+require 'config/test_config.rb'
+
+include RR
+
+describe "PostgreSQLReplication", :shared => true do
+  before(:each) do
+  end
+
+  it "create_replication_trigger should also work if language plpgsql does not yet exist" do
+    session = nil
+    begin
+      session = Session.new
+      session.left.begin_db_transaction
+      unless session.left.select_all("select lanname from pg_language where lanname = 'plpgsql'").empty?
+        session.left.execute "DROP LANGUAGE plpgsql"
+      end
+      params = {
+        :trigger_name => 'rr_trigger_test',
+        :table => 'trigger_test',
+        :keys => ['first_id', 'second_id'],
+        :log_table => 'rr_pending_changes',
+        :key_sep => '|',
+        :exclude_rr_activity => false,
+      }
+      session.left.create_replication_trigger params
+      session.left.insert_record 'trigger_test', {
+        'first_id' => 1,
+        'second_id' => 2,
+        'name' => 'bla'
+      }
+
+      row = session.left.select_one("select * from rr_pending_changes")
+      row.delete 'id'
+      row.delete 'change_time'
+      row.should == {
+        'change_table' => 'trigger_test',
+        'change_key' => 'first_id|1|second_id|2',
+        'change_new_key' => nil,
+        'change_type' => 'I'
+      }
+      
+    ensure
+      session.left.rollback_db_transaction if session
+    end
+  end
+end

data/spec/postgresql_schema_support_spec.rb

@@ -0,0 +1,212 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+include RR
+
+require File.dirname(__FILE__) + "/../config/test_config.rb"
+
+describe "PostgreSQL schema support" do
+  before(:each) do
+    config = deep_copy(standard_config)
+    config.options[:rep_prefix] = 'rx'
+    config.left[:schema_search_path] = 'rr'
+    config.right[:schema_search_path] = 'rr'
+    Initializer.configuration = config
+  end
+
+  after(:each) do
+  end
+
+  if Initializer.configuration.left[:adapter] == 'postgresql'
+    it "tables should show the tables from the schema and no others" do
+      session = Session.new
+      session.left.tables.include?('rr_simple').should be_true
+      session.left.tables.include?('scanner_records').should be_false
+    end
+
+    it "tables should not show the tables from other schemas" do
+      session = Session.new standard_config
+      session.left.tables.include?('scanner_records').should be_true
+      session.left.tables.include?('rr_simple').should be_false
+    end
+
+    it "primary_key_names should work" do
+      session = Session.new
+      session.left.primary_key_names('rr_simple').should == ['id']
+    end
+
+    it "primary_key_names should pick the table in the target schema" do
+      session = Session.new
+      session.left.primary_key_names('rr_duplicate').should == ['id']
+    end
+
+    it "column_names should work" do
+      session = Session.new
+      session.left.column_names('rr_simple').should == ['id', 'name']
+    end
+
+    it "column_names should pick the table in the target schema" do
+      session = Session.new
+      session.left.column_names('rr_duplicate').should == ['id', 'name']
+    end
+
+    it "referenced_tables should work" do
+      session = Session.new
+      session.left.referenced_tables(['rr_referencing']).should == {
+        'rr_referencing' => ['rr_referenced']
+      }
+    end
+
+    it "table_select_query should work" do
+      session = Session.new
+      session.left.table_select_query('rr_simple').
+        should == 'select "id", "name" from "rr_simple" order by "id"'
+    end
+
+    it "TypeCastingCursor should work" do
+      session = Session.new
+      org_cursor = session.left.select_cursor(
+        :query => "select id, name from rr_simple where id = 1",
+        :type_cast => false
+      )
+      cursor = TypeCastingCursor.new session.left, 'rr_simple', org_cursor
+
+      row = cursor.next_row
+
+      row.should == {
+        'id' => 1,
+        'name' => 'bla'
+      }
+    end
+
+    it "sequence_values should pick the table in the target schema" do
+      session = Session.new
+      session.left.sequence_values('rx', 'rr_duplicate').keys.should == ["rr_duplicate_id_seq"]
+    end
+
+    it "clear_sequence_setup should pick the table in the target schema" do
+      session = nil
+      begin
+        session = Session.new
+        initializer = ReplicationInitializer.new(session)
+        session.left.begin_db_transaction
+        session.right.begin_db_transaction
+        table_pair = {:left => 'rr_duplicate', :right => 'rr_duplicate'}
+        initializer.ensure_sequence_setup table_pair, 5, 2, 1
+        id1, id2 = get_example_sequence_values(session, 'rr_duplicate')
+        (id2 - id1).should == 5
+        (id1 % 5).should == 2
+
+        initializer.clear_sequence_setup :left, 'rr_duplicate'
+        id1, id2 = get_example_sequence_values(session, 'rr_duplicate')
+        (id2 - id1).should == 1
+      ensure
+        [:left, :right].each do |database|
+          initializer.clear_sequence_setup database, 'rr_duplicate' rescue nil if session
+          session.send(database).execute "delete from rr_duplicate" rescue nil if session
+          session.send(database).rollback_db_transaction rescue nil if session
+        end
+      end
+
+    end
+
+    it "sequence setup should work" do
+      session = nil
+      begin
+        session = Session.new
+        initializer = ReplicationInitializer.new(session)
+        session.left.begin_db_transaction
+        session.right.begin_db_transaction
+        table_pair = {:left => 'rr_sequence_test', :right => 'rr_sequence_test'}
+        initializer.ensure_sequence_setup table_pair, 5, 2, 1
+        id1, id2 = get_example_sequence_values(session, 'rr_sequence_test')
+        (id2 - id1).should == 5
+        (id1 % 5).should == 2
+      ensure
+        [:left, :right].each do |database|
+          initializer.clear_sequence_setup database, 'rr_sequence_test' rescue nil if session
+          session.send(database).execute "delete from rr_sequence_test" rescue nil if session
+          session.send(database).rollback_db_transaction rescue nil if session
+        end
+      end
+    end
+
+    it "clear_sequence_setup should work" do
+      session = nil
+      begin
+        session = Session.new
+        initializer = ReplicationInitializer.new(session)
+        session.left.begin_db_transaction
+        session.right.begin_db_transaction
+        table_pair = {:left => 'rr_sequence_test', :right => 'rr_sequence_test'}
+        initializer.ensure_sequence_setup table_pair, 5, 2, 2
+        initializer.clear_sequence_setup :left, 'rr_sequence_test'
+        id1, id2 = get_example_sequence_values(session, 'rr_sequence_test')
+        (id2 - id1).should == 1
+      ensure
+        [:left, :right].each do |database|
+          initializer.clear_sequence_setup database, 'rr_sequence_test' if session
+          session.send(database).execute "delete from rr_sequence_test" if session
+          session.send(database).rollback_db_transaction if session
+        end
+      end
+    end
+
+    it "initializer should create tables in target schema" do
+      session = nil
+      begin
+        config = deep_copy(Initializer.configuration)
+        config.options[:rep_prefix] = 'ry'
+        session = Session.new config
+        session.left.begin_db_transaction
+
+        initializer = ReplicationInitializer.new(session)
+        initializer.create_change_log(:left)
+
+        # no exception ==> means table was created in target schema
+        session.left.select_one("select id from rr.ry_pending_changes")
+      ensure
+        session.left.rollback_db_transaction if session
+      end
+    end
+
+    it "create_trigger, trigger_exists? and drop_trigger should work" do
+      session = nil
+      begin
+        session = Session.new
+        initializer = ReplicationInitializer.new(session)
+        session.left.begin_db_transaction
+
+        initializer.create_trigger :left, 'rr_trigger_test'
+        initializer.trigger_exists?(:left, 'rr_trigger_test').
+          should be_true
+
+        # Verify that the trigger can find the pending_changes table even if
+        # current search_path does not include it.
+        session.left.execute "set search_path = 'public'"
+        session.left.execute <<-EOF
+          insert into rr.rr_trigger_test(first_id, second_id) values(10, 11)
+        EOF
+        session.left.execute "set search_path = 'rr'"
+        session.left.select_one("select change_key from rx_pending_changes")['change_key'].
+          should == "first_id|10|second_id|11"
+
+        initializer.drop_trigger(:left, 'rr_trigger_test')
+        initializer.trigger_exists?(:left, 'rr_trigger_test').
+          should be_false
+      ensure
+        session.left.rollback_db_transaction if session
+      end
+    end
+
+    it "should work with complex search paths" do
+      config = deep_copy(standard_config)
+      config.left[:schema_search_path] = 'public,rr'
+      config.right[:schema_search_path] = 'public,rr'
+      session = Session.new(config)
+
+      tables = session.left.tables
+      tables.include?('rr_simple').should be_true
+      tables.include?('scanner_records').should be_true
+    end
+  end
+end

data/spec/postgresql_support_spec.rb

@@ -0,0 +1,63 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+include RR
+
+require File.dirname(__FILE__) + "/../config/test_config.rb"
+
+describe "PostgreSQL support" do
+  before(:each) do
+    Initializer.configuration = standard_config
+  end
+
+  after(:each) do
+  end
+
+  if Initializer.configuration.left[:adapter] == 'postgresql'
+
+    it "should read & write microsecond times" do
+      session = nil
+      begin
+        session = Session.new
+        session.left.begin_db_transaction
+        session.left.insert_record('extender_type_check',
+          {'id' => 2, 'timestamp' => Time.local(2009, "feb", 16, 20, 48, 1, 543)}
+        )
+
+        org_cursor = session.left.select_cursor(
+          :query => "select id, timestamp from extender_type_check where id = 2",
+          :type_cast => false
+        )
+        cursor = TypeCastingCursor.new session.left, 'extender_type_check', org_cursor
+
+        row = cursor.next_row
+        row['timestamp'].usec.should == 543
+        cursor.clear
+      ensure
+        session.left.rollback_db_transaction if session
+      end
+    end
+
+    it "should not round microsecond times to incorrect value" do
+      session = nil
+      begin
+        session = Session.new
+        session.left.begin_db_transaction
+        session.left.insert_record('extender_type_check',
+          {'id' => 2, 'timestamp' => Time.local(2009, "feb", 16, 13, 37, 11, 126291)}
+        )
+
+        org_cursor = session.left.select_cursor(
+          :query => "select id, timestamp from extender_type_check where id = 2",
+          :type_cast => false
+        )
+        cursor = TypeCastingCursor.new session.left, 'extender_type_check', org_cursor
+
+        row = cursor.next_row
+        row['timestamp'].usec.should == 126291
+        cursor.clear
+      ensure
+        session.left.rollback_db_transaction if session
+      end
+    end
+  end
+end

data/spec/progress_bar_spec.rb

@@ -0,0 +1,77 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+include RR
+include ScanProgressPrinters
+
+describe ProgressBar do
+  before(:each) do
+    @org_stdout = $stdout
+    $stdout = StringIO.new
+    @old_arg = ProgressBar.arg
+  end
+
+  after(:each) do
+    ProgressBar.arg = @old_arg
+    $stdout = @org_stdout
+  end
+
+  it "arg should store the command line argument and max_markers return the correct marker number" do
+    ProgressBar.arg = nil
+    ProgressBar.new(100, Session.new(standard_config), 'bla', 'blub').max_markers.should == ProgressBar::MAX_MARKERS
+    ProgressBar.arg = "2"
+    ProgressBar.new(100, Session.new(standard_config), 'bla', 'blub').max_markers.should == 2
+  end
+
+  it "step should use ANSI codes if options :use_ansi is set" do
+    session = Session.new(deep_copy(standard_config))
+    session.configuration.options[:use_ansi] = true
+    bar = ProgressBar.new(10, session, 'bla', 'blub')
+    bar.step 1
+    bar.step 1
+    $stdout.string.should =~ Regexp.new(Regexp.escape("\e[1"))
+  end
+
+  it "step should not use ANSI codes if options :use_ansi is not true" do
+    session = Session.new(deep_copy(standard_config))
+    session.configuration.options[:use_ansi] = false
+    bar = ProgressBar.new(10, session, 'bla', 'blub')
+    bar.step 1
+    bar.step 1
+    $stdout.string.should_not =~ Regexp.new(Regexp.escape("\e[1"))
+  end
+
+  it "should register itself with ScanRunner" do
+    RR::ScanProgressPrinters.printers[:progress_bar][:printer_class].
+      should == ProgressBar
+  end
+
+  it "step should print the correct progress" do
+    bar = ProgressBar.new(1000, Session.new(standard_config), 'bla', 'blub')
+    bar.step 200
+    bar.step 300
+    $stdout.string.count('.').should == ProgressBar::MAX_MARKERS / 2
+    bar.step 500
+    $stdout.string.count('.').should == ProgressBar::MAX_MARKERS
+  end
+
+  it "step should work around weird floating point rounding issues" do
+    bar = ProgressBar.new(56, Session.new(standard_config), 'bla', 'blub')
+    (1..56).each {bar.step 1}
+    $stdout.string.count('.').should == ProgressBar::MAX_MARKERS
+  end
+
+  it "if max_steps is 0, any step call should go to 100% progress" do
+    config = deep_copy(standard_config)
+    config.options[:use_ansi] = true
+    bar = ProgressBar.new(0, Session.new(config), 'bla', 'blub')
+    bar.step
+    $stdout.string.count('.').should == ProgressBar::MAX_MARKERS
+    $stdout.string.should =~ /100%/
+
+    # Ensure we don't go over 100%
+    $stdout = StringIO.new
+    bar.step
+    $stdout.string.should == ''
+  end
+
+end