rspec.rb

  def set_type
    if @type == 'policy' or @type == ''
      @status = 'policy'
      puts @type
    elsif @type == 'role'
      @status = 'role'
      puts @type
    else
      @status = 'error'
    end
  end
  ## RSPEC
   describe "#set_type" do
      it {is_expected.to respond_to :set_type}
      context '--type is policy' do
        let(:type) { 'policy' }
        it {expect(subject.status).to eq 'policy'}
      end
      context '--type is role' do
        let(:type) { 'role' }
        it {expect(subject.status).to eq 'role'}
      end
      context '--type is nil' do
        let(:type) { '' }
        it {expect(subject.status).to eq 'policy'}
      end
    end