Fix test: StdinTestCase

Signed-off-by: Nico Schottelius <nico@brief.schottelius.org>

cdist/test/autorequire/__init__.py

@@ -42,7 +42,6 @@ class AutorequireTestCase(test.CdistTestCase):
     def setUp(self):
         self.orig_environ = os.environ
         os.environ = os.environ.copy()
-        self.target_host = 'localhost'
         self.temp_dir = self.mkdtemp()
 
         self.out_dir = os.path.join(self.temp_dir, "out")

cdist/test/emulator/__init__.py

@@ -20,6 +20,7 @@
 #
 #
 
+import io
 import os
 import shutil
 import string
@@ -141,7 +142,6 @@ class ArgumentsTestCase(test.CdistTestCase):
 
     def setUp(self):
         self.temp_dir = self.mkdtemp()
-        self.target_host = 'localhost'
         out_path = self.temp_dir
         handle, self.script = self.mkstemp(dir=self.temp_dir)
         os.close(handle)
@@ -236,38 +236,59 @@ class StdinTestCase(test.CdistTestCase):
     def setUp(self):
         self.orig_environ = os.environ
         os.environ = os.environ.copy()
-        self.target_host = 'localhost'
+
         self.temp_dir = self.mkdtemp()
-        os.environ['__cdist_out_dir'] = self.temp_dir
+        out_path = os.path.join(self.temp_dir, "out")
 
-        self.context = cdist.context.Context(
+        self.local = local.Local(
             target_host=self.target_host,
-            remote_copy='scp -o User=root -q',
+            out_path=out_path,
-            remote_exec='ssh -o User=root -q',
             exec_path=test.cdist_exec_path,
-            debug=False)
+            add_conf_dirs=[conf_dir])
-        self.config = config.Config(self.context)
 
+        self.local.create_files_dirs()
+
+        self.manifest = core.Manifest(
+            target_host=self.target_host,
+            local = self.local)
+ 
     def tearDown(self):
         os.environ = self.orig_environ
         shutil.rmtree(self.temp_dir)
 
     def test_file_from_stdin(self):
-        handle, destination = self.mkstemp(dir=self.temp_dir)
+        """
-        os.close(handle)
+        Test whether reading from stdin works
-        source_handle, source = self.mkstemp(dir=self.temp_dir)
+        """
-        candidates = string.ascii_letters+string.digits
-        with os.fdopen(source_handle, 'w') as fd:
-            for x in range(100):
-                fd.write(''.join(random.sample(candidates, len(candidates))))
 
-        handle, initial_manifest = self.mkstemp(dir=self.temp_dir)
+        ######################################################################
-        with os.fdopen(handle, 'w') as fd:
+        # Create string with random content
-            fd.write('__file_from_stdin %s --source %s\n' % (destination, source))
+        random_string = str(random.sample(range(1000), 800))
-        self.context.initial_manifest = initial_manifest
+        random_buffer = io.BytesIO(bytes(random_string, 'utf-8'))
-        self.config.stage_prepare()
 
-        cdist_type = core.CdistType(self.config.local.type_path, '__file')
+        ######################################################################
-        cdist_object = core.CdistObject(cdist_type, self.config.local.object_path, destination)
+        # Prepare required args and environment for emulator
-        # Test weither stdin has been stored correctly
+        type_name = '__file'
-        self.assertTrue(filecmp.cmp(source, os.path.join(cdist_object.absolute_path, 'stdin')))
+        object_id = "cdist-test-id"
+        argv = [type_name, object_id]
+
+        initial_manifest_path = "/cdist-test/path/that/does/not/exist"
+        env = self.manifest.env_initial_manifest(initial_manifest_path)
+
+        ######################################################################
+        # Create path where stdin should reside at
+        cdist_type = core.CdistType(self.local.type_path, type_name)
+        cdist_object = core.CdistObject(cdist_type, self.local.object_path, object_id)
+        stdin_out_path = os.path.join(cdist_object.absolute_path, 'stdin')
+
+        ######################################################################
+        # Run emulator
+        emu = emulator.Emulator(argv, stdin=random_buffer, env=env)
+        emu.run()
+
+        ######################################################################
+        # Read where emulator should have placed stdin
+        with open(stdin_out_path, 'r') as fd:
+            stdin_saved_by_emulator = fd.read()
+
+        self.assertEqual(random_string, stdin_saved_by_emulator)

cdist/test/emulator/fixtures/conf/type/__file

@@ -1 +0,0 @@
-../../../../../conf/type/__file