tests: Add first_stage_test that completes the first stage

Add a test that completes the first stage without any error.

Signed-off-by: Marc Hartmayer <[email protected]>

Author: mhartmay

tests/first_stage_test.py

@@ -50,3 +50,44 @@ def test_valid_syntax(self):
             )
         finally:
             fp.close()
+
+    def test_stage(self):
+        options = mitogen.parent.Options(max_message_size=123)
+        conn = mitogen.parent.Connection(options, self.router)
+        conn.context = mitogen.core.Context(None, 123)
+        args = conn.get_boot_command()
+        preamble = conn.get_preamble()
+
+        # The boot command should write all ECO markers to stdout, read the
+        # preamble from stdin, then execute it.
+
+        # This test writes the preamble to STDIN and closes it then to create an
+        # EOF situation.
+        # 1. Fork child tries to read from STDIN, but stops as EOF is received.
+        # 2. Fork child writes all ECO markers to stdout
+        # TBD
+
+        proc = mitogen.parent.popen(args=args,
+                                    stdout=subprocess.PIPE,
+                                    stderr=subprocess.PIPE,
+                                    stdin=subprocess.PIPE,
+                                    )
+        try:
+            try:
+                stdout, stderr = proc.communicate(input=preamble, timeout=10)
+            except TypeError:
+                stdout, stderr = proc.communicate(input=preamble)
+        except:
+            proc.kill()
+            self.fail("First stage did not finish")
+        else:
+            # proc was killed by SIGTERM?!? TODO Where does this come from?
+            self.assertEqual(-15, proc.returncode)
+            self.assertEqual(stdout,
+                             mitogen.parent.BootstrapProtocol.EC0_MARKER + b('\n') +
+                             mitogen.parent.BootstrapProtocol.EC1_MARKER + b('\n') +
+                             mitogen.parent.BootstrapProtocol.EC2_MARKER + b('\n'))
+            self.assertEqual(
+                b(""),
+                stderr,
+            )