Module: BoltSpec::Plans

Defined in:
lib/bolt_spec/plans.rb,
lib/bolt_spec/plans/action_stubs.rb,
lib/bolt_spec/plans/mock_executor.rb,
lib/bolt_spec/plans/action_stubs/task_stub.rb,
lib/bolt_spec/plans/action_stubs/script_stub.rb,
lib/bolt_spec/plans/action_stubs/upload_stub.rb,
lib/bolt_spec/plans/action_stubs/command_stub.rb

Defined Under Namespace

Classes: ActionDouble, ActionStub, CommandStub, MockExecutor, MockPuppetDBClient, ScriptStub, TaskStub, UnexpectedInvocation, UploadStub

Constant Summary collapse

MOCKED_ACTIONS = 
%i[command script task upload].freeze

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.init ⇒ Object 



131
132
133
134
135
136
137
138
139
 
# File 'lib/bolt_spec/plans.rb', line 131

def self.init
  # Ensure tasks are enabled when rspec-puppet sets up an environment so we get task loaders.
  # Note that this is probably not safe to do in modules that also test Puppet manifest code.
  Bolt::PAL.load_puppet
  Puppet[:tasks] = true

  # Ensure logger is initialized with Puppet levels so 'notice' works when running plan specs.
  Logging.init :debug, :info, :notice, :warn, :error, :fatal, :any
end

Instance Method Details

#allow_apply ⇒ Object 



209
210
211
212
 
# File 'lib/bolt_spec/plans.rb', line 209

def allow_apply
  executor.stub_apply
  nil
end

#allow_apply_prep ⇒ Object 



204
205
206
207
 
# File 'lib/bolt_spec/plans.rb', line 204

def allow_apply_prep
  allow_task('apply_helpers::custom_facts')
  nil
end

#allow_get_resources ⇒ Object 



214
215
216
217
 
# File 'lib/bolt_spec/plans.rb', line 214

def allow_get_resources
  allow_task('apply_helpers::query_resources')
  nil
end

#config ⇒ Object

Override in your tests



149
150
151
152
153
154
155
 
# File 'lib/bolt_spec/plans.rb', line 149

def config
  @config ||= begin
    conf = Bolt::Config.new(Bolt::Boltdir.new('.'), {})
    conf.modulepath = [modulepath].flatten
    conf
  end
end

#executor ⇒ Object

intended to be private below here



244
245
246
 
# File 'lib/bolt_spec/plans.rb', line 244

def executor
  @executor ||= BoltSpec::Plans::MockExecutor.new(modulepath)
end

#inventory ⇒ Object

Override in your tests



158
159
160
 
# File 'lib/bolt_spec/plans.rb', line 158

def inventory
  @inventory ||= Bolt::Inventory.new({})
end

#modulepath ⇒ Object

Override in your tests if needed



142
143
144
145
146
 
# File 'lib/bolt_spec/plans.rb', line 142

def modulepath
  [RSpec.configuration.module_path]
rescue NoMethodError
  raise "RSpec.configuration.module_path not defined set up rspec puppet or define modulepath for this test"
end

#puppetdb_client ⇒ Object 



165
166
167
 
# File 'lib/bolt_spec/plans.rb', line 165

def puppetdb_client
  @puppetdb_client ||= MockPuppetDBClient.new
end

#run_plan(name, params) ⇒ Object 



169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
 
# File 'lib/bolt_spec/plans.rb', line 169

def run_plan(name, params)
  pal = Bolt::PAL.new(config.modulepath, config.hiera_config)
  result = pal.run_plan(name, params, executor, inventory, puppetdb_client)

  if executor.error_message
    raise executor.error_message
  end

  begin
    executor.assert_call_expectations
  rescue StandardError => e
    raise "#{e.message}\nPlan result: #{result}"
  end

  result
end