require "bundler/inline"

gemfile(true) do
  source "https://rubygems.org"

  git_source(:github) { |repo| "https://github.com/#{repo}.git" }

  gem "rails", '~> 7.1.3'
  gem "pg"
end

require "active_record"
require "minitest/autorun"
require "logger"

# You may need to change your pg settings accordingly
db_config = {
  adapter: 'postgresql',
  database: 'infinityrange_test',
  host: 'localhost', # Change this to your PostgreSQL host
  port: 5432,
  username: 'postgres', # Change this to your PostgreSQL username
  password: 'postgres'  # Change this to your PostgreSQL password
}

begin
  ActiveRecord::Base.establish_connection(db_config.except(:database))
  ActiveRecord::Base.connection.drop_database(db_config[:database]) rescue nil
  ActiveRecord::Base.connection.create_database(db_config[:database])
end
ActiveRecord::Base.establish_connection(db_config)
ActiveRecord::Base.logger = Logger.new(STDOUT)

ActiveRecord::Schema.define do
  create_table :infinity_ranges do |t|
    t.tsrange :date_range
  end
end

class InfinityRange < ActiveRecord::Base; end

class InfinityRangeTest < Minitest::Test
  def setup
    @time = Time.utc(2024)
  end

  def test_date_range_infinity
    # Create a range with infinity end including end 2024-01-01 00:00:00 UTC..
    # There are multiple ways to create a range with infinity end but all of 
    # these give the same result
    # range = (@time..Float::INFINITY)
    # range = (@time..DateTime::Infinity.new)
    # range = (@time..Date::Infinity.new)
    range = (@time..)

    record = InfinityRange.new(date_range: range)

    # Range is including end
    refute_predicate record.date_range, :exclude_end?

    # Save an reload the record to read back the value from db
    record.save!
    record.reload

    # Range is no longer including range and returns 2024-01-01 00:00:00 UTC...
    # but it should be 2024-01-01 00:00:00 UTC..
    assert_equal (@time..), record.date_range
    refute_predicate record.date_range, :exclude_end?

    # To read back excluding end it needs to be stored like this
    assert_equal(
      "[\"#{@time.to_fs(:db)}\",infinity]",
      record.read_attribute_before_type_cast(:date_range),
    )
  end

  # This test is to shows that the range can be inserted directly into the
  # database and then read back as expected
  def test_direct_insert
    # Insert the equivalent of (@time..) directly into the database
    sql = "INSERT INTO infinity_ranges (date_range) VALUES ('[#{@time},infinity]')"
    ActiveRecord::Base.connection.execute(sql)

    record = InfinityRange.last

    assert_equal (@time..), record.date_range
    refute_predicate record.date_range, :exclude_end?

    assert_equal(
      "[\"#{@time.to_fs(:db)}\",infinity]",
      record.read_attribute_before_type_cast(:date_range),
    )
  end

  def test_direct_insert_nil
    # Insert the equivalent of (@time..) directly into the database
    sql = "INSERT INTO infinity_ranges (date_range) VALUES ('[#{@time},]')"
    ActiveRecord::Base.connection.execute(sql)

    record = InfinityRange.last

    assert_equal (@time..), record.date_range
    refute_predicate record.date_range, :exclude_end?

    assert_equal(
      "[\"#{@time.to_fs(:db)}\",infinity]",
      record.read_attribute_before_type_cast(:date_range),
    )
  end
end