diff options
author | Erik Dalén <dalen@spotify.com> | 2012-08-07 23:36:20 +0200 |
---|---|---|
committer | Jeff McCune <jeff@puppetlabs.com> | 2012-08-14 08:43:36 -0700 |
commit | 247b69cbc01741257c3d3b72ad75bb62325eaf49 (patch) | |
tree | 9c65541213d5c30e91d3b7e413efc49bf586363c /spec/unit/puppet/parser/functions | |
parent | deafe88e02499a7bdababf0b5dc264fcc3edecf1 (diff) | |
download | puppet-stdlib-247b69cbc01741257c3d3b72ad75bb62325eaf49.tar.gz puppet-stdlib-247b69cbc01741257c3d3b72ad75bb62325eaf49.tar.bz2 |
(#15872) Add to_bytes function
Converts a string like "2 MB" to the value in bytes. Useful for
comparisons on facts that return a human readable number instead of
machine readable.
Diffstat (limited to 'spec/unit/puppet/parser/functions')
-rwxr-xr-x | spec/unit/puppet/parser/functions/to_bytes_spec.rb | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/spec/unit/puppet/parser/functions/to_bytes_spec.rb b/spec/unit/puppet/parser/functions/to_bytes_spec.rb new file mode 100755 index 0000000..d1ea4c8 --- /dev/null +++ b/spec/unit/puppet/parser/functions/to_bytes_spec.rb @@ -0,0 +1,58 @@ +#! /usr/bin/env ruby -S rspec + +require 'spec_helper' + +describe "the to_bytes function" do + let(:scope) { PuppetlabsSpec::PuppetInternals.scope } + + it "should exist" do + Puppet::Parser::Functions.function("to_bytes").should == "function_to_bytes" + end + + it "should raise a ParseError if there is less than 1 arguments" do + lambda { scope.function_to_bytes([]) }.should( raise_error(Puppet::ParseError)) + end + + it "should convert kB to B" do + result = scope.function_to_bytes(["4 kB"]) + result.should(eq(4096)) + end + + it "should work without B in unit" do + result = scope.function_to_bytes(["4 k"]) + result.should(eq(4096)) + end + + it "should work without a space before unit" do + result = scope.function_to_bytes(["4k"]) + result.should(eq(4096)) + end + + it "should work without a unit" do + result = scope.function_to_bytes(["5678"]) + result.should(eq(5678)) + end + + it "should convert fractions" do + result = scope.function_to_bytes(["1.5 kB"]) + result.should(eq(1536)) + end + + it "should convert scientific notation" do + result = scope.function_to_bytes(["1.5e2 B"]) + result.should(eq(150)) + end + + it "should do nothing with a positive number" do + result = scope.function_to_bytes([5678]) + result.should(eq(5678)) + end + + it "should should raise a ParseError if input isn't a number" do + lambda { scope.function_to_bytes(["foo"]) }.should( raise_error(Puppet::ParseError)) + end + + it "should should raise a ParseError if prefix is unknown" do + lambda { scope.function_to_bytes(["5 uB"]) }.should( raise_error(Puppet::ParseError)) + end +end |