require_relative '../test/helper'

describe DBRotator do
  after do
    `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.split.each do |schema|
      if /^#{TEST_SCHEMA_PREFIX}/.match(schema)
        `#{TEST_MYSQL_COMMAND} -e "DROP SCHEMA #{schema}"`
      end
    end
  end

  def dbname_with(time)
    "__minitest_#{ time.strftime(DBRotator::TIME_FORMAT) }"
  end

  describe "integration test" do
    it "rotates databases correctly" do
      dbr = DBRotator.new(dummy_config)

      t1 = Time.new(2013, 1, 1)
      t2 = Time.new(2013, 1, 2)
      t3 = Time.new(2013, 1, 3)

      Time.stub(:now, t1) do
        dbr.rotate
        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t1))
        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW TABLES FROM #{dbname_with(t1)};'`.must_include("test")
      end

      Time.stub(:now, t2) do
        dbr.rotate
        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t2))
        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t1))
      end

      Time.stub(:now, t3) do
        dbr.rotate
        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t3))
        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.must_include(dbname_with(t2))

        `#{TEST_MYSQL_COMMAND} -B -e 'SHOW SCHEMAS;'`.wont_include(dbname_with(t1))
      end
    end
  end

  describe "callbacks" do
    before do
      `test -e /tmp/success && rm /tmp/success`
      `test -e /tmp/failure && rm /tmp/failure`
    end

    describe "on_success" do
      it "calls on_success when specified" do
        dbr = DBRotator.new(dummy_config(on_success: 'touch /tmp/success'))
        dbr.rotate
        FileTest.exists?('/tmp/success').must_equal true
        FileTest.exists?('/tmp/failure').must_equal false
      end
    end

    describe "on_failure" do
      it "calls on_failure" do
        dbr = DBRotator.new(dummy_config(
                                   on_failure: 'touch /tmp/failure && true',
                                   mysql_command: 'crap command 2> /dev/null'))
        dbr.rotate
        FileTest.exists?('/tmp/success').must_equal false
        FileTest.exists?('/tmp/failure').must_equal true
      end
    end
  end
end