module RabbotDb::TestSupport

# File lib/rabbot_db/test_support.rb, line 14
def available?
  !test_url.empty?
end

# File lib/rabbot_db/test_support.rb, line 22
def reset!(bot_id: "test")
  raise "RABBOT_TEST_DATABASE_URL is not set" unless available?

  RabbotDb.database_url = test_url
  RabbotDb.current_bot_id = bot_id
  RabbotDb.reset!(url: test_url)
end

# File lib/rabbot_db/test_support.rb, line 18
def test_url
  ENV["RABBOT_TEST_DATABASE_URL"].to_s.strip.empty? ? DEFAULT_TEST_URL : ENV["RABBOT_TEST_DATABASE_URL"].to_s.strip
end

# File lib/rabbot_db/test_support.rb, line 30
def with_test_db(bot_id: "test")
  previous_url = RabbotDb.instance_variable_get(:@database_url)
  previous_bot = RabbotDb.instance_variable_get(:@current_bot_id)
  reset!(bot_id: bot_id)
  yield
ensure
  RabbotDb.database_url = previous_url
  RabbotDb.current_bot_id = previous_bot if previous_bot
end