require 'spec_helper'
require 'support/time_machine/structure'

# STI cases
#
# - https://github.com/ifad/chronomodel/issues/5
# - https://github.com/ifad/chronomodel/issues/47
#
describe 'models with STI' do
  include ChronoTest::TimeMachine::Helpers

  adapter.create_table 'elements', :temporal => true do |t|
    t.string :title
    t.string :type
  end

  class ::Element < ActiveRecord::Base
    include ChronoModel::TimeMachine
  end

  class ::Publication < Element
  end

  describe '.descendants' do
    subject { Element.descendants }

    it { is_expected.to_not include(Element::History) }
    it { is_expected.to     include(Publication) }
  end

  describe '.descendants_with_history' do
    subject { Element.descendants_with_history }

    it { is_expected.to include(Element::History) }
    it { is_expected.to include(Publication) }
  end

  describe 'timeline' do
    let(:publication) do
      pub = ts_eval { Publication.create! :title => 'wrong title' }
      ts_eval(pub) { update_attributes! :title => 'correct title' }

      pub
    end

    it { expect(publication.history.map(&:title)).to eq ['wrong title', 'correct title'] }
  end

  describe 'identity' do
    adapter.create_table 'animals', temporal: true do |t|
      t.string :type
    end

    class ::Animal < ActiveRecord::Base
      include ChronoModel::TimeMachine
    end

    class ::Dog < Animal
    end

    class ::Goat < Animal
    end

    before do
      Dog.create!
      @later = Time.new
      Goat.create!
    end

    after do
      tables = ['temporal.animals', 'history.animals']
      ActiveRecord::Base.connection.execute "truncate #{tables.join(', ')} cascade"
    end

    specify "select" do
      expect(Animal.first).to be_a(Animal)
      expect(Animal.as_of(@later).first).to be_a(Animal)

      expect(Animal.where(type: 'Dog').first).to be_a(Dog)
      expect(Dog.first).to be_a(Dog)
      expect(Dog.as_of(@later).first).to be_a(Dog)

      expect(Animal.where(type: 'Goat').first).to be_a(Goat)
      expect(Goat.first).to be_a(Goat)
      expect(Goat.as_of(@later).first).to be(nil)
      expect(Goat.as_of(Time.now).first).to be_a(Goat)
    end

    specify "count" do
      expect(Animal.count).to eq(2)
      expect(Animal.as_of(@later).count).to eq(1)

      expect(Dog.count).to eq(1)
      expect(Dog.as_of(@later).count).to eq(1)

      expect(Goat.count).to eq(1)
      expect(Goat.as_of(@later).count).to eq(0)
    end
  end
end