From 79a4339e1f614647d8b3e074752d8bf0e583aef2 Mon Sep 17 00:00:00 2001
From: Thomas Bellman <bellman@nsc.liu.se>
Date: Tue, 4 Aug 2009 14:47:12 +0200
Subject: Add shellquote() function.

This adds a new function shellquote() which can be used for quoting
arguments in shell commands used in the exec type.

This only supports Unixoid operating systems.  Other systems would
likely require some other quoting.

Signed-off-by: Thomas Bellman <bellman@nsc.liu.se>
---
 spec/unit/parser/functions/shellquote.rb | 92 ++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100755 spec/unit/parser/functions/shellquote.rb

(limited to 'spec/unit/parser')

diff --git a/spec/unit/parser/functions/shellquote.rb b/spec/unit/parser/functions/shellquote.rb
new file mode 100755
index 000000000..283a4de1e
--- /dev/null
+++ b/spec/unit/parser/functions/shellquote.rb
@@ -0,0 +1,92 @@
+#! /usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../../spec_helper'
+
+describe "the shellquote function" do
+
+    before :each do
+        @scope = Puppet::Parser::Scope.new()
+    end
+
+    it "should exist" do
+        Puppet::Parser::Functions.function("shellquote").should == "function_shellquote"
+    end
+
+
+    it "should handle no arguments" do
+        result = @scope.function_shellquote([])
+        result.should(eql(""))
+    end
+
+    it "should handle several simple arguments" do
+        result = @scope.function_shellquote(
+            ['foo', 'bar@example.com', 'localhost:/dev/null', 'xyzzy+-4711,23'])
+        result.should(eql(
+            'foo bar@example.com localhost:/dev/null xyzzy+-4711,23'))
+    end
+
+    it "should handle array arguments" do
+        result = @scope.function_shellquote(
+            ['foo', ['bar@example.com', 'localhost:/dev/null'],
+             'xyzzy+-4711,23'])
+        result.should(eql(
+            'foo bar@example.com localhost:/dev/null xyzzy+-4711,23'))
+    end
+
+    it "should quote unsafe characters" do
+        result = @scope.function_shellquote(
+            ['/etc/passwd ', '(ls)', '*', '[?]', "'&'"])
+        result.should(eql(
+            '"/etc/passwd " "(ls)" "*" "[?]" "\'&\'"'))
+    end
+
+    it "should deal with double quotes" do
+        result = @scope.function_shellquote(
+           ['"foo"bar"'])
+        result.should(eql(
+            '\'"foo"bar"\''))
+    end
+
+    it "should cope with dollar signs" do
+        result = @scope.function_shellquote(
+           ['$PATH', 'foo$bar', '"x$"'])
+        result.should(eql(
+            "'$PATH' 'foo$bar' '\"x$\"'"))
+    end
+
+    it "should deal with apostrophes (single quotes)" do
+        result = @scope.function_shellquote(
+           ["'foo'bar'", "`$'EDITOR'`"])
+        result.should(eql(
+            '"\'foo\'bar\'" "\\`\\$\'EDITOR\'\\`"'))
+    end
+
+    it "should cope with grave accents (backquotes)" do
+        result = @scope.function_shellquote(
+           ['`echo *`', '`ls "$MAILPATH"`'])
+        result.should(eql(
+            "'`echo *`' '`ls \"$MAILPATH\"`'"))
+    end
+
+    it "should deal with both single and double quotes" do
+        result = @scope.function_shellquote(
+           ['\'foo"bar"xyzzy\'', '"foo\'bar\'xyzzy"'])
+        result.should(eql(
+            '"\'foo\\"bar\\"xyzzy\'" "\\"foo\'bar\'xyzzy\\""'))
+    end
+
+    it "should handle multiple quotes *and* dollars and backquotes" do
+        result = @scope.function_shellquote(
+           ['\'foo"$x`bar`"xyzzy\''])
+        result.should(eql(
+            '"\'foo\\"\\$x\\`bar\\`\\"xyzzy\'"'))
+    end
+
+    it "should handle linefeeds" do
+        result = @scope.function_shellquote(
+           ["foo \n bar"])
+        result.should(eql(
+            "\"foo \n bar\""))
+    end
+
+end
-- 
cgit