diff options
author | Ashley Penney <ashley.penney@puppetlabs.com> | 2014-03-07 19:48:40 -0500 |
---|---|---|
committer | Ashley Penney <ashley.penney@puppetlabs.com> | 2014-03-07 19:48:40 -0500 |
commit | 9e1cf99ccf12de2edaf73af73a8410e5cac6dd72 (patch) | |
tree | 2624c5c96549324719fbf6eaca781888d6d4646d /spec/lib/puppet_spec/compiler.rb | |
parent | fecb53d46ed9e926973cdf5be1289c1ea71c2f68 (diff) | |
parent | 3854e076ccb75d1bcb1ddd29f5976b194d857765 (diff) |
Merge pull request #230 from apenney/fix-testing
[WIP] Spec overhaul.
Diffstat (limited to 'spec/lib/puppet_spec/compiler.rb')
-rw-r--r-- | spec/lib/puppet_spec/compiler.rb | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/spec/lib/puppet_spec/compiler.rb b/spec/lib/puppet_spec/compiler.rb new file mode 100644 index 0000000..22e923d --- /dev/null +++ b/spec/lib/puppet_spec/compiler.rb @@ -0,0 +1,46 @@ +module PuppetSpec::Compiler + def compile_to_catalog(string, node = Puppet::Node.new('foonode')) + Puppet[:code] = string + Puppet::Parser::Compiler.compile(node) + end + + def compile_to_ral(manifest) + catalog = compile_to_catalog(manifest) + ral = catalog.to_ral + ral.finalize + ral + end + + def compile_to_relationship_graph(manifest, prioritizer = Puppet::Graph::SequentialPrioritizer.new) + ral = compile_to_ral(manifest) + graph = Puppet::Graph::RelationshipGraph.new(prioritizer) + graph.populate_from(ral) + graph + end + + if Puppet.version.to_f >= 3.3 + def apply_compiled_manifest(manifest, prioritizer = Puppet::Graph::SequentialPrioritizer.new) + transaction = Puppet::Transaction.new(compile_to_ral(manifest), + Puppet::Transaction::Report.new("apply"), + prioritizer) + transaction.evaluate + transaction.report.finalize_report + + transaction + end + else + def apply_compiled_manifest(manifest) + transaction = Puppet::Transaction.new(compile_to_ral(manifest), Puppet::Transaction::Report.new("apply")) + transaction.evaluate + transaction.report.finalize_report + + transaction + end + end + + def order_resources_traversed_in(relationships) + order_seen = [] + relationships.traverse { |resource| order_seen << resource.ref } + order_seen + end +end |