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

@@ -52,3 +52,50 @@ 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 = subprocess.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,
+            )