From 9d2ec219bbd77bfca48a72b52fe5d0d3fcc0dcf7 Mon Sep 17 00:00:00 2001 From: Pieter van de Bruggen Date: Sun, 17 Apr 2011 22:23:44 -0700 Subject: (#7013) Add support for required options. This adds another hook into the generated wrapper, which invokes a method to validate arguments. This is used to raise an exception when required options have not been passed to the method. Reviewed-By: Daniel Pittman --- spec/unit/interface/option_builder_spec.rb | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'spec/unit/interface/option_builder_spec.rb') diff --git a/spec/unit/interface/option_builder_spec.rb b/spec/unit/interface/option_builder_spec.rb index b32b316f6..e9346852c 100755 --- a/spec/unit/interface/option_builder_spec.rb +++ b/spec/unit/interface/option_builder_spec.rb @@ -56,5 +56,20 @@ describe Puppet::Interface::OptionBuilder do end end.to raise_error ArgumentError, /takes three arguments/ end + + it "should support simple required declarations" do + opt = Puppet::Interface::OptionBuilder.build(face, "--foo") do + required + end + opt.should be_required + end + + it "should support arguments to the required property" do + opt = Puppet::Interface::OptionBuilder.build(face, "--foo") do + required(false) + end + opt.should_not be_required + end + end end -- cgit