From 2fd7a079c19ce42c7705e9e87bc5e807d45891b2 Mon Sep 17 00:00:00 2001
From: Soren Hansen <soren@canonical.com>
Date: Fri, 4 Sep 2009 11:16:09 -0500
Subject: Add a unittest for veriyfing that shell scripts passed in user-data
 get called.

---
 tests.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'tests.py')

diff --git a/tests.py b/tests.py
index eb7401f3..d9583390 100644
--- a/tests.py
+++ b/tests.py
@@ -56,6 +56,7 @@ class RunUserDataApplianceConfigEBS(RunUserDataApplianceTestCase):
 class RunUserDataApplianceConfigScript(RunUserDataApplianceTestCase):
     def setUp(self):
         self.ec2_run_user_data = __import__('ec2-run-user-data')
+        reload(self.ec2_run_user_data)
         self.fake_handle_shell_script_counter = 0
         self.expected_scripts = []
         # Override install_remove_package 
@@ -95,11 +96,25 @@ echo hey'''
         self.handle_xml(xml)
         self.assertEqual(self.fake_handle_shell_script_counter, 1) 
 
+class RunUserDataScript(unittest.Testcase):
+    def setUp(self):
+        self.ec2_run_user_data = __import__('ec2-run-user-data')
+        reload(self.ec2_run_user_data)
+
+    def testHandleDetectedScript(self):
+        self.ec2_run_user_data.get_user_data = lambda : '''#!/bin/sh
+touch /tmp/ec2-run-user-data-test
+'''
+        self.ec2_run_user_data.main()
+        self.assertTrue(os.path.exists('/tmp/ec2-run-user-data-test'))
+        os.unlink('/tmp/ec2-run-user-data-test')
+
 class RunUserDataApplianceConfigPackageHandling(RunUserDataApplianceTestCase):
     def setUp(self):
         self.fake_install_remove_package_counter = 0
 
         self.ec2_run_user_data = __import__('ec2-run-user-data')
+        reload(self.ec2_run_user_data)
 
         # Override install_remove_package 
         self.ec2_run_user_data.install_remove_package = self.fake_install_remove_package
-- 
cgit v1.2.3