require 'spec_helper'

provider_class = Puppet::Type.type(:rabbitmq_parameter).provider(:rabbitmqctl)
describe provider_class do
  let(:resource) do
    Puppet::Type.type(:rabbitmq_parameter).new(
      name: 'documentumShovel@/',
      component_name: 'shovel',
      value: {
        'src-uri'    => 'amqp://',
        'src-queue'  => 'my-queue',
        'dest-uri'   => 'amqp://remote-server',
        'dest-queue' => 'another-queue'
      }
    )
  end
  let(:provider) { provider_class.new(resource) }

  after do
    described_class.instance_variable_set(:@parameters, nil)
  end

  describe '#prefetch' do
    it 'exists' do
      expect(described_class).to respond_to :prefetch
    end

    it 'matches' do
      provider_class.expects(:rabbitmqctl_list).with('vhosts').returns <<-EOT
/
EOT
      provider_class.expects(:rabbitmqctl_list).with('parameters', '-p', '/').returns <<-EOT
shovel  documentumShovel  {"src-uri":"amqp://","src-queue":"my-queue","dest-uri":"amqp://remote-server","dest-queue":"another-queue"}
EOT
      provider_class.prefetch('documentumShovel@/' => resource)
    end
  end

  describe '#instances' do
    it 'exists' do
      expect(described_class).to respond_to :instances
    end

    it 'fail with invalid output from list' do
      provider_class.expects(:rabbitmqctl_list).with('vhosts').returns <<-EOT
/
EOT
      provider.class.expects(:rabbitmqctl_list).with('parameters', '-p', '/').returns 'foobar'
      expect { provider_class.instances }.to raise_error Puppet::Error, %r{cannot parse line from list_parameter}
    end

    it 'return no instance' do
      provider_class.expects(:rabbitmqctl_list).with('vhosts').returns <<-EOT
/
EOT
      provider_class.expects(:rabbitmqctl_list).with('parameters', '-p', '/').returns ''
      instances = provider_class.instances
      expect(instances.size).to eq(0)
    end
    it 'return one instance' do
      provider_class.expects(:rabbitmqctl_list).with('vhosts').returns <<-EOT
/
EOT
      provider_class.expects(:rabbitmqctl_list).with('parameters', '-p', '/').returns <<-EOT
shovel  documentumShovel  {"src-uri":"amqp://","src-queue":"my-queue","dest-uri":"amqp://remote-server","dest-queue":"another-queue"}
EOT
      instances = provider_class.instances
      expect(instances.size).to eq(1)
      expect(instances.map do |prov|
        {
          name: prov.get(:name),
          component_name: prov.get(:component_name),
          value: prov.get(:value)
        }
      end).to eq(
        [
          {
            name: 'documentumShovel@/',
            component_name: 'shovel',
            value: {
              'src-uri'    => 'amqp://',
              'src-queue'  => 'my-queue',
              'dest-uri'   => 'amqp://remote-server',
              'dest-queue' => 'another-queue'
            }
          }
        ]
      )
    end
    # rubocop:enable RSpec/MultipleExpectations
    it 'return multiple instances' do
      provider_class.expects(:rabbitmqctl_list).with('vhosts').returns <<-EOT
/
EOT
      provider_class.expects(:rabbitmqctl_list).with('parameters', '-p', '/').returns <<-EOT
shovel  documentumShovel1  {"src-uri":"amqp://","src-queue":"my-queue","dest-uri":"amqp://remote-server","dest-queue":"another-queue"}
shovel  documentumShovel2  {"src-uri":["amqp://cl1","amqp://cl2"],"src-queue":"my-queue","dest-uri":"amqp://remote-server","dest-queue":"another-queue"}
EOT
      instances = provider_class.instances
      expect(instances.size).to eq(2)
      expect(instances.map do |prov|
        {
          name: prov.get(:name),
          component_name: prov.get(:component_name),
          value: prov.get(:value)
        }
      end).to eq(
        [
          {
            name: 'documentumShovel1@/',
            component_name: 'shovel',
            value: {
              'src-uri'    => 'amqp://',
              'src-queue'  => 'my-queue',
              'dest-uri'   => 'amqp://remote-server',
              'dest-queue' => 'another-queue'
            }
          },
          {
            name: 'documentumShovel2@/',
            component_name: 'shovel',
            value: {
              'src-uri'    => ['amqp://cl1', 'amqp://cl2'],
              'src-queue'  => 'my-queue',
              'dest-uri'   => 'amqp://remote-server',
              'dest-queue' => 'another-queue'
            }
          }
        ]
      )
    end

    it 'return different instances' do
      provider_class.expects(:rabbitmqctl_list).with('vhosts').returns <<-EOT
/
EOT
      provider_class.expects(:rabbitmqctl_list).with('parameters', '-p', '/').returns <<-EOT
shovel  documentumShovel1  {"src-uri":"amqp://","src-queue":"my-queue","dest-uri":"amqp://remote-server","dest-queue":"another-queue"}
federation  documentumFederation2  {"uri":"amqp://","expires":"360000"}
EOT
      instances = provider_class.instances
      expect(instances.size).to eq(2)
      expect(instances.map do |prov|
        {
          name: prov.get(:name),
          component_name: prov.get(:component_name),
          value: prov.get(:value)
        }
      end).to eq(
        [
          {
            name: 'documentumShovel1@/',
            component_name: 'shovel',
            value: {
              'src-uri'    => 'amqp://',
              'src-queue'  => 'my-queue',
              'dest-uri'   => 'amqp://remote-server',
              'dest-queue' => 'another-queue'
            }
          },
          {
            name: 'documentumFederation2@/',
            component_name: 'federation',
            value: {
              'uri'        => 'amqp://',
              'expires'    => '360000'
            }
          }
        ]
      )
    end
    # rubocop:enable RSpec/MultipleExpectations
  end

  describe '#create' do
    it 'create parameter' do
      provider.expects(:rabbitmqctl).with('set_parameter', '-p', '/', 'shovel', 'documentumShovel',
                                          '{"src-uri":"amqp://","src-queue":"my-queue","dest-uri":"amqp://remote-server","dest-queue":"another-queue"}')
      provider.create
    end
  end

  describe '#destroy' do
    it 'destroy parameter' do
      provider.expects(:rabbitmqctl).with('clear_parameter', '-p', '/', 'shovel', 'documentumShovel')
      provider.destroy
    end
  end
end