From 64edfc464038215bbf969ef7a5c775b4524f08af Mon Sep 17 00:00:00 2001
From: Steven Armstrong <steven@icarus.ethz.ch>
Date: Tue, 18 Oct 2011 13:19:37 +0200
Subject: [PATCH] add tests for emulator
Signed-off-by: Steven Armstrong <steven@icarus.ethz.ch>
---
lib/cdist/test/emulator/__init__.py | 75 +++++++++++++++++++++++++++++
1 file changed, 75 insertions(+)
create mode 100644 lib/cdist/test/emulator/__init__.py
diff --git a/lib/cdist/test/emulator/__init__.py b/lib/cdist/test/emulator/__init__.py
new file mode 100644
index 00000000..66e7918a
--- /dev/null
+++ b/lib/cdist/test/emulator/__init__.py
@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+#
+# 2010-2011 Steven Armstrong (steven-cdist at armstrong.cc)
+#
+# This file is part of cdist.
+#
+# cdist is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# cdist 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 cdist. If not, see <http://www.gnu.org/licenses/>.
+#
+#
+
+import os
+import tempfile
+import unittest
+import shutil
+
+from cdist import test
+from cdist.exec import local
+from cdist import emulator
+from cdist import core
+
+local_base_path = test.cdist_base_path
+
+
+class EmulatorTestCase(unittest.TestCase):
+
+ def mkdtemp(self, **kwargs):
+ return tempfile.mkdtemp(prefix='tmp.cdist.test.', **kwargs)
+
+ def mkstemp(self, **kwargs):
+ return tempfile.mkstemp(prefix='tmp.cdist.test.', **kwargs)
+
+ def setUp(self):
+ self.orig_environ = os.environ
+ os.environ = os.environ.copy()
+ self.temp_dir = self.mkdtemp()
+ handle, self.script = self.mkstemp(dir=self.temp_dir)
+ self.target_host = 'localhost'
+ out_path = self.temp_dir
+ self.local = local.Local(self.target_host, local_base_path, out_path)
+ self.local.create_directories()
+ self.env = {
+ 'PATH': "%s:%s" % (self.local.bin_path, os.environ['PATH']),
+ '__target_host': self.target_host,
+ '__global': self.local.out_path,
+ '__cdist_type_base_path': self.local.type_path, # for use in type emulator
+ '__manifest': self.local.manifest_path,
+ '__cdist_manifest': self.script,
+ }
+
+ def tearDown(self):
+ os.environ = self.orig_environ
+ shutil.rmtree(self.temp_dir)
+
+ def test_nonexistent_type_exec(self):
+ argv = ['__does-not-exist']
+ os.environ.update(self.env)
+ self.assertRaises(core.NoSuchTypeError, emulator.Emulator, argv)
+
+ def test_nonexistent_type_requirement(self):
+ argv = ['__file', '/tmp/foobar']
+ os.environ.update(self.env)
+ os.environ['require'] = '__does-not-exist/some-id'
+ emu = emulator.Emulator(argv)
+ self.assertRaises(core.NoSuchTypeError, emu.run)