require 'test_helper'

class BulkInsertWorkerTest < ActiveSupport::TestCase
  setup do
    @insert = BulkInsert::Worker.new(
      Testing.connection,
      Testing.table_name,
      %w(greeting age happy created_at updated_at color))
    @now = Time.now
  end

  test "empty insert is not pending" do
    assert_equal false, @insert.pending?
  end

  test "pending_count should describe size of pending set" do
    assert_equal 0, @insert.pending_count
    @insert.add ["Hello", 15, true, @now, @now]
    assert_equal 1, @insert.pending_count
  end

  test "default set size" do
    assert_equal 500, @insert.set_size
  end

  test "adding row to insert makes insert pending" do
    @insert.add ["Hello", 15, true, @now, @now]
    assert_equal true, @insert.pending?
  end

  test "add should default timestamp columns to current time" do
    now = Time.now

    @insert.add ["Hello", 15, true]
    @insert.save!

    record = Testing.first
    assert_operator record.created_at, :>=, now
    assert_operator record.updated_at, :>=, now
  end

  test "default timestamp columns should be equivalent for the entire batch" do
    @insert.add ["Hello", 15, true]
    @insert.add ["Howdy", 20, false]
    @insert.save!

    first, second = Testing.all
    assert_equal first.created_at.to_f, second.created_at.to_f
    assert_equal first.created_at.to_f, first.updated_at.to_f
  end

  test "add should use database default values when present" do
    @insert.add greeting: "Hello", age: 20, happy: false
    @insert.save!

    record = Testing.first
    assert_equal record.color, "chartreuse"
  end

  test "explicit nil should override defaults" do
    @insert.add greeting: "Hello", age: 20, happy: false, color: nil
    @insert.save!

    record = Testing.first
    assert_nil record.color
  end

  test "add should allow values given as Hash" do
    @insert.add greeting: "Yo", age: 20, happy: false, created_at: @now, updated_at: @now
    @insert.save!

    record = Testing.first
    assert_not_nil record
    assert_equal "Yo", record.greeting
    assert_equal 20, record.age
    assert_equal false, record.happy?
  end

  test "add should save automatically when overflowing set size" do
    @insert.set_size = 1
    @insert.add ["Hello", 15, true, @now, @now]
    @insert.add ["Yo", 20, false, @now, @now]
    assert_equal 1, Testing.count
    assert_equal "Hello", Testing.first.greeting
  end

  test "add_all should append all items to the set" do
    @insert.add_all [
      [ "Hello", 15, true ],
      { greeting: "Hi", age: 55, happy: true }
    ]
    assert_equal 2, @insert.pending_count
  end

  test "save! makes insert not pending" do
    @insert.add ["Hello", 15, true, @now, @now]
    @insert.save!
    assert_equal false, @insert.pending?
  end

  test "save! when not pending should do nothing" do
    assert_no_difference 'Testing.count' do
      @insert.save!
    end
  end

  test "save! inserts pending records" do
    @insert.add ["Yo", 15, false, @now, @now]
    @insert.add ["Hello", 25, true, @now, @now]
    @insert.save!

    yo = Testing.find_by(greeting: 'Yo')
    hello = Testing.find_by(greeting: 'Hello')

    assert_not_nil yo
    assert_equal 15, yo.age
    assert_equal false, yo.happy?

    assert_not_nil hello
    assert_equal 25, hello.age
    assert_equal true, hello.happy?
  end

  test "save! calls the after_save handler" do
    x = 41

    @insert.after_save do
      x += 1
    end

    @insert.add ["Yo", 15, false, @now, @now]
    @insert.add ["Hello", 25, true, @now, @now]
    @insert.save!

    assert_equal 42, x
  end

  test "after_save stores a block as a proc" do
    @insert.after_save do
      "hello"
    end

    assert_equal "hello", @insert.after_save_callback.()
  end

  test "after_save_callback can be set as a proc" do
    @insert.after_save_callback = -> do
      "hello"
    end

    assert_equal "hello", @insert.after_save_callback.()
  end
end