Added script handling to appliance config handling.

Added test cases for this.

1 files changed, 42 insertions, 0 deletions

diff --git a/tests.py b/tests.py
index fa3383a2..33c45ec7 100644
--- a/tests.py
+++ b/tests.py
@@ -20,6 +20,48 @@
 
 import unittest
 
+class RunUserDataApplianceConfigScript(unittest.TestCase):
+    def setUp(self):
+        self.ec2_run_user_data = __import__('ec2-run-user-data')
+        self.fake_handle_shell_script_counter = 0
+        self.expected_scripts = []
+        # Override install_remove_package 
+        self.ec2_run_user_data.content_type_handlers['text/x-shellscript'] = self.fake_handle_shell_script
+
+    def fake_handle_shell_script(self, txt):
+        self.fake_handle_shell_script_counter += 1
+        self.assertEqual(self.expected_scripts.pop(0), txt)
+
+    def handle_xml(self, xml):
+        msg = self.ec2_run_user_data.parse_user_data(xml)
+        self.ec2_run_user_data.handle_part(msg)
+
+    def testApplianceConfigPackageScriptSingle(self):
+        script = '''#!/bin/sh
+echo hey'''
+        xml = '<appliance><script>%s</script></appliance>' % script
+        self.expected_scripts += [script]
+        self.handle_xml(xml)
+        self.assertEqual(self.fake_handle_shell_script_counter, 1) 
+
+    def testApplianceConfigPackageScriptMultiple(self):
+        script1 = '''#!/bin/sh
+echo hey'''
+        script2 = '''#!/usr/bin/python
+print "hey"'''
+        xml = '<appliance><script>%s</script><script>%s</script></appliance>' % (script1, script2)
+        self.expected_scripts += [script1, script2]
+        self.handle_xml(xml)
+        self.assertEqual(self.fake_handle_shell_script_counter, 2) 
+
+    def testApplianceConfigPackageScriptCDATA(self):
+        script = '''#!/bin/sh
+echo hey'''
+        xml = '<appliance><script><![CDATA[%s]]></script></appliance>' % (script, )
+        self.expected_scripts += [script]
+        self.handle_xml(xml)
+        self.assertEqual(self.fake_handle_shell_script_counter, 1) 
+
 class RunUserDataApplianceConfigPackageHandling(unittest.TestCase):
     def setUp(self):
         self.fake_install_remove_package_counter = 0