require 'test_helper'

class AuditTest < ActiveSupport::TestCase
  include ActiveModel::Lint::Tests

  # Use Widget as a fake model to test auditing.
  class Widget
    include ActiveModel::Conversion
    include ActiveModel::Validations
    extend ActiveModel::Naming
    extend ActiveModel::Callbacks

    define_model_callbacks :create, :update, :save

    include Challah::Audit

    attr_accessor :name, :created_by, :created_at, :updated_by, :updated_at

    def initialize(attributes = {})
      attributes.each do |name, value|
        send("#{name}=", value)
      end

      @attributes = {}
    end

    def persisted?
      false
    end

    def new_record?
      !@saved
    end

    def save
      # Saving doesn't do anything, just a stub
      run_callbacks :save do
        @saved = true
      end

      true
    end

    def column_for_attribute(attr_name)
      self.respond_to?("#{attr_name}=") ? attr_name : nil
    end

    def write_attribute(attr_name, value)
      self.send("#{attr_name}=", value)
    end

    # Stub for tests
    def changed_attributes
      {}
    end

    # Stubs
    def attribute_changed?(attr_name)
      false
    end
  end

  # Runs the ActiveModel::Lint tests against a new Widget instance.
  def setup
    @model = Widget.new
  end

  context "An audited model" do
    should "be able to receive a current user" do
      user = create(:user)
      user_two = create(:user, :first_name => 'User', :last_name => 'Test 2')

      assert_equal 0, @model.current_user_id
      assert_equal true, @model.new_record?

      # For a new record, setting current_user should update both attributes
      @model.current_user = user
      assert_equal user.id, @model.current_user_id

      @model.save

      assert_equal user.id, @model.created_by
      assert_equal user.id, @model.updated_by

      assert_equal false, @model.new_record?

      # For an existing record, setting current_user (or current_user_id) should update only updated_by
      @model.current_user_id = user_two.id
      assert_equal user_two.id, @model.current_user_id

      @model.save

      assert_equal user.id, @model.created_by
      assert_equal user_two.id, @model.updated_by
    end

    should "be able to clear audit attributes" do
      @model.created_by = 1
      @model.updated_by = 2

      @model.send(:initialize_dup, nil)

      assert_equal nil, @model.created_by
      assert_equal nil, @model.updated_by
    end
  end
end