require 'test_plugin_helper'

class ::ForemanSalt::Api::V2::SaltStatesControllerTest < ActionController::TestCase
  test 'should get index' do
    get :index, {}
    assert_response :success
    assert_template 'api/v2/salt_states/index'
  end

  test 'should show state' do
    state = ForemanSalt::SaltModule.create(:name => 'foo.bar.baz')
    get :show, :id => state.id
    assert_response :success
    assert_template 'api/v2/salt_states/show'
  end

  test 'should create state' do
    post :create, :state => { :name => 'unicorn' }
    assert_response :success
    assert ForemanSalt::SaltModule.find_by_name('unicorn')
    assert_template 'api/v2/salt_states/create'
  end

  test 'should delete state' do
    state = ForemanSalt::SaltModule.create(:name => 'foo.bar.baz')
    assert_difference('ForemanSalt::SaltModule.count', -1) do
      delete :destroy, :id => state.id
    end
    assert_response :success
  end

  context 'importing' do
    setup do
      @proxy = FactoryGirl.create :smart_proxy, :with_salt_feature
      @states = { 'env1' => %w(state1 state2 state3),
                  'env2' => %w(state1 state2) }

      ::ProxyAPI::Salt.any_instance.stubs(:states_list).returns(@states)
    end

    test 'should import' do
      post :import, :smart_proxy_id => @proxy.id

      assert_response :success

      @states.each do |env, states|
        environment = ::ForemanSalt::SaltEnvironment.find_by_name(env)
        assert_empty environment.salt_modules.map(&:name) - states
      end
    end

    test 'should import only from a given environment' do
      post :import, :smart_proxy_id => @proxy.id, :salt_environments => ['env2']
      assert_response :success
      refute ::ForemanSalt::SaltEnvironment.where(:name => 'env1').first
      assert ::ForemanSalt::SaltEnvironment.where(:name => 'env2').first
    end

    test 'should limit actions to add' do
      env   = FactoryGirl.create :salt_environment
      state = FactoryGirl.create :salt_module, :salt_environments => [env]

      post :import, :smart_proxy_id => @proxy.id, :actions => ['add']
      assert_response :success
      assert ::ForemanSalt::SaltModule.where(:id => state).first
      assert ::ForemanSalt::SaltModule.where(:name => 'state1').first
    end

    test 'should limit actions to remove' do
      state = FactoryGirl.create :salt_module
      post :import, :smart_proxy_id => @proxy.id, :actions => ['remove']
      assert_response :success
      refute ::ForemanSalt::SaltModule.where(:id => state).first
      refute ::ForemanSalt::SaltModule.where(:name => 'state1').first
    end

    test 'dryrun should do nothing' do
      post :import, :smart_proxy_id => @proxy.id, :dryrun => true
      assert_response :success
      refute ::ForemanSalt::SaltModule.all.any?
      refute ::ForemanSalt::SaltEnvironment.all.any?
    end
  end
end