summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--tests.py149
1 files changed, 0 insertions, 149 deletions
diff --git a/tests.py b/tests.py
deleted file mode 100644
index d9583390..00000000
--- a/tests.py
+++ /dev/null
@@ -1,149 +0,0 @@
-#!/usr/bin/python
-#
-# Unit tests for EC2-init
-# Copyright (C) 2008-2009 Canonical Ltd.
-#
-# Author: Soren Hansen <soren@canonical.com>
-#
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU General Public License version 3, as
-# published by the Free Software Foundation.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU General Public License for more details.
-#
-# You should have received a copy of the GNU General Public License
-# along with this program. If not, see <http://www.gnu.org/licenses/>.
-#
-
-import re
-import os
-import unittest
-
-class RunUserDataApplianceTestCase(unittest.TestCase):
- def handle_xml(self, xml):
- msg = self.ec2_run_user_data.parse_user_data(xml)
- self.ec2_run_user_data.handle_part(msg)
-
-class RunUserDataApplianceConfigEBS(RunUserDataApplianceTestCase):
- def setUp(self):
- self.ec2_run_user_data = __import__('ec2-run-user-data')
- reload(self.ec2_run_user_data)
- self.real_mount_ebs_volume = self.ec2_run_user_data.mount_ebs_volume
- self.ec2_run_user_data.mount_ebs_volume = self.fake_mount_ebs_volume
-
- def fake_mount_ebs_volume(self, device, paths):
- self.assertEqual(device, '/dev/sdc')
- self.assertEqual(paths, ['/etc/alfresco', '/var/lib/mysql'])
-
- def testApplianceConfigEBS(self):
- os.environ['EBSMOUNT_DEBUG'] = 'yes, please'
- xml = '<appliance><storage device="/dev/sdc"><path>/etc/alfresco</path><path>/var/lib/mysql</path></storage></appliance>'
- self.handle_xml(xml)
-
- def testMountEBSVolume(self):
- output = self.real_mount_ebs_volume('/dev/sdh', ['/foo', '/bar'])
- lines = output.strip().split('\n')
- self.assertEqual(len(lines), 11)
- match = re.match('mount /dev/sdh (/var/run/ec2/tmp.[a-zA-Z0-9]+)', lines[0])
- self.assertNotEqual(match, None)
- tmpdir = match.group(1)
- for (i, s) in zip(range(10), ['mkdir %s/_foo', 'cp -a /foo %s/_foo', 'chown --reference /foo %s/_foo', 'chmod --reference /foo %s/_foo', 'mount --bind %s/_foo /foo', 'mkdir %s/_bar', 'cp -a /bar %s/_bar', 'chown --reference /bar %s/_bar', 'chmod --reference /bar %s/_bar', 'mount --bind %s/_bar /bar']):
- self.assertEqual(s % tmpdir, lines[i+1])
-
-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
- 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 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
-
- def fake_install_remove_package(self, package, action):
- self.fake_install_remove_package_counter += 1
- mapping = { 'foobarplus': 'install',
- 'foobarminus': 'remove' }
- self.assert_(package in mapping)
- self.assertEqual(action, mapping[package])
-
- def handle_xml(self, xml):
- msg = self.ec2_run_user_data.parse_user_data(xml)
- self.ec2_run_user_data.handle_part(msg)
-
- def testApplianceConfigPackageInstall(self):
- xml = '<appliance><package>foobarplus</package></appliance>'
- self.handle_xml(xml)
- self.assertEqual(self.fake_install_remove_package_counter, 1)
-
- def testApplianceConfigPackageRemove(self):
- xml = '<appliance><package action="remove">foobarminus</package></appliance>'
- self.handle_xml(xml)
- self.assertEqual(self.fake_install_remove_package_counter, 1)
-
- def testApplianceConfigPackageInstallAndRemove(self):
- xml = '<appliance><package>foobarplus</package><package action="remove">foobarminus</package></appliance>'
- self.handle_xml(xml)
- self.assertEqual(self.fake_install_remove_package_counter, 2)
-
-if __name__ == "__main__":
- unittest.main()